|
|
|
import asyncio
|
|
import os
|
|
from logging import getLogger
|
|
from typing import Union
|
|
|
|
from pymongo import MongoClient
|
|
from pyrogram.errors import (
|
|
FloodWait,
|
|
InputUserDeactivated,
|
|
PeerIdInvalid,
|
|
UserIsBlocked,
|
|
)
|
|
from pyrogram.types import Message
|
|
|
|
from Mikobot import DB_NAME, MONGO_DB_URI
|
|
|
|
client = MongoClient(MONGO_DB_URI)
|
|
dbname = client[DB_NAME]
|
|
|
|
LOGGER = getLogger(__name__)
|
|
BANNED = {}
|
|
|
|
|
|
|
|
|
|
|
|
class temp(object):
|
|
BANNED_USERS = []
|
|
BANNED_CHATS = []
|
|
ME = None
|
|
CURRENT = int(os.environ.get("SKIP", 2))
|
|
CANCEL = False
|
|
MELCOW = {}
|
|
U_NAME = None
|
|
B_NAME = None
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def broadcast_messages(user_id, message):
|
|
try:
|
|
message.copy(chat_id=user_id)
|
|
return True, "Succes"
|
|
except FloodWait as e:
|
|
asyncio.sleep(e.x)
|
|
return broadcast_messages(user_id, message)
|
|
except InputUserDeactivated:
|
|
dbname.delete_user(int(user_id))
|
|
LOGGER.info(f"{user_id}-Removed from Database, since deleted account.")
|
|
return False, "Deleted"
|
|
except UserIsBlocked:
|
|
LOGGER.info(f"{user_id} -Blocked the bot.")
|
|
return False, "Blocked"
|
|
except PeerIdInvalid:
|
|
dbname.delete_user(int(user_id))
|
|
LOGGER.info(f"{user_id} - PeerIdInvalid")
|
|
return False, "Error"
|
|
except Exception:
|
|
return False, "Error"
|
|
|
|
|
|
def get_size(size):
|
|
"""Get size in readable format"""
|
|
|
|
units = ["Bytes", "KB", "MB", "GB", "TB", "PB", "EB"]
|
|
size = float(size)
|
|
i = 0
|
|
while size >= 1024.0 and i < len(units):
|
|
i += 1
|
|
size /= 1024.0
|
|
return "%.2f %s" % (size, units[i])
|
|
|
|
|
|
def get_file_id(msg: Message):
|
|
if msg.media:
|
|
for message_type in (
|
|
"photo",
|
|
"animation",
|
|
"audio",
|
|
"document",
|
|
"video",
|
|
"video_note",
|
|
"voice",
|
|
"sticker",
|
|
):
|
|
if obj := getattr(msg, message_type):
|
|
setattr(obj, "message_type", message_type)
|
|
return obj
|
|
|
|
|
|
def extract_user(message: Message) -> Union[int, str]:
|
|
"""extracts the user from a message"""
|
|
|
|
user_id = None
|
|
user_first_name = None
|
|
if message.reply_to_message:
|
|
user_id = message.reply_to_message.from_user.id
|
|
user_first_name = message.reply_to_message.from_user.first_name
|
|
|
|
elif len(message.command) > 1:
|
|
if len(message.entities) > 1 and message.entities[1].type == "text_mention":
|
|
required_entity = message.entities[1]
|
|
user_id = required_entity.user.id
|
|
user_first_name = required_entity.user.first_name
|
|
else:
|
|
user_id = message.command[1]
|
|
|
|
user_first_name = user_id
|
|
try:
|
|
user_id = int(user_id)
|
|
except ValueError:
|
|
pass
|
|
else:
|
|
user_id = message.from_user.id
|
|
user_first_name = message.from_user.first_name
|
|
return (user_id, user_first_name)
|
|
|
|
|
|
|
|
|
