| 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 |
|
|
| |
| logging.config.fileConfig("logging.conf") |
| logger = logging.getLogger(__name__) |
|
|
| 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.") |
|
|
| 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.") |
|
|
| 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 |
| 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) |
|
|
| 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() |
| 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}") |
|
|
| Bot().run() |
