addall rmall addtoday rmtoday
This commit is contained in:
wt committed
1 parent
b99fb5a72f
commit
4dc52d7add
6 files changed
+168
-69
No files matched your search
@@ -1,6 +1,8 @@
|
||||
import datetime
|
||||
|
||||
from termcolor import cprint
|
||||
from bot import bot
|
||||
import locale
|
||||
import locales
|
||||
from changelocale import change_locale
|
||||
from database.models import User, Transaction
|
||||
|
||||
@@ -15,59 +17,147 @@ def main_bot(messages):
|
||||
for message in messages:
|
||||
if message.text.split(' ')[0] == '/start' or message.text.split(' ')[0] == '/help':
|
||||
mes = None
|
||||
if locale.LOCALE == 'ru':
|
||||
mes = locale.ru['help_message']
|
||||
elif locale.LOCALE == 'en':
|
||||
mes = locale.en['help_message']
|
||||
if locales.LOCALE == 'ru':
|
||||
mes = locales.ru['help_message']
|
||||
elif locales.LOCALE == 'en':
|
||||
mes = locales.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)
|
||||
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))
|
||||
mes = ''
|
||||
if locales.LOCALE == 'ru':
|
||||
mes = 'Баланс: '
|
||||
elif locales.LOCALE == 'en':
|
||||
mes = 'Balance: '
|
||||
mes += to_fixed(user.balance, 2)
|
||||
bot.send_message(message.chat.id, mes)
|
||||
elif message.text.split(' ')[0] == '/add' or message.text.split(' ')[0] == '+':
|
||||
if len(message.text.split(' ')) < 2:
|
||||
mes = None
|
||||
if locale.LOCALE == 'ru':
|
||||
mes = locale.ru['erradd']
|
||||
elif locale.LOCALE == 'en':
|
||||
mes = locale.en['erradd']
|
||||
if locales.LOCALE == 'ru':
|
||||
mes = locales.ru['erradd']
|
||||
elif locales.LOCALE == 'en':
|
||||
mes = locales.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)
|
||||
)
|
||||
if len(message.text.split(' ')) == 3:
|
||||
Transaction.create(
|
||||
sum=int(message.text.split(' ')[1]),
|
||||
type='incom',
|
||||
user=User.get(username=message.chat.id),
|
||||
category=message.text.split(' ')[2]
|
||||
)
|
||||
else:
|
||||
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 = locale.ru['errrm']
|
||||
elif locale.LOCALE == 'en':
|
||||
mes = locale.en['errrm']
|
||||
if locales.LOCALE == 'ru':
|
||||
mes = locales.ru['errrm']
|
||||
elif locales.LOCALE == 'en':
|
||||
mes = locales.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)
|
||||
)
|
||||
if len(message.text.split(' ')) == 3:
|
||||
Transaction.create(
|
||||
sum=int(message.text.split(' ')[1]),
|
||||
type='consuption',
|
||||
user=User.get(username=message.chat.id),
|
||||
category=message.text.split(' ')[2]
|
||||
)
|
||||
else:
|
||||
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))
|
||||
elif message.text.split(' ')[0] == '/addtoday':
|
||||
transactions = Transaction.select().where(
|
||||
Transaction.type == 'incom',
|
||||
Transaction.date == datetime.datetime.today(),
|
||||
Transaction.user == User.get(username=message.chat.id).id
|
||||
)
|
||||
mes = ''
|
||||
if locales.LOCALE == 'ru':
|
||||
mes = 'Ваши доходы за сегодня:\n\n'
|
||||
elif locales.LOCALE == 'en':
|
||||
mes = 'Your incomes today:\n\n'
|
||||
iteration = 1
|
||||
for i in transactions:
|
||||
mes += str(iteration) + '. ' + str(i.sum) + ' ' + i.category + '\n\n'
|
||||
iteration += 1
|
||||
|
||||
bot.send_message(message.chat.id, mes)
|
||||
elif message.text.split(' ')[0] == '/rmtoday':
|
||||
transactions = Transaction.select().where(
|
||||
Transaction.type == 'consuption',
|
||||
Transaction.date == datetime.datetime.today(),
|
||||
Transaction.user == User.get(username=message.chat.id).id
|
||||
)
|
||||
mes = ''
|
||||
if locales.LOCALE == 'ru':
|
||||
mes = 'Ваши расходы за сегодня:\n\n'
|
||||
elif locales.LOCALE == 'en':
|
||||
mes = 'Your consuptions today:\n\n'
|
||||
iteration = 1
|
||||
for i in transactions:
|
||||
mes += str(iteration) + '. ' + str(i.sum) + ' ' + i.category + '\n\n'
|
||||
iteration += 1
|
||||
|
||||
bot.send_message(message.chat.id, mes)
|
||||
elif message.text.split(' ')[0] == '/rmall':
|
||||
transactions = Transaction.select().where(
|
||||
Transaction.type == 'consuption',
|
||||
Transaction.user == User.get(username=message.chat.id).id
|
||||
)
|
||||
mes = ''
|
||||
if locales.LOCALE == 'ru':
|
||||
mes = 'Ваши расходы за все время:\n\n'
|
||||
elif locales.LOCALE == 'en':
|
||||
mes = 'Your all consuptions:\n\n'
|
||||
iteration = 1
|
||||
for i in transactions:
|
||||
mes += str(iteration) + '. ' + str(i.sum) + ' ' + i.category + '\n\n'
|
||||
iteration += 1
|
||||
|
||||
bot.send_message(message.chat.id, mes)
|
||||
elif message.text.split(' ')[0] == '/addall':
|
||||
transactions = Transaction.select().where(
|
||||
Transaction.type == 'incom',
|
||||
Transaction.user == User.get(username=message.chat.id).id
|
||||
)
|
||||
mes = ''
|
||||
if locales.LOCALE == 'ru':
|
||||
mes = 'Ваши доходы за все время:\n\n'
|
||||
elif locales.LOCALE == 'en':
|
||||
mes = 'Your all incomes:\n\n'
|
||||
iteration = 1
|
||||
for i in transactions:
|
||||
mes += str(iteration) + '. ' + str(i.sum) + ' ' + i.category + '\n\n'
|
||||
iteration += 1
|
||||
|
||||
bot.send_message(message.chat.id, mes)
|
||||
else:
|
||||
mes = None
|
||||
if locale.LOCALE == 'ru':
|
||||
mes = locale.ru['error404_message']
|
||||
elif locale.LOCALE == 'en':
|
||||
mes = locale.en['error404_message']
|
||||
if locales.LOCALE == 'ru':
|
||||
mes = locales.ru['error404_message']
|
||||
elif locales.LOCALE == 'en':
|
||||
mes = locales.en['error404_message']
|
||||
bot.send_message(message.chat.id, mes)
|
||||
|
||||
|
||||
@@ -75,11 +165,11 @@ bot.set_update_listener(main_bot)
|
||||
|
||||
try:
|
||||
bot.polling(none_stop=True)
|
||||
while True:
|
||||
pass
|
||||
except KeyboardInterrupt:
|
||||
exit('byby\n')
|
||||
cprint('\ngoodbye', 'green')
|
||||
exit()
|
||||
except Exception as e:
|
||||
cprint(str(e), 'red')
|
||||
bot.polling(none_stop=True)
|
||||
|
||||
while True:
|
||||
pass
|
||||
Reference in new issue
Block a user