mirror of
https://github.com/LonamiWebs/Telethon.git
synced 2024-11-25 02:43:45 +03:00
2c9d43d600
The assistant example will now simply be the "core" that initializes the rest of plugins, allowing for more updates to the *bot* without cluttering Telethon's git history.
71 lines
1.7 KiB
Python
71 lines
1.7 KiB
Python
"""
|
|
This file is only the "core" of the bot. It is responsible for loading the
|
|
plugins module and initializing it. You may obtain the plugins by running:
|
|
|
|
git clone https://github.com/Lonami/TelethonianBotExt plugins
|
|
|
|
In the same folder where this file lives. As a result, the directory should
|
|
look like the following:
|
|
|
|
assistant.py
|
|
plugins/
|
|
...
|
|
"""
|
|
import asyncio
|
|
import os
|
|
import sys
|
|
import time
|
|
|
|
from telethon import TelegramClient
|
|
|
|
try:
|
|
# Standalone script assistant.py with folder plugins/
|
|
import plugins
|
|
except ImportError:
|
|
try:
|
|
# Running as a module with `python -m assistant` and structure:
|
|
#
|
|
# assistant/
|
|
# __main__.py (this file)
|
|
# plugins/ (cloned)
|
|
from . import plugins
|
|
except ImportError:
|
|
print('could not load the plugins module, does the directory exist '
|
|
'in the correct location?', file=sys.stderr)
|
|
|
|
exit(1)
|
|
|
|
|
|
def get_env(name, message, cast=str):
|
|
if name in os.environ:
|
|
return os.environ[name]
|
|
while True:
|
|
value = input(message)
|
|
try:
|
|
return cast(value)
|
|
except ValueError as e:
|
|
print(e, file=sys.stderr)
|
|
time.sleep(1)
|
|
|
|
|
|
API_ID = get_env('TG_API_ID', 'Enter your API ID: ', int)
|
|
API_HASH = get_env('TG_API_HASH', 'Enter your API hash: ')
|
|
TOKEN = get_env('TG_TOKEN', 'Enter the bot token: ')
|
|
NAME = TOKEN.split(':')[0]
|
|
|
|
|
|
async def main():
|
|
bot = TelegramClient(NAME, API_ID, API_HASH)
|
|
|
|
await bot.start(bot_token=TOKEN)
|
|
|
|
try:
|
|
await plugins.init(bot)
|
|
await bot.run_until_disconnected()
|
|
finally:
|
|
await bot.disconnect()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
asyncio.run(main())
|