Commited backup

hikariatama/ftg/waifu.py

@@ -0,0 +1,128 @@
+#             █ █ ▀ █▄▀ ▄▀█ █▀█ ▀
+#             █▀█ █ █ █ █▀█ █▀▄ █
+#              © Copyright 2022
+#           https://t.me/hikariatama
+#
+# 🔒      Licensed under the GNU AGPLv3
+# 🌐 https://www.gnu.org/licenses/agpl-3.0.html
+
+# meta pic: https://static.dan.tatar/waifu_icon.png
+# meta banner: https://mods.hikariatama.ru/badges/waifu.jpg
+# meta developer: @hikarimods
+# scope: inline
+# scope: hikka_only
+# scope: hikka_min 1.2.10
+
+import functools
+import logging
+
+import requests
+from telethon.tl.types import Message
+
+from .. import loader, utils
+
+logger = logging.getLogger(__name__)
+categories = [
+    "waifu",
+    "neko",
+    "shinobu",
+    "megumin",
+    "cuddle",
+    "cry",
+    "hug",
+    "awoo",
+    "kiss",
+    "lick",
+    "pat",
+    "smug",
+    "bonk",
+    "yeet",
+    "blush",
+    "smile",
+    "wave",
+    "highfive",
+    "handhold",
+    "nom",
+    "bite",
+    "glomp",
+    "slap",
+]
+nsfw_categories = ["waifu", "neko", "trap", "blowjob"]
+
+
+async def photo(type_: str, category: str) -> list:
+    if category in nsfw_categories and category not in categories:
+        type_ = "nsfw"
+
+    ans = (
+        await utils.run_sync(
+            requests.post,
+            f"https://api.waifu.pics/many/{type_}/{category}",
+            json={"exclude": []},
+        )
+    ).json()
+
+    if "files" not in ans:
+        logger.error(ans)
+        return []
+
+    return ans["files"]
+
+
+@loader.tds
+class WaifuMod(loader.Module):
+    """Unleash best waifus of all time"""
+
+    strings = {"name": "Waifu"}
+
+    async def waifucmd(self, message: Message):
+        """[nsfw] [category] - Send waifu"""
+        category = (
+            [
+                category
+                for category in categories + nsfw_categories
+                if category in utils.get_args_raw(message)
+            ]
+            or [
+                (
+                    categories[0]
+                    if "nsfw" not in utils.get_args_raw(message)
+                    else nsfw_categories[0]
+                )
+            ]
+        )[0]
+
+        await self.inline.gallery(
+            message=message,
+            next_handler=functools.partial(
+                photo,
+                type_=("nsfw" if "nsfw" in utils.get_args_raw(message) else "sfw"),
+                category=category,
+            ),
+            caption=(
+                f"<b>{('🔞 NSFW' if 'nsfw' in utils.get_args_raw(message) else '👨‍👩‍👧 SFW')}</b>:"
+                f" <i>{category}</i>"
+            ),
+            preload=10,
+        )
+
+    async def waifuscmd(self, message: Message):
+        """Show available categories"""
+        await utils.answer(
+            message,
+            "\n".join(
+                [
+                    " | ".join(i)
+                    for i in utils.chunks([f"<code>{i}</code>" for i in categories], 5)
+                ]
+            )
+            + "\n<b>NSFW:</b>\n"
+            + "\n".join(
+                [
+                    " | ".join(i)
+                    for i in utils.chunks(
+                        [f"<code>{i}</code>" for i in nsfw_categories], 5
+                    )
+                ]
+            ),
+        )