Tea

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} литров. ☕️')