Exemplo n.º 1
0
    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)
Exemplo n.º 3
0
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):