profitroll/LibBotUniversal
profitroll/LibBotUniversal
1
0
Fork 0
Code Issues 3 Pull Requests Actions Packages Releases 30 Activity

owner, scoped_commands and i18n_bot_info can be provided in Client's kwargs now

Browse Source
This commit is contained in:
Profitroll 2024-01-03 23:16:26 +01:00
parent 327b161b41
commit fc14cad3ff
Signed by: profitroll
GPG Key ID: FA35CAB49DACD3B2
1 changed files with 53 additions and 42 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/libbot/pyrogram/classes/client.py
View File

@ -46,6 +46,7 @@ class PyroClient(Client):
def __init__( def __init__(
self, self,
name: str = "bot_client", name: str = "bot_client",
owner: Union[int, None] = None,
config: Union[Dict[str, Any], None] = None, config: Union[Dict[str, Any], None] = None,
config_path: Union[str, Path] = Path("config.json"), config_path: Union[str, Path] = Path("config.json"),
api_id: Union[int, None] = None, api_id: Union[int, None] = None,
@ -58,6 +59,8 @@ class PyroClient(Client):
sleep_threshold: int = 120, sleep_threshold: int = 120,
max_concurrent_transmissions: int = 1, max_concurrent_transmissions: int = 1,
commands_source: Union[Dict[str, dict], None] = None, commands_source: Union[Dict[str, dict], None] = None,
scoped_commands: Union[bool, None] = None,
i18n_bot_info: bool = False,
scheduler: Union[AsyncIOScheduler, BackgroundScheduler, None] = None, scheduler: Union[AsyncIOScheduler, BackgroundScheduler, None] = None,
**kwargs, **kwargs,
): ):
@ -93,12 +96,16 @@ class PyroClient(Client):
else max_concurrent_transmissions, else max_concurrent_transmissions,
**kwargs, **kwargs,
) )
self.owner: int = self.config["bot"]["owner"] self.owner: int = self.config["bot"]["owner"] if owner is None else owner
self.commands: List[PyroCommand] = [] self.commands: List[PyroCommand] = []
self.commands_source: Dict[str, dict] = ( self.commands_source: Dict[str, dict] = (
self.config["commands"] if commands_source is None else commands_source self.config["commands"] if commands_source is None else commands_source
) )
self.scoped_commands: bool = self.config["bot"]["scoped_commands"] self.scoped_commands: bool = (
self.config["bot"]["scoped_commands"]
if scoped_commands is None
else scoped_commands
)
self.start_time: float = 0 self.start_time: float = 0
self.bot_locale: BotLocale = BotLocale( self.bot_locale: BotLocale = BotLocale(
@ -116,6 +123,8 @@ class PyroClient(Client):
self.scopes_placeholders: Dict[str, int] = {"owner": self.owner} self.scopes_placeholders: Dict[str, int] = {"owner": self.owner}
self.i18n_bot_info: bool = i18n_bot_info
async def start(self, register_commands: bool = True) -> None: async def start(self, register_commands: bool = True) -> None:
await super().start() await super().start()
@ -129,6 +138,7 @@ class PyroClient(Client):
getpid(), getpid(),
) )
if self.i18n_bot_info:
# Register default bot's info # Register default bot's info
try: try:
await self.set_bot_info( await self.set_bot_info(
@ -153,7 +163,8 @@ class PyroClient(Client):
if "metadata" not in locale or ("codes" not in locale["metadata"]): if "metadata" not in locale or ("codes" not in locale["metadata"]):
logger.warning( logger.warning(
"Locale %s is missing metadata or metadata.codes key", locale_code "Locale %s is missing metadata or metadata.codes key",
locale_code,
) )
continue continue