def menu_choice(bot, update, user_data):
clear_user_data(user_data, "categories_menu")
user = update.message.from_user
choice = update.message.text
if choice == action_choices.back:
update.message.reply_text(f"{user.first_name}, слушаю тебя",
reply_markup=get_keyboard("main_menu"))
clear_user_data(user_data, "categories_menu")
return ConversationHandler.END
category_type = "income" if choice in [
action_choices.income_add, action_choices.income_remove
] else "expense"
action = "add" if choice in [
action_choices.income_add, action_choices.expense_add
] else "remove"
categories = data_observer.get_all_category_names(user.id, category_type)
if action == "remove":
user_data["delete_category_type"] = category_type
reply_text = "Выберите категорию, которую хотите удалить:"
update.message.reply_text(reply_text,
reply_markup=get_keyboard_from_list(
categories, one_time_keyboard=True))
return "delete_category"
elif action == "add":
user_data["add_category_type"] = category_type
update.message.reply_text("Какое название будет у новой категории?")
return "add_category"
def index():
if not current_user.is_authenticated:
return redirect(url_for("user.login"))
user_id = current_user.user_id
return render_template("category/index.html",
categories_list=sorted(
data_observer.get_all_category_names(user_id)))
def confirm_delete_category(bot, update, user_data):
user = update.message.from_user
choice = update.message.text
# Lets check whether everything is correct
# (category_name and category_type)
if ("delete_category_name" in user_data
and "delete_category_type" in user_data):
# We know category_name and category_type
category_name = user_data["delete_category_name"]
category_type = user_data["delete_category_type"]
# user confirmed that he want to delete category
if choice.lower() == confirmation.yes.lower():
if data_manipulator.delete_category(user.id, category_name,
category_type):
logging.info(f"category {category_name} deleted")
reply_text = f"Категория {category_name} удалена! Что дальше?"
else:
logging.info(f"couldn't delete category {category_name}")
reply_text = f"Не получилось удалить категорию {category_name}"
update.message.reply_text(reply_text,
reply_markup=get_keyboard("main_menu"))
clear_user_data(user_data, "categories_menu")
return ConversationHandler.END
# user cancel
elif choice.lower() == confirmation.no.lower():
reply_text = "Выберите категорию, которую хотите удалить:"
categories = data_observer.get_all_category_names(
user.id, user_data["delete_category_type"])
update.message.reply_text(reply_text,
reply_markup=get_keyboard_from_list(
categories, one_time_keyboard=True))
return "delete_category"
# user reply is neither yes nor no - wrong answer
else:
category_name = user_data["delete_category_name"]
reply_text = f"Не понимаю твой ответ. Ты уверен, \
что хочешь удалить категорию {category_name}?"
update.message.reply_text(reply_text,
reply_markup=get_keyboard(
"confirmation",
one_time_keyboard=True))
return "confirm_delete_category"
else:
logging.info("delete_category_type is not in user_data")
update.message.reply_text(
"Не получается определить категорию, выбери еще раз желаемое \
действие",
reply_markup=get_keyboard("main_menu"))
clear_user_data(user_data, "categories_menu")
return ConversationHandler.END
def add_category(bot, update, user_data):
user = update.message.from_user
new_category = update.message.text.capitalize()
try:
if "add_category_type" not in user_data:
raise CategoryTypeNotInUserData
category_added = data_manipulator.add_category(
new_category, user.id, user_data["add_category_type"])
if category_added:
reply_text = f"Категория {new_category} добавлена!"
else:
reply_text = f"Не получилось добавить категорию {new_category}"
update.message.reply_text(reply_text,
reply_markup=get_keyboard("main_menu"))
clear_user_data(user_data, "categories_menu")
return ConversationHandler.END
except CategoryTypeNotInUserData:
logging.debug("add_category: category_type is not defined!")
update.message.reply_text("не получается определить тип категории, \
выбери еще раз желаемое действие",
reply_markup=get_keyboard("main_menu"))
clear_user_data(user_data, "categories_menu")
return ConversationHandler.END
except ObjectAlreadyExist:
reply_text = "Такая категория уже существует, кстати, вот список всех \
категорий, которые у тебя есть:"
for i, category in enumerate(
sorted(
data_observer.get_all_category_names(
user.id,
user_data["add_category_type"],
status="active"))):
reply_text = f"{reply_text}\n{i+1}. {category}"
reply_text = f"{reply_text}\n Введи другое название"
update.message.reply_text(reply_text)
return "add_category"
def delete_category(bot, update, user_data):
user = update.message.from_user
category_name = update.message.text
# Something goes wrong and we don't know category_type
if "delete_category_type" not in user_data:
logging.info("category_type is not defined!")
update.message.reply_text("Не получается определить тип категории, \
выбери еще раз желаемое действие",
reply_markup=get_keyboard("main_menu"))
clear_user_data(user_data, "categories_menu")
return ConversationHandler.END
# User decided to cancel
if category_name == action_choices.back:
clear_user_data(user_data, "categories_menu")
update.message.reply_text("Выбери, что именно нужно сделать",
reply_markup=get_keyboard_from_list(
list(action_choices),
cancel=False,
one_time_keyboard=True))
return "menu_choice"
# Get necessary information and check whether the category_name is
# correct. In case everythin is ok ask for confirmation
categories = data_observer.get_all_category_names(
user.id, user_data["delete_category_type"])
if category_name in categories:
user_data["delete_category_name"] = category_name
update.message.reply_text(
f"Уверен, что хочешь удалить {category_name}?",
reply_markup=get_keyboard("confirmation", one_time_keyboard=True))
return "confirm_delete_category"
else:
update.message.reply_text(f"Не могу найти {category_name}, \
выбери категорию при помощи клавиатуры",
reply_markup=get_keyboard_from_list(
categories, one_time_keyboard=True))
return "delete_category"
def start_chat(bot, update):
user = update.effective_user
text = f"Привет, {user.first_name},"
if is_existing_user(user.id):
text = f"{text} слушаю тебя"
else:
text = f"{text} я помогу тебе управлять личными финансами. Для начала \
я заведу тебе несколько стандартных категорий, пару кошельков и вид \
дохода - Зарплата"
data_manipulator.add_user(user.id, user.first_name, user.username)
text = f"{text}\n\nТвои категории расходов:"
for i, category in enumerate(
sorted(data_observer.get_all_category_names(user.id))):
text = f"{text}\n{i+1}. {category}"
text = f"{text}\n\nТвои счета:"
for i, account in enumerate(
sorted(data_observer.get_all_account_names(user.id))):
text = f"{text}\n{i+1}. {account}"
update.message.reply_text(text, reply_markup=get_keyboard())
def parse_transaction(line, user_id):
text = line.lower()
try:
# Search for transaction amount and currency
amount_mo = re.search(amount_with_currency_pattern(), text)
transaction_amount_currency = amount_mo.group().strip()
except NoCurrencies:
logging.error("There are no currencies in database")
raise
transaction_amount = re.search(AMOUNT_PATTERN,
transaction_amount_currency).group()
transaction_currency = transaction_amount_currency.replace(
transaction_amount, "").strip()
transaction_amount = transaction_amount.replace(",", ".")
text = text.replace(transaction_amount_currency, "")
# Search for category
expense_category_list = [
category for category in data_observer.get_all_category_names(user_id)
if category.lower() in text
]
transaction_category = ""
transaction_type = ""
if expense_category_list:
transaction_category = max(expense_category_list, key=len)
text = text.replace(transaction_category, "")
transaction_type = "expense"
else:
income_category_list = [
category for category in data_observer.get_all_category_names(
user_id, "income") if category.lower() in text
]
if income_category_list:
transaction_category = max(income_category_list, key=len)
text = text.replace(transaction_category, "")
transaction_type = "income"
else:
logging.debug("при вводе транзакции указана неверная категория")
raise WrongCategory
# Search for account
account_list = [
account
for account in data_observer.get_all_account_names(user_id,
with_info=False)
if account.lower() in text
]
transaction_account = ""
if account_list:
transaction_account = max(account_list, key=len)
text = text.replace(transaction_account, "")
else:
# считаем этот счет счетом по-умолчанию,
# далее добавим возможность выбора"
transaction_account = "Наличные"
return {
"amount": transaction_amount,
"currency": transaction_currency,
"category": transaction_category,
"type": transaction_type,
"account": transaction_account
}
def update():
if not current_user.is_authenticated:
return redirect(url_for("user.login"))
user_id = current_user.user_id
categories_list = sorted(data_observer.get_all_category_names(user_id))
return json.dumps(categories_list)