Update bot.py

bot.py

@@ -1,93 +1,36 @@
-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 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
-import uvicorn
-from fastapi_app import app as fastapi_app
+# Dockerfile
+FROM python:3.10
 
-# Get logging configurations
-logging.config.fileConfig("logging.conf")
-logger = logging.getLogger(__name__)
+# Create a non-root user with valid username format
+RUN adduser --disabled-password --gecos '' professor && \
+    mkdir -p /app && \
+    chown -R professor:professor /app
 
-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.")
+# Install system dependencies
+RUN apt-get update && apt-get upgrade -y
 
-    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 requirements and install
+COPY requirements.txt /requirements.txt
+RUN pip install --upgrade pip && \
+    pip install -r /requirements.txt
 
-        await super().start()
-        logger.info("Pyrogram client started.")
+# Switch to non-root user
+USER professor
+WORKDIR /app
 
-        await Media.ensure_indexes()
-        logger.info("Indexes ensured for Media collection.")
+# Copy application files
+COPY --chown=professor:professor . .
 
-        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)
+# Set environment variables
+ENV LOG_PATH=/app/BotLog.txt
+ENV UVICORN_HOST=0.0.0.0
+ENV UVICORN_PORT=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}")
+# Ensure the log file is writable
+RUN touch $LOG_PATH && chmod 666 $LOG_PATH
 
-        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......🕸️")
+# Expose port for FastAPI
+EXPOSE 8000
 
-        # Run FastAPI app in a separate task
-        asyncio.create_task(uvicorn.run(fastapi_app, host="0.0.0.0", port=8000, log_level="info"))
-
-    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()
-    asyncio.run(bot.start())
+# Start the FastAPI app
+CMD ["uvicorn", "fastapi_app:app", "--host", "$UVICORN_HOST", "--port", "$UVICORN_PORT", "--log-level", "info"]