from dataclasses import dataclass
from typing import List, Union
from pyrogram.types import (
    BotCommandScopeAllChatAdministrators,
    BotCommandScopeAllGroupChats,
    BotCommandScopeAllPrivateChats,
    BotCommandScopeChat,
    BotCommandScopeChatAdministrators,
    BotCommandScopeChatMember,
    BotCommandScopeDefault,
    BotCommand,
)


@dataclass
class CommandSet:
    """Command stored in PyroClient's 'commands' attribute"""

    commands: List[BotCommand]
    scope: Union[
        BotCommandScopeDefault,
        BotCommandScopeAllPrivateChats,
        BotCommandScopeAllGroupChats,
        BotCommandScopeAllChatAdministrators,
        BotCommandScopeChat,
        BotCommandScopeChatAdministrators,
        BotCommandScopeChatMember,
    ] = BotCommandScopeDefault()
    language_code: str = ""