diff --git a/bot/__main__.py b/bot/__main__.py index da32186..65f9001 100644 --- a/bot/__main__.py +++ b/bot/__main__.py @@ -1,7 +1,22 @@ +from importlib import import_module +from pathlib import Path from bot import TelegramBot, logger +from bot.config import Telegram from bot.server import server +def load_plugins(): + count = 0 + for path in Path('bot/plugins').rglob('*.py'): + import_module(f'bot.plugins.{path.stem}') + count += 1 + logger.info(f'Loaded {count} {"plugins" if count > 1 else "plugin"}.') + if __name__ == '__main__': - logger.info('Initializing...') + logger.info('Intializing...') TelegramBot.loop.create_task(server.serve()) - TelegramBot.run() + TelegramBot.start(bot_token=Telegram.BOT_TOKEN) + logger.info('Telegram client is now started.') + logger.info('Loading bot plugins...') + load_plugins() + logger.info('Bot is now ready!') + TelegramBot.run_until_disconnected()