Commited backup

CakesTwix/Hikka-Modules/flibusta.py

@@ -0,0 +1,169 @@
+"""
+
+    █▀▀ ▄▀█ █▄▀ █▀▀ █▀ ▀█▀ █░█░█ █ ▀▄▀
+    █▄▄ █▀█ █░█ ██▄ ▄█ ░█░ ▀▄▀▄▀ █ █░█
+
+    Copyleft 2022 t.me/CakesTwix
+    This program is free software; you can redistribute it and/or modify
+
+"""
+
+__version__ = (1, 0, 0)
+
+# meta pic: https://allvpn.ru/assets/upload/t-200x200-7439447981535195421.png
+# meta developer: @cakestwix_mods
+# requires: httpx bs4
+# scope: inline
+
+import logging
+import xml.etree.ElementTree as ET
+from typing import Union
+
+import httpx
+from aiogram.types import (InlineKeyboardButton, InlineKeyboardMarkup,
+                           InlineQueryResultArticle, InputTextMessageContent)
+from bs4 import BeautifulSoup
+
+from .. import loader, utils
+from ..inline import GeekInlineQuery, rand
+
+logger = logging.getLogger(__name__)
+
+# From Hikka https://github.com/hikariatama/Hikka/blob/master/hikka/utils.py#L459-L461
+def chunks(_list: Union[list, tuple, set], n: int, /) -> list:
+    """Split provided `_list` into chunks of `n`"""
+    return [_list[i : i + n] for i in range(0, len(_list), n)]
+
+async def search_book(query=None):
+    books_list = []
+    async with httpx.AsyncClient() as client:
+        if query:
+            xml_ = (await client.get(f"http://flibusta.is/opds/opensearch?searchTerm={query}&searchType=books&pageNumber=1"))
+        else:
+            xml_ = await client.get("http://flibusta.is/opds/opensearch?searchType=books&pageNumber=1")
+
+        myroot = ET.fromstring(xml_.text)
+
+        for book in myroot.findall('.//{http://www.w3.org/2005/Atom}entry'):
+            books_dict = {"Books": {}, "Name": book.find('./{http://www.w3.org/2005/Atom}title').text, "Author": ""}
+
+
+            for item in book.findall('./{http://www.w3.org/2005/Atom}author'):
+                books_dict["Author"] += item.find('./{http://www.w3.org/2005/Atom}name').text + " "
+
+            # Links and Formats
+            for item in book.findall('./{http://www.w3.org/2005/Atom}link'):
+                if "application/" in item.attrib["type"] and "related" not in item.attrib["rel"]:
+                    books_dict["Books"][item.attrib["type"].split("/")[1].replace("+zip", "").replace("x-mobipocket-ebook", "mobi")] = item.attrib["href"]
+
+                # Read
+                if "title" in item.attrib:
+                    books_dict["Read"] = [item.attrib["title"], item.attrib["href"]]
+
+            if books_dict["Books"] != {}:
+                books_list.append(books_dict)
+
+        return books_list
+
+@loader.tds
+class FlibustaMod(loader.Module):
+    """Get books from flibusta"""
+
+    strings = {
+        "name": "Flibusta",
+        "no_args": "🎞 <b>You need to specify book name</b>",
+        "no_book": "🎞 <b>No books by your query :(</b>",
+    }
+
+    # Just commands
+
+    async def bookcmd(self, message):
+        """🔎 Sending the form with the books. Send message with args if you want to find a book by title"""
+        # Get args from message
+        if args:= utils.get_args_raw(message):
+            books = await search_book(args)
+        else:
+            books = await search_book()
+
+        # No books ((
+        if books == []:
+            return await utils.answer(message, self.strings["no_book"])
+
+        string = f'📙 <b>{books[0]["Name"]}</b>'
+        string += f"\n<b>Author:</b> {books[0]['Author']}"
+
+        # epub pdf fb2 mobi / Read
+        reply_keyboard = [[{"text": file_, "url": f"http://flibusta.is{books[0]['Books'][file_]}"} for file_ in books[0]["Books"]], [{"text": "Read", "url": f"http://flibusta.is{books[0]['Read'][1]}"}]]
+
+        btn_buff = [{"text": str(i), "callback": self.change_book__callback, "args": (book, args or None)} for i, book in enumerate(books, start=1)]
+
+        reply_keyboard.extend(iter(chunks(btn_buff, 5)))
+        try:
+            reply_keyboard.remove([])
+        except ValueError:
+            pass
+
+        await self.inline.form(
+            text=string,
+            message=message,
+            reply_markup=reply_keyboard,
+        )
+
+    # Just callbacks
+
+    async def change_book__callback(self, call, book, search=None):
+        books = await search_book(search)
+
+        string = f'📙 <b>{book["Name"]}</b>'
+        string += f"\n<b>Author:</b> {book['Author']}"
+
+        # epub pdf fb2 mobi / Read
+        reply_keyboard = [[{"text": file_, "url": f"http://flibusta.is{book['Books'][file_]}"} for file_ in book["Books"]], [{"text": "Read", "url": f"http://flibusta.is{book['Read'][1]}"}]]
+
+        btn_buff = [{"text": str(i), "callback": self.change_book__callback, "args": (book, search or None)} for i, book in enumerate(books, start=1)]
+
+        reply_keyboard.extend(iter(chunks(btn_buff, 5)))
+        try:
+            reply_keyboard.remove([])
+        except ValueError:
+            pass
+        await call.edit(text=string, reply_markup=reply_keyboard)
+
+    # Just Inline
+
+    async def book_inline_handler(self, query: GeekInlineQuery) -> None:
+        """
+        🔎 Sending the form with the books. Send message with args if you want to find a book by title (Inline)
+        """
+        if text := query.args:
+            books = await search_book(text)
+        else:
+            books = await search_book()
+        
+        # No books ((
+        if books == []:
+            return await query.e404()
+
+        InlineQueryResult = []
+        for book in books:
+            markup = InlineKeyboardMarkup(row_width=3)
+            for file_ in book["Books"]:
+                markup.insert(InlineKeyboardButton(file_, f"http://flibusta.is{book['Books'][file_]}"))
+            InlineQueryResult.append(
+                InlineQueryResultArticle(
+                id=utils.rand(50),
+                title=book["Name"],
+                description=book["Author"],
+                input_message_content=InputTextMessageContent(
+                    f'📙 <b>{book["Name"]}</b>\n<b>Author:</b> {book["Author"]}',
+                    "HTML",
+                    disable_web_page_preview=True,
+                    ),
+                reply_markup=markup,
+                )
+            )
+        
+        await query.answer(InlineQueryResult, cache_time=0)
+
+
+