gcast.py

import asyncio

from pyrogram import Client, filters
from pyrogram.errors import FloodWait
from pyrogram.types import Message

from utils.misc import modules_help, prefix


@Client.on_message(filters.command("gcast", prefix) & filters.me)
async def gcast(client: Client, message: Message):
    if message.reply_to_message:
        msg = message.reply_to_message.text
    elif len(message.command) > 1:
        msg = " ".join(message.command[1:])
    else:
        await message.edit("Provide text or reply to a message to broadcast globally.")
        return

    await message.edit("Starting global broadcast...")
    done, errors = 0, 0

    async for dialog in client.get_dialogs():
        if dialog.chat.type.value in ["supergroup", "group"]:
            try:
                await client.send_message(dialog.chat.id, msg)
                done += 1
            except FloodWait as e:
                await asyncio.sleep(e.value + 10)
            except Exception as e:
                errors += 1
                print(f"Error: {e}")

    await message.edit(f"Broadcast completed: {done} successful, {errors} failed.")


@Client.on_message(filters.command("gucast", prefix) & filters.me)
async def gucast(client: Client, message: Message):
    if message.reply_to_message:
        msg = message.reply_to_message.text
    elif len(message.command) > 1:
        msg = " ".join(message.command[1:])
    else:
        await message.edit("Provide text or reply to a message to broadcast globally to users.")
        return

    await message.edit("Starting global user broadcast...")
    done, errors = 0, 0

    async for dialog in client.get_dialogs():
        if dialog.chat.type.value == "private" and not dialog.chat.is_bot:
            try:
                await client.send_message(dialog.chat.id, msg)
                done += 1
            except Exception as e:
                errors += 1
                print(f"Error: {e}")

    await message.edit(f"Broadcast completed: {done} successful, {errors} failed.")


modules_help["gcast"] = {
    "gcast [text/reply to text]*": "Broadcast message to all groups.",
    "gucast [text/reply to text]*": "Broadcast message to all private users."
}