diff --git a/tea.py b/tea.py index 3bd4715..5ad1c2c 100644 --- a/tea.py +++ b/tea.py @@ -37,18 +37,23 @@ def tea_message(message): user_id = str(message.from_user.id) user_name = message.from_user.username + if not user_id in USERS_BALANCE: + USERS_BALANCE[user_id] = [0.0, ''] + # проверяем, можно ли выпить чай - if user_id in USERS_BALANCE: - last_teapot_time = datetime.strptime(USERS_BALANCE[user_id][1], '%Y-%m-%d %H:%M:%S') - print(last_teapot_time) + try: + last_teapot_time = datetime.strptime(USERS_BALANCE[user_id][1], '%Y-%m-%d %H:%M:%S') time_delta = datetime.now() - last_teapot_time if time_delta.seconds < 1800: bot.reply_to(message, f'Ты уже пил чай {time_delta.seconds//30} минут назад. \n' f'Чай будет готов через {(30 - time_delta.seconds//30)} минут. ☕️') return + except: + pass # обновляем баланс пользователя - balance = USERS_BALANCE[str(user_id)] + balance = USERS_BALANCE[user_id] + print(balance) balance[0] += TEA_PRICE balance[1] = datetime.now().strftime('%Y-%m-%d %H:%M:%S') USERS_BALANCE[user_id] = balance @@ -60,10 +65,10 @@ def tea_message(message): @bot.message_handler(commands=['bal']) def balance_message(message): - user_id = message.from_user.id + user_id = str(message.from_user.id) # проверяем баланс пользователя - user_balance = USERS_BALANCE.get(user_id, [0, ''])[0] + user_balance = USERS_BALANCE[user_id][0] bot.reply_to(message, f'Твой баланс: {user_balance} литров. ☕️')