diff --git a/hlna_bot.py b/hlna_bot.py index bdfa757..a9380b9 100755 --- a/hlna_bot.py +++ b/hlna_bot.py @@ -9,36 +9,35 @@ import colorama import hlna list_config = hlna.list_config -k_api = "" + def read_config(): - with open(f"{hlna.dir_config} + hlna_bot", "r") as yamlfile: - data = yaml.load(yamlfile, Loader=yaml.FullLoader) - return data[0] # возвращаем словарь со всеми значениями - - -def create_config(): + if not f"{hlna.dir_config}hlna_bot": + key_api = input("Укажите ключ бота\n") + create_config(key_api) + else: + with open(f"{hlna.dir_config}hlna_bot", "r") as yamlfile: + data = yaml.load(yamlfile, Loader=yaml.FullLoader) + return data[0] +def create_config(key_api): k_api = key_api config_hlna_bot = [ { 'key_api': k_api, } ] - with open(f"hlna.config_hlna", 'w') as yamlfile: - yaml.dump(f"{config_hlna_bot} + hlna_bot", yamlfile) + with open(f"{hlna.dir_config}hlna_bot", 'w') as yamlfile: + yaml.dump(config_hlna_bot, yamlfile) print(colorama.Fore.GREEN + "Конфиг создан" + colorama.Style.RESET_ALL) return k_api -read_config() -if k_api == "": - key_api = input("Укажите ключ бота\n") - create_config() +data = read_config() - -read_config() +k_api = data['key_api'] hlna_bot = telebot.TeleBot(f"{k_api}", parse_mode=None) + @hlna_bot.message_handler(commands=["start"]) def m_start(m): menu = types.ReplyKeyboardMarkup(resize_keyboard=True)