mirror of
https://github.com/yasirarism/MissKatyPyro.git
synced 2025-12-29 09:44:50 +00:00
89b2b9a2d6
This commit fixes the style issues introduced in f9f107e according to the output
from isort, Ruff Formatter and Yapf.
Details: None
50 lines
1.7 KiB
Python
50 lines
1.7 KiB
Python
import asyncio
|
|
import datetime
|
|
import time
|
|
|
|
from async_pymongo import AsyncClient
|
|
from pyrogram import filters
|
|
from pyrogram.types import Message
|
|
|
|
from misskaty import DATABASE_URI, app
|
|
from misskaty.vars import SUDO
|
|
from utils import broadcast_messages
|
|
|
|
|
|
@app.on_message(filters.command("broadcast") & filters.user(SUDO) & filters.reply)
|
|
async def broadcast(_, ctx: Message):
|
|
mongo = AsyncClient(DATABASE_URI)
|
|
userdb = mongo["MissKatyBot"]["peers"]
|
|
b_msg = ctx.reply_to_message
|
|
sts = await ctx.reply_msg("Broadcasting your messages...")
|
|
start_time = time.time()
|
|
total_users = await userdb.count_documents({})
|
|
done = 0
|
|
blocked = 0
|
|
deleted = 0
|
|
failed = 0
|
|
|
|
success = 0
|
|
async for chat in userdb.find({"type": "user"}):
|
|
if chat["type"] != "user":
|
|
continue
|
|
pti, sh = await broadcast_messages(int(chat["_id"]), b_msg)
|
|
if pti:
|
|
success += 1
|
|
elif pti is False:
|
|
if sh == "Blocked":
|
|
blocked += 1
|
|
elif sh == "Deleted":
|
|
deleted += 1
|
|
elif sh == "Error":
|
|
failed += 1
|
|
done += 1
|
|
await asyncio.sleep(2)
|
|
if not done % 20:
|
|
await sts.edit_msg(
|
|
f"Broadcast in progress:\n\nTotal Users {total_users}\nCompleted: {done} / {total_users}\nSuccess: {success}\nBlocked: {blocked}\nDeleted: {deleted}"
|
|
)
|
|
time_taken = datetime.timedelta(seconds=int(time.time() - start_time))
|
|
await sts.edit_msg(
|
|
f"Broadcast Completed:\nCompleted in {time_taken} seconds.\n\nTotal Users {total_users}\nCompleted: {done} / {total_users}\nSuccess: {success}\nBlocked: {blocked}\nDeleted: {deleted}"
|
|
)
|