database

main.py

@@ -2,6 +2,11 @@ from termcolor import cprint
 from bot import bot
 import locale
 from changelocale import change_locale
+from database.models import User
+
+
+def to_fixed(num_obj, digits=0):
+    return f"{num_obj:.{digits}f}"
 
 
 def main_bot(messages):
@@ -12,9 +17,39 @@ def main_bot(messages):
                 mes = locale.ru['help_message']
             elif locale.LOCALE == 'en':
                 mes = locale.en['help_message']
+            _, user = User.get_or_create(username=message.chat.id)
             bot.send_message(message.chat.id, mes)
         elif message.text.split(' ')[0] == '/ru' or message.text.split(' ')[0] == '/en':
             change_locale(message)
+        elif message.text.split(' ')[0] == '/balance':
+            user = User.get(username=message.chat.id)
+            bot.send_message(message.chat.id, to_fixed(user.balance, 2))
+        elif message.text.split(' ')[0] == '/add' or message.text.split(' ')[0] == '+':
+            if len(message.text.split(' ')) < 2:
+                mes = None
+                if locale.LOCALE == 'ru':
+                    mes = 'Используйте /add [сумма] или + [сумма]'
+                elif locale.LOCALE == 'en':
+                    mes = 'Use /add [sum] or + [sum]'
+                bot.send_message(message.chat.id, mes)
+            else:
+                user = User.get(username=message.chat.id)
+                user.balance += float(message.text.split(' ')[1])
+                user.save()
+                bot.send_message(message.chat.id, to_fixed(user.balance, 2))
+        elif message.text.split(' ')[0] == '/rm' or message.text.split(' ')[0] == '-':
+            if len(message.text.split(' ')) < 2:
+                mes = None
+                if locale.LOCALE == 'ru':
+                    mes = 'Используйте /rm [сумма] или - [сумма]'
+                elif locale.LOCALE == 'en':
+                    mes = 'Use /rm [sum] or - [sum]'
+                bot.send_message(message.chat.id, mes)
+            else:
+                user = User.get(username=message.chat.id)
+                user.balance -= float(message.text.split(' ')[1])
+                user.save()
+                bot.send_message(message.chat.id, to_fixed(user.balance, 2))
         else:
             mes = None
             if locale.LOCALE == 'ru':