diff --git a/template/{{ part_dir }}/src/{% if 'kurigram_userbot' in backend_services %}userbot{% endif %}/__init__.py.jinja b/template/{{ part_dir }}/src/{% if 'kurigram_userbot' in backend_services %}userbot{% endif %}/__init__.py.jinja index e74ab29..e9daadd 100644 --- a/template/{{ part_dir }}/src/{% if 'kurigram_userbot' in backend_services %}userbot{% endif %}/__init__.py.jinja +++ b/template/{{ part_dir }}/src/{% if 'kurigram_userbot' in backend_services %}userbot{% endif %}/__init__.py.jinja @@ -1,3 +1,3 @@ -from .modules.client import PyroClient +from .modules.client import DEVICE_MODEL, PyroClient -__all__ = ["PyroClient"] +__all__ = ["DEVICE_MODEL", "PyroClient"] diff --git a/template/{{ part_dir }}/src/{% if 'kurigram_userbot' in backend_services %}userbot{% endif %}/modules/client.py.jinja b/template/{{ part_dir }}/src/{% if 'kurigram_userbot' in backend_services %}userbot{% endif %}/modules/client.py.jinja index f392709..e612058 100644 --- a/template/{{ part_dir }}/src/{% if 'kurigram_userbot' in backend_services %}userbot{% endif %}/modules/client.py.jinja +++ b/template/{{ part_dir }}/src/{% if 'kurigram_userbot' in backend_services %}userbot{% endif %}/modules/client.py.jinja @@ -1,20 +1,30 @@ from pyrogram import Client, enums +DEVICE_MODEL = "{{ project_name[:32] }}" + class PyroClient(Client): def __init__( - self, name: str, *, workdir: str = "sessions", load_handlers: bool = True + self, + name: str, + *, + workdir: str = "sessions", + device_model: str | None = None, + load_handlers: bool = True, ) -> None: super().__init__( name, workdir=workdir, api_id=2040, api_hash="b18441a1ff607e10a989891a5462e627", - device_model="Desktop", + device_model=device_model or DEVICE_MODEL, system_version="Windows 11 x64", - app_version="6.2.4 x64", + app_version="7.0.8 x64", lang_pack="tdesktop", + lang_code="en", + system_lang_code="en-US", client_platform=enums.ClientPlatform.DESKTOP, + max_concurrent_transmissions=4, ) if load_handlers: @@ -24,4 +34,4 @@ class PyroClient(Client): self.add_handler(*handler) -__all__ = ["PyroClient"] +__all__ = ["DEVICE_MODEL", "PyroClient"]