Update bot.py

bot.py

@@ -3,16 +3,14 @@ import os, math, logging, datetime, pytz, logging.config
 from aiohttp import web
 from pyrogram import Client, types
 from database.users_chats_db import db
-from database.ia_filterdb import  Media
+from database.ia_filterdb import Media
 from typing import Union, Optional, AsyncGenerator
 from utils import temp, __repo__, __license__, __copyright__, __version__
 from info import API_ID, API_HASH, BOT_TOKEN, LOG_CHANNEL, UPTIME, WEB_SUPPORT, LOG_MSG
 
 # Get logging configurations
 logging.config.fileConfig("logging.conf")
-logging.getLogger(__name__).setLevel(logging.INFO)
-logging.getLogger("cinemagoer").setLevel(logging.ERROR)
-
+logger = logging.getLogger(__name__)
 
 class Bot(Client):
     def __init__(self):
@@ -23,14 +21,21 @@ class Bot(Client):
             bot_token=BOT_TOKEN,
             plugins=dict(root="plugins")
         )
+        logger.info("Bot initialized.")
 
     async def start(self):
+        logger.info("Starting bot...")
         b_users, b_chats = await db.get_banned()
         temp.BANNED_USERS = b_users
-        temp.BANNED_CHATS = b_chats        
-        
+        temp.BANNED_CHATS = b_chats
+        logger.info("Banned users and chats loaded.")
+
         await super().start()
+        logger.info("Pyrogram client started.")
+
         await Media.ensure_indexes()
+        logger.info("Indexes ensured for Media collection.")
+
         me = await self.get_me()
         temp.U_NAME = me.username
         temp.B_NAME = me.first_name
@@ -43,37 +48,39 @@ class Bot(Client):
         curr = datetime.datetime.now(pytz.timezone("Asia/Kolkata"))
         date = curr.strftime('%d %B, %Y')
         tame = curr.strftime('%I:%M:%S %p')
-        logging.info(LOG_MSG.format(me.first_name, date, tame, __repo__, __version__, __license__, __copyright__))
-        
-        try: await self.send_message(LOG_CHANNEL, text=LOG_MSG.format(me.first_name, date, tame, __repo__, __version__, __license__, __copyright__), disable_web_page_preview=True)   
-        except Exception as e: logging.warning(f"Bot Isn't Able To Send Message To LOG_CHANNEL \n{e}")
-        
+        log_message = LOG_MSG.format(me.first_name, date, tame, __repo__, __version__, __license__, __copyright__)
+        logger.info(log_message)
+
+        try:
+            await self.send_message(LOG_CHANNEL, text=log_message, disable_web_page_preview=True)
+            logger.info("Log message sent to LOG_CHANNEL.")
+        except Exception as e:
+            logger.warning(f"Bot Isn't Able To Send Message To LOG_CHANNEL \n{e}")
+
         if bool(WEB_SUPPORT) is True:
             app = web.AppRunner(web.Application(client_max_size=30000000))
             await app.setup()
             await web.TCPSite(app, "0.0.0.0", 8080).start()
-            logging.info("Web Response Is Running......🕸️")
-            
+            logger.info("Web Response Is Running......🕸️")
+
     async def stop(self, *args):
+        logger.info("Stopping bot...")
         await super().stop()
-        logging.info(f"Bot Is Restarting ⟳...")
+        logger.info(f"Bot Is Restarting ⟳...")
 
-    async def iter_messages(self, chat_id: Union[int, str], limit: int, offset: int = 0) -> Optional[AsyncGenerator["types.Message", None]]:                       
+    async def iter_messages(self, chat_id: Union[int, str], limit: int, offset: int = 0) -> Optional[AsyncGenerator["types.Message", None]]:
+        logger.info(f"Iterating messages in chat_id: {chat_id}, limit: {limit}, offset: {offset}")
         current = offset
         while True:
             new_diff = min(200, limit - current)
             if new_diff <= 0:
+                logger.info("No more messages to iterate.")
                 return
             messages = await self.get_messages(chat_id, list(range(current, current+new_diff+1)))
+            logger.info(f"Retrieved {len(messages)} messages.")
             for message in messages:
                 yield message
                 current += 1
+                logger.info(f"Yielding message with ID: {message.id}")
 
-
-        
-Bot().run()
-
-
-
-
-
+Bot().run()