def __init__(self): # get the sensitive information from config file login="" password="" dbaseName="" dbaseHost="" for line in open(".config","r"): if not line.rfind("DBUser") == -1: login = line.split(" ")[1].split("\n")[0] elif not line.rfind("DBPass") == -1: password = line.split(" ")[1].split("\n")[0] elif not line.rfind("DBHost") == -1: dbaseHost = line.split(" ")[1].split("\n")[0] elif not line.rfind("DBName") == -1: dbaseName = line.split(" ")[1].split("\n")[0] self.sql = SQLHandler(dbaseName,login,password,dbaseHost)
login = "" password = "" dbaseName = "" dbaseHost = "" for line in open("../.config", "r"): if not line.rfind("DBUser") == -1: login = line.split(" ")[1].split("\n")[0] elif not line.rfind("DBPass") == -1: password = line.split(" ")[1].split("\n")[0] elif not line.rfind("DBHost") == -1: dbaseHost = line.split(" ")[1].split("\n")[0] elif not line.rfind("DBName") == -1: dbaseName = line.split(" ")[1].split("\n")[0] sql = SQLHandler(dbaseName, login, password, dbaseHost) ############### ### METHODS ### ############### def rmSRMdir(dir): global srmHost if os.path.exists(dir): log.output(" ---> Going into directory " + dir) tmp = os.listdir(dir)
from aiogram.dispatcher import FSMContext from aiogram import Bot, Dispatcher, executor, types from aiogram.contrib.fsm_storage.memory import MemoryStorage from stateHandler import STATES from sqlHandler import SQLHandler from configHandler import ConfigHandler from sqlHandler.consts import TYPES, CONSTS, STRINGS from errorHandling import errorHandler logging.basicConfig(level=logging.INFO) bot = Bot(token=ConfigHandler.token) dispatcher = Dispatcher(bot, storage=MemoryStorage()) database = SQLHandler() def api_count_check() -> list: country_codes = TYPES.params_list result = [] for param in country_codes: response = requests.get( ConfigHandler.count_link, params=param ).json() result.append(response.get('count')) return result @dispatcher.message_handler(commands=['start']) async def start(message: types.Message):