AutoAnimeBot v2.0

Author: TechShreyash

.gitignore

@@ -2,4 +2,8 @@ __pycache__/config.cpython-39.pyc
 *.pyc
 *.session
 *.session-journal
-main/modules/text.py
+main/modules/text.py
+logs.txt
+.vscode/launch.json
+logs.txt
+delete all.py

AutoAnimeBot/__init__.py

@@ -0,0 +1,3 @@
+from AutoAnimeBot.core.bot import AutoAnimeBot
+
+app = AutoAnimeBot()

AutoAnimeBot/__main__.py

@@ -0,0 +1,45 @@
+import asyncio
+from pyrogram import filters
+from pyrogram.types import Message
+
+from uvloop import install
+from contextlib import closing, suppress
+from pyrogram import idle
+import AutoAnimeBot.modules.vote
+from AutoAnimeBot import app
+
+loop = asyncio.get_event_loop()
+
+
+@app.on_message(filters.command(["start", "help", "ping"]))
+async def start(bot, message: Message):
+    await message.reply_photo(
+        "assets/thumb.jpg",
+        caption="⭐️ **Bot Is Online...**\n\n**Updates :** @TechZBots **| Support :** @TechZBots_Support",
+    )
+
+
+@app.on_message(filters.command("logs"))
+async def logs(bot, message: Message):
+    await message.reply_document(
+        "logs.txt",
+        caption="AutoAnimeBot Logs, Send this to @TechZBots_Support if you need help",
+    )
+
+
+async def main():
+    await app.start()
+
+    await idle()
+    app.logger.info("BOT STOPPED")
+    await app.stop()
+    for task in asyncio.all_tasks():
+        task.cancel()
+
+
+if __name__ == "__main__":
+    install()
+    with closing(loop):
+        with suppress(asyncio.exceptions.CancelledError):
+            loop.run_until_complete(main())
+            loop.run_until_complete(asyncio.sleep(3.0))

AutoAnimeBot/core/bot.py

@@ -0,0 +1,71 @@
+from pyrogram import Client
+from config import *
+import asyncio
+from AutoAnimeBot.modules.parser import auto_parser
+from pyrogram.errors import MessageNotModified
+from AutoAnimeBot.modules.tg_handler import tg_handler
+from AutoAnimeBot.inline import button1
+from pyrogram.types import Message
+from AutoAnimeBot.core.log import LOGGER
+
+
+class AutoAnimeBot(Client):
+    def __init__(self):
+        self.logger = LOGGER("AutoAnimeBot")
+        self.logger.info("Starting AutoAnimeBot")
+        super().__init__(
+            "AutoAnimeBot", api_id=int(API_ID), api_hash=API_HASH, bot_token=BOT_TOKEN
+        )
+
+    async def start(self):
+        await super().start()
+
+        self.logger.info("Getting Channel IDs")
+        self.INDEX_CHANNEL_ID = (await self.get_chat(INDEX_CHANNEL_USERNAME)).id
+        self.UPLOADS_CHANNEL_ID = (await self.get_chat(UPLOADS_CHANNEL_USERNAME)).id
+
+        self.logger.info("Getting Status Messages")
+        self.queue = []
+        self.status: Message = await self.get_messages(
+            self.UPLOADS_CHANNEL_ID, int(STATUS_MSG_ID)
+        )
+        self.schedule: Message = await self.get_messages(
+            self.UPLOADS_CHANNEL_ID, int(SCHEDULE_MSG_ID)
+        )
+
+        self.logger.info("==================================")
+        self.logger.info("AutoAnimeBot Started Bot Successfully")
+        self.logger.info("==========JOIN @TECHZBOTS=========")
+
+        self.logger.info("Adding Parsing Task")
+        asyncio.create_task(auto_parser(TECHZ_API_KEY, self))
+        asyncio.create_task(tg_handler(self, TECHZ_API_KEY))
+
+    async def update_status(self, text):
+        try:
+            logger = LOGGER("Status")
+            logger.info(text)
+            text = self.status_text(text)
+            await self.status.edit_text(text, reply_markup=button1)
+        except MessageNotModified:
+            pass
+        except Exception as e:
+            logger.warning(str(e))
+
+    def status_text(self, text):
+        stat = """
+⭐️ **Status :** {}
+
+⏳ **Queue :** 
+
+{}
+    """
+
+        queue_text = ""
+        for i in self.queue[:10]:
+            queue_text += "📌 " + i.replace("-", " ").title().strip() + "\n"
+
+        if queue_text == "":
+            queue_text = "❌ Empty"
+
+        return stat.format(text, queue_text)

AutoAnimeBot/core/log.py

@@ -0,0 +1,18 @@
+import logging
+from logging.handlers import RotatingFileHandler
+
+logging.basicConfig(
+    level=logging.INFO,
+    format="[%(asctime)s - %(levelname)s] - %(name)s - %(message)s",
+    datefmt="%d-%b-%y %H:%M:%S",
+    handlers=[
+        RotatingFileHandler("logs.txt", maxBytes=5000000, backupCount=10),
+        logging.StreamHandler(),
+    ],
+)
+
+logging.getLogger("pyrogram").setLevel(logging.ERROR)
+
+
+def LOGGER(name: str) -> logging.Logger:
+    return logging.getLogger(name)

AutoAnimeBot/inline.py

@@ -0,0 +1,36 @@
+from config import (
+    INDEX_CHANNEL_USERNAME,
+    SCHEDULE_MSG_ID,
+    STATUS_MSG_ID,
+    UPLOADS_CHANNEL_USERNAME,
+    COMMENTS_GROUP_LINK,
+)
+from pyrogram.types import InlineKeyboardButton, InlineKeyboardMarkup
+
+button1 = InlineKeyboardMarkup(
+    [
+        [
+            InlineKeyboardButton(
+                text="View Schedule",
+                url=f"https://t.me/{UPLOADS_CHANNEL_USERNAME}/{SCHEDULE_MSG_ID}",
+            )
+        ],
+        [
+            InlineKeyboardButton(
+                text="Index Channel", url=f"https://t.me/{INDEX_CHANNEL_USERNAME}"
+            ),
+            InlineKeyboardButton(text="Discussion Group", url=COMMENTS_GROUP_LINK),
+        ],
+    ]
+)
+
+button2 = InlineKeyboardMarkup(
+    [
+        [
+            InlineKeyboardButton(
+                text="Check Queue",
+                url=f"https://t.me/{UPLOADS_CHANNEL_USERNAME}/{STATUS_MSG_ID}",
+            )
+        ]
+    ]
+)

main/modules/anilist.py AutoAnimeBot/modules/anilist.pymain/modules/anilist.py renamed to AutoAnimeBot/modules/anilist.py

@@ -1,11 +1,5 @@
-import asyncio
-from main.modules.utils import format_text
+from AutoAnimeBot.modules.utils import format_text
 import requests
-import time
-import os
-from bs4 import BeautifulSoup
-from datetime import datetime
-from string import digits
 
 ANIME_QUERY = """
 query ($id: Int, $idMal:Int, $search: String) {
@@ -62,21 +56,24 @@
 
 ANIME_DB = {}
 
+
 async def return_json_senpai(query: str, vars_: dict):
     url = "https://graphql.anilist.co"
     anime = vars_["search"]
     db = ANIME_DB.get(anime)
 
     if db:
-      return db
+        return db
     data = requests.post(url, json={"query": query, "variables": vars_}).json()
     ANIME_DB[anime] = data
 
     return data
 
+
 temp = []
 
-async def get_anime(vars_,less):
+
+async def get_anime(vars_, less):
     if 1 == 1:
         result = await return_json_senpai(ANIME_QUERY, vars_)
 
@@ -88,8 +85,8 @@ async def get_anime(vars_,less):
             data = temp[0]
             temp.pop(0)
         else:
-          data = result["data"]["Media"]   
-          temp.append(data)
+            data = result["data"]["Media"]
+            temp.append(data)
         idm = data.get("id")
         title = data.get("title")
         tit = title.get("english")
@@ -98,43 +95,46 @@ async def get_anime(vars_,less):
 
         tit = format_text(tit)
         title_img = f"https://img.anili.st/media/{idm}"
-        
+
         if less == True:
-          return idm, title_img, tit
+            return idm, title_img, tit
 
         return data
 
+
 async def get_anime_img(query):
     vars_ = {"search": query}
-    idm, title_img, title = await get_anime(vars_,less=True)
 
-    return idm, title_img, title
-    
+    return await get_anime(vars_, less=True)
+
+
 def get_anime_name(title):
     x = title.split(" - ")[-1]
-    x = title.replace(x,'').replace('-','').strip()
-    x = x.split(' ')
+    x = title.replace(x, "").replace("-", "").strip()
+    x = x.split(" ")
     x = x[:4]
-    y = ''
+    y = ""
     for i in x:
-        y += i + ' '
+        y += i + " "
     return y
 
+
 atext = """
 📺 **{}**
   ({})
 
-🎭 : {}
-🧬 : {}
-📡 : {}
-🗓 : {}
-💾 : {}
-⭐️ : {}/100
+🎭 Genre : `{}`
+🧬 Type : `{}`
+📡 Status : `{}`
+🗓 Episodes : `{}`
+💾 Duration : `{}`
+⭐️ Rating : `{}/100`
 """
 
+
 async def get_anilist_data(name):
     vars_ = {"search": name}
-    data = await get_anime(vars_,less=False)
+    data = await get_anime(vars_, less=False)
 
     id_ = data.get("id")
     title = data.get("title")
@@ -152,41 +152,33 @@ async def get_anilist_data(name):
     title2 = title.get("romaji")
 
     if title2 == None:
-      title2 = title.get("native")
+        title2 = title.get("native")
 
     if title1 == None:
-      title1 = title2
+        title1 = title2
 
     # genre
 
     genre = ""
 
     for i in genres:
-      genre += i + ", "
+        genre += i + ", "
 
     genre = genre[:-2]
 
-
     caption = atext.format(
-      title1,
-      title2,
-      genre,
-      form,
-      status,
-      episodes,
-      duration,
-      averageScore
+        title1, title2, genre, form, status, episodes, duration, averageScore
     )
 
     if trailer != None:
-      ytid = trailer.get("id")
-      site = trailer.get("site")
+        ytid = trailer.get("id")
+        site = trailer.get("site")
     else:
-      site = None
+        site = None
 
     if site == "youtube":
-      caption += f"\n[Trailer](https://www.youtube.com/watch?v={ytid}) | [More Info](https://anilist.co/anime/{id_})"
+        caption += f"\n[Trailer](https://www.youtube.com/watch?v={ytid}) | [More Info](https://anilist.co/anime/{id_})"
     else:
-      caption += f"\n[More Info](https://anilist.co/anime/{id_})"
+        caption += f"\n[More Info](https://anilist.co/anime/{id_})"
 
     return img, caption