diff --git a/bot/__init__.py b/bot/__init__.py index bffe857..38fc273 100644 --- a/bot/__init__.py +++ b/bot/__init__.py @@ -3,7 +3,6 @@ import configparser import aria2p from telegram.ext import Updater - logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', level=logging.INFO) @@ -31,8 +30,7 @@ aria2 = aria2p.API( secret="", ) ) -aria2.autopurge() -aria2.remove_all(True) + DOWNLOAD_DIR = None BOT_TOKEN = None try: diff --git a/bot/__main__.py b/bot/__main__.py index dc87c3e..5ae23d9 100644 --- a/bot/__main__.py +++ b/bot/__main__.py @@ -1,6 +1,8 @@ from telegram.ext import CommandHandler, run_async -from bot import dispatcher, LOGGER, updater +from bot import dispatcher, LOGGER, updater, aria2 import bot.mirror +from bot.helper import fs_utils +import signal @run_async @@ -13,10 +15,12 @@ LOGGER.info('__main__.py') def main(): + start_handler = CommandHandler('start', start) dispatcher.add_handler(start_handler) LOGGER.info("Bot Started!") updater.start_polling() + signal.signal(signal.SIGINT, fs_utils.exit_clean_up) main()