53 lines
1.3 KiB
Python
53 lines
1.3 KiB
Python
import logging
|
|
import configparser
|
|
import aria2p
|
|
from telegram.ext import Updater
|
|
|
|
logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
|
|
level=logging.INFO)
|
|
|
|
config = configparser.ConfigParser()
|
|
config.read('bot/config.ini')
|
|
|
|
|
|
def getConfig(name: str):
|
|
return config['DEFAULT'][name]
|
|
|
|
|
|
LOGGER = logging.getLogger(__name__)
|
|
|
|
try:
|
|
if bool(config['DEFAULT']['_____REMOVE_THIS_LINE_____']):
|
|
logging.ERROR('The README.md file there to be read! Exiting now!')
|
|
exit()
|
|
except KeyError:
|
|
pass
|
|
|
|
aria2 = aria2p.API(
|
|
aria2p.Client(
|
|
host="http://localhost",
|
|
port=6800,
|
|
secret="",
|
|
)
|
|
)
|
|
|
|
DOWNLOAD_DIR = None
|
|
BOT_TOKEN = None
|
|
|
|
status_reply_dict = {}
|
|
download_dict = {}
|
|
try:
|
|
BOT_TOKEN = getConfig('BOT_TOKEN')
|
|
parent_id = getConfig('GDRIVE_FOLDER_ID')
|
|
DOWNLOAD_DIR = getConfig('DOWNLOAD_DIR')
|
|
if DOWNLOAD_DIR[-1] != '/' or DOWNLOAD_DIR[-1] != '\\':
|
|
DOWNLOAD_DIR = DOWNLOAD_DIR + '/'
|
|
DOWNLOAD_STATUS_UPDATE_INTERVAL = int(getConfig('DOWNLOAD_STATUS_UPDATE_INTERVAL'))
|
|
|
|
except KeyError as e:
|
|
LOGGER.error("One or more env variables missing! Exiting now")
|
|
exit(1)
|
|
|
|
updater = Updater(token=BOT_TOKEN, use_context=True)
|
|
dispatcher = updater.dispatcher
|