example.py

import asyncio
from os import getenv
from time import time
from typing import List, Union, cast

from FB2 import FictionBook2
from FB2.Author import Author
from httpx import AsyncClient, Timeout

from Classes.Book import Book
from Classes.Dataclasses import Pages
from Classes.Functions import Authorize, Logoff, SetSessionHeaders


async def main():
    async with AsyncClient(base_url=Pages.main) as client:
        client._timeout = Timeout(3)
        SetSessionHeaders(client)
        t = time()
        password = getenv('AUTHOR_TODAY_PASSWORD')
        email = getenv('AUTHOR_TODAY_EMAIL')
        if await Authorize(client, email, password):
            print(f"Authorized as {email}")
            book: Book = Book(client)
            await book.GetBookFromUrl("work/40323")
            print(book.header)
            print(
                "-----------------\nTable of Contents\n-----------------",
                end="\n",
            )
            for chapterHeader in book.header.tableOfContents:
                print(chapterHeader)
            fb2 = FictionBook2()
            fb2.titleInfo.title = book.header.title
            fb2.titleInfo.authors = cast(
                List[Union[str, Author]],
                book.header.authors,
            )
            fb2.titleInfo.annotation = book.header.annotation
            fb2.titleInfo.genres = book.header.genres
            fb2.titleInfo.lang = "ru"
            fb2.titleInfo.sequences = (
                [(book.header.sequence.name, book.header.sequence.number)]
                if book.header.sequence
                else None
            )
            fb2.titleInfo.keywords = book.header.tags
            fb2.titleInfo.coverPageImages = (
                [book.header.coverImageData]
                if book.header.coverImageData
                else None
            )
            fb2.titleInfo.date = (book.header.publicationDate, None)

            fb2.chapters = list(
                map(
                    lambda chapter: (chapter.header.title, chapter.paragraphs),
                    book.chapters,
                )
            )
            fb2.write(f"./Output/{fb2.titleInfo.title}.fb2")
            await Logoff(client)
        print(f"All requests took {time() - t} seconds.")


asyncio.get_event_loop().run_until_complete(main())