diff --git a/main.py b/main.py index 05b71b0..1e792a7 100644 --- a/main.py +++ b/main.py @@ -52,18 +52,6 @@ def telegram_bot(): if LOGS: bot.send_message(IMPORTANT_LOGS, BotDB.get_error_message_from_db(7)) - markup = types.ReplyKeyboardMarkup(resize_keyboard=True, one_time_keyboard=True) - item1 = types.KeyboardButton("📢Предложить свой пост") - item2 = types.KeyboardButton("📩Связаться с админами") - item3 = types.KeyboardButton("❌Удалить пост") - item4 = types.KeyboardButton("🎧Войти в режим стендапа") - is_need_sticker = BotDB.get_info_about_stickers(user_id=message.from_user.id) - if is_need_sticker == 0: - item5 = types.KeyboardButton("🤪Хочу ваши стикеры") - BotDB.update_info_about_stickers(user_id=message.from_user.id) - markup.add(item1, item2, item3, item4, item5) - else: - markup.add(item1, item2, item3, item4) try: user_id = message.from_user.id first_name = message.from_user.first_name @@ -74,10 +62,29 @@ def telegram_bot(): time_UTC = int(time.time()) date_added = datetime.fromtimestamp(time_UTC) date_changed = datetime.fromtimestamp(time_UTC) + + markup = types.ReplyKeyboardMarkup(resize_keyboard=True, one_time_keyboard=True) + item1 = types.KeyboardButton("📢Предложить свой пост") + item2 = types.KeyboardButton("📩Связаться с админами") + item3 = types.KeyboardButton("❌Удалить пост") + item4 = types.KeyboardButton("🎧Войти в режим стендапа") if BotDB.user_exists(user_id): - pass + is_need_sticker = BotDB.get_info_about_stickers(user_id=message.from_user.id) + if is_need_sticker == 0: + item5 = types.KeyboardButton("🤪Хочу ваши стикеры") + BotDB.update_info_about_stickers(user_id=message.from_user.id) + markup.add(item1, item2, item3, item4, item5) + else: + markup.add(item1, item2, item3, item4) else: BotDB.add_new_user_in_db(user_id, first_name, full_name, username, is_bot, language_code, date_added, date_changed) + is_need_sticker = BotDB.get_info_about_stickers(user_id=message.from_user.id) + if is_need_sticker == 0: + item5 = types.KeyboardButton("🤪Хочу ваши стикеры") + BotDB.update_info_about_stickers(user_id=message.from_user.id) + markup.add(item1, item2, item3, item4, item5) + else: + markup.add(item1, item2, item3, item4) hello_message = BotDB.get_message_from_db('start_message', first_name) bot.send_message(message.chat.id, hello_message, parse_mode='html', reply_markup=markup, disable_web_page_preview=not PREVIEW_LINK) except: