Update bot.py

bot.py

@@ -1,36 +1,96 @@
-# Dockerfile
-FROM python:3.10
+# bot.py
 
-# Create a non-root user with valid username format
-RUN adduser --disabled-password --gecos '' professor && \
-    mkdir -p /app && \
-    chown -R professor:professor /app
+import os, math, logging, datetime, pytz, logging.config
+from pyrogram import Client, types
+from database.users_chats_db import db
+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
+import asyncio
 
-# Install system dependencies
-RUN apt-get update && apt-get upgrade -y
+# Get logging configurations
+logging.config.fileConfig("logging.conf")
+logger = logging.getLogger(__name__)
 
-# Copy requirements and install
-COPY requirements.txt /requirements.txt
-RUN pip install --upgrade pip && \
-    pip install -r /requirements.txt
+class Bot(Client):
+    def __init__(self):
+        super().__init__(
+            name="Professor-Bot",
+            api_id=API_ID,
+            api_hash=API_HASH,
+            bot_token=BOT_TOKEN,
+            plugins=dict(root="plugins")
+        )
+        logger.info("Bot initialized.")
 
-# Switch to non-root user
-USER professor
-WORKDIR /app
+    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
+        logger.info("Banned users and chats loaded.")
 
-# Copy application files
-COPY --chown=professor:professor . .
+        await super().start()
+        logger.info("Pyrogram client started.")
 
-# Set environment variables
-ENV LOG_PATH=/app/BotLog.txt
-ENV UVICORN_HOST=0.0.0.0
-ENV UVICORN_PORT=8000
+        await Media.ensure_indexes()
+        logger.info("Indexes ensured for Media collection.")
 
-# Ensure the log file is writable
-RUN touch $LOG_PATH && chmod 666 $LOG_PATH
+        me = await self.get_me()
+        temp.U_NAME = me.username
+        temp.B_NAME = me.first_name
+        self.id = me.id
+        self.name = me.first_name
+        self.mention = me.mention
+        self.username = me.username
+        self.log_channel = LOG_CHANNEL
+        self.uptime = UPTIME
+        curr = datetime.datetime.now(pytz.timezone("Asia/Kolkata"))
+        date = curr.strftime('%d %B, %Y')
+        tame = curr.strftime('%I:%M:%S %p')
+        log_message = LOG_MSG.format(me.first_name, date, tame, __repo__, __version__, __license__, __copyright__)
+        logger.info(log_message)
 
-# Expose port for FastAPI
-EXPOSE 8000
+        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}")
 
-# Start the FastAPI app
-CMD ["uvicorn", "fastapi_app:app", "--host", "$UVICORN_HOST", "--port", "$UVICORN_PORT", "--log-level", "info"]
+        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()
+            logger.info("Web Response Is Running......🕸️")
+
+    async def stop(self, *args):
+        logger.info("Stopping bot...")
+        await super().stop()
+        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]]:
+        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}")
+
+if __name__ == "__main__":
+    bot = Bot()
+    loop = asyncio.get_event_loop()
+    try:
+        loop.run_until_complete(bot.start())
+        loop.run_forever()
+    except KeyboardInterrupt:
+        loop.run_until_complete(bot.stop())
+    finally:
+        loop.close()