forked from kate/uniborg
Handle bot tokens
This commit is contained in:
parent
881266f96e
commit
3cdb595b55
|
@ -14,7 +14,7 @@ import telethon.events
|
||||||
from . import hacks
|
from . import hacks
|
||||||
|
|
||||||
class Uniborg(TelegramClient):
|
class Uniborg(TelegramClient):
|
||||||
def __init__(self, session, *, plugin_path="plugins", **kwargs):
|
def __init__(self, session, *, plugin_path="plugins", bot_token=None, **kwargs):
|
||||||
# TODO: handle non-string session
|
# TODO: handle non-string session
|
||||||
self._name = session
|
self._name = session
|
||||||
self._logger = logging.getLogger(session)
|
self._logger = logging.getLogger(session)
|
||||||
|
@ -30,7 +30,7 @@ class Uniborg(TelegramClient):
|
||||||
# precedence
|
# precedence
|
||||||
self._event_builders = hacks.ReverseList()
|
self._event_builders = hacks.ReverseList()
|
||||||
|
|
||||||
self._loop.run_until_complete(self._async_init())
|
self._loop.run_until_complete(self._async_init(bot_token=bot_token))
|
||||||
|
|
||||||
core_plugin = Path(__file__).parent / "_core.py"
|
core_plugin = Path(__file__).parent / "_core.py"
|
||||||
self.load_plugin_from_file(core_plugin)
|
self.load_plugin_from_file(core_plugin)
|
||||||
|
@ -38,8 +38,8 @@ class Uniborg(TelegramClient):
|
||||||
for p in Path().glob(f"{self._plugin_path}/*.py"):
|
for p in Path().glob(f"{self._plugin_path}/*.py"):
|
||||||
self.load_plugin_from_file(p)
|
self.load_plugin_from_file(p)
|
||||||
|
|
||||||
async def _async_init(self):
|
async def _async_init(self, **kwargs):
|
||||||
await self.start()
|
await self.start(**kwargs)
|
||||||
|
|
||||||
self.me = await self.get_me()
|
self.me = await self.get_me()
|
||||||
self.uid = telethon.utils.get_peer_id(self.me)
|
self.uid = telethon.utils.get_peer_id(self.me)
|
||||||
|
|
Loading…
Reference in New Issue