From b7863746974b0f133fcdbec627bcc281ad0089c6 Mon Sep 17 00:00:00 2001
From: TolaMironcenko <mironcenkotola@gmail.com>
Date: Thu, 20 Oct 2022 08:26:59 +0700
Subject: [PATCH] transaction table added

---
 database.sqlite    | Bin 16384 -> 16384 bytes
 database/models.py |   9 ++++++++-
 locale.py          |   8 ++++++--
 main.py            |  22 +++++++++++++++++-----
 4 files changed, 31 insertions(+), 8 deletions(-)

diff --git a/database.sqlite b/database.sqlite
index 857d18ab649c28ac74b237e0b29a91c2529cecb9..d80da12e9097010d51c018ce4fe0dbae18061c17 100644
GIT binary patch
delta 280
zcmZo@U~Fh$oRG!woPocHFPQiF#>P@!&Manjabso1M()Y~c_joxiW2jR6O&6a^Yd7o
zgIpa$ChPMl`67ubDQGB_7N-`)XQn7=Hpwutiz_QLHrJOVCgr3eDMMm1gUxYO2yt}s
zaaB+XnLLkgKDUxfVo9o!f{SB_>trMTD=Z%v_&;nGRCvZO!^g<NAkS!KWN2<-WN2c+
z#LK|Iz{G!=f&T^n>CJ)y`}rNEnb;T<895b`^Ye;J3qVfgGBPkS(ls>DH8K!mWMdEq
lN@nII=jZYu#JRx6d}rYQ!~Y#<(QAHA7G_RHCT3<%P5?@~OpX8m

delta 175
zcmZo@U~Fh$oRG!wje)<2FO&D%W<h}#Ud}>hc5!26#zx7>A9y7sN{SNmiW8GdGV}9T
zoP%5)LtGU?{X7FhCrk1vF_uiW;ggVJV`3LqR%UDtE=f$vNk!67GC7Y=5~!$yZ~kOk
z{*x>(8TelU?YPe`E6T{iAkS`QWN2<-WN2bx^MC;ictNDVW<i5z{1YdLuyLVq0hFjO
A`2YX_

diff --git a/database/models.py b/database/models.py
index 63994c9..27e4e54 100644
--- a/database/models.py
+++ b/database/models.py
@@ -1,3 +1,5 @@
+import datetime
+
 from peewee import *
 from .base import conn
 
@@ -18,11 +20,16 @@ class User(BaseModel):
         table_name = 'User'
 
 
+# модель транзакции
 class Transaction(BaseModel):
     id = AutoField(column_name='TransactionId')
     type = CharField(column_name='TransactionType', max_length=10, null=False, default='incom')
     sum = FloatField(column_name='TransactionSum', null=False, default=0)
-    user = ForeignKeyField(User, unique=True)
+    user = ForeignKeyField(User, unique=False)
+    date = DateField(column_name='Date', default=datetime.datetime.today())
+
+    class Meta:
+        table_name = 'Transaction'
 
 
 conn.create_tables([User, Transaction])
diff --git a/locale.py b/locale.py
index 44bab3d..0d817ec 100644
--- a/locale.py
+++ b/locale.py
@@ -11,7 +11,9 @@ ru = {
 /add или + - добавить доход
 /rm или - - добавить расход
         ''',
-    'error404_message': 'Я не могу это сделать.'
+    'error404_message': 'Я не могу это сделать.',
+    'erradd': 'Используйте /add [сумма] или + [сумма]',
+    'errrm': 'Используйте /rm [сумма] или - [сумма]'
 }
 
 en = {
@@ -25,5 +27,7 @@ Hi i am finance bot and i can do this:
 /add или + - add incom
 /rm или - - add consuption
         ''',
-    'error404_message': 'I can\'t do this.'
+    'error404_message': 'I can\'t do this.',
+    'erradd': 'Use /add [sum] or + [sum]',
+    'errrm': 'Use /rm [sum] or - [sum]'
 }
diff --git a/main.py b/main.py
index f680e88..b7b09b5 100644
--- a/main.py
+++ b/main.py
@@ -2,13 +2,15 @@ from termcolor import cprint
 from bot import bot
 import locale
 from changelocale import change_locale
-from database.models import User
+from database.models import User, Transaction
 
 
+# функция ограничения количества знаков после запятой
 def to_fixed(num_obj, digits=0):
     return f"{num_obj:.{digits}f}"
 
 
+# основной обработчик для бота
 def main_bot(messages):
     for message in messages:
         if message.text.split(' ')[0] == '/start' or message.text.split(' ')[0] == '/help':
@@ -28,27 +30,37 @@ def main_bot(messages):
             if len(message.text.split(' ')) < 2:
                 mes = None
                 if locale.LOCALE == 'ru':
-                    mes = 'Используйте /add [сумма] или + [сумма]'
+                    mes = locale.ru['erradd']
                 elif locale.LOCALE == 'en':
-                    mes = 'Use /add [sum] or + [sum]'
+                    mes = locale.en['erradd']
                 bot.send_message(message.chat.id, mes)
             else:
                 user = User.get(username=message.chat.id)
                 user.balance += float(message.text.split(' ')[1])
                 user.save()
+                Transaction.create(
+                    sum=int(message.text.split(' ')[1]),
+                    type='incom',
+                    user=User.get(username=message.chat.id)
+                )
                 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 [сумма] или - [сумма]'
+                    mes = locale.ru['errrm']
                 elif locale.LOCALE == 'en':
-                    mes = 'Use /rm [sum] or - [sum]'
+                    mes = locale.en['errrm']
                 bot.send_message(message.chat.id, mes)
             else:
                 user = User.get(username=message.chat.id)
                 user.balance -= float(message.text.split(' ')[1])
                 user.save()
+                Transaction.create(
+                    sum=int(message.text.split(' ')[1]),
+                    type='consuption',
+                    user=User.get(username=message.chat.id)
+                )
                 bot.send_message(message.chat.id, to_fixed(user.balance, 2))
         else:
             mes = None