def check_random_month_expenses(message):
"""
Handler to check about of money spent for a specific category in the past
:param message:
:return:
"""
available_sheets = list_all_sheets_in_finance_directory()
month_selected_msg = bot.reply_to(
message, 'Which month expenses do you want to check? ')
bot.send_message(
message.chat.id, "\n".join(
[f"{i + 1}. {elem}" for i, elem in enumerate(available_sheets)]))
bot.register_next_step_handler(month_selected_msg, check_selected_month)
def add_current_month_income_input(message):
"""
Method callback that is inserting the income for current month
:param message:
"""
sh = open_google_sheets(formatted_data)
worksheet = sh.worksheet("Income")
input_data = message.text.split(':')
read_row_str = add_input_data_to_google_sheet(message, input_data,
worksheet)
bot.send_message(
message.chat.id,
f"{read_row_str} has been added to {worksheet.title} worksheet into"
f" {formatted_data} file.")
def handle_start_help(message):
"""
Shows start menu with all supported commands.
"""
bot.send_message(
message.chat.id, "All available commands:\n"
"/start or /help shows help menu\n\n"
"/CurrentMonthBalance or /CMB or /cb\n shows current month balance\n\n"
"/CurrentMonthExpenseByCategory or /CME or /ce\n retrieves expenses for current"
"month for every category\n\n"
"/CheckRandomMonthCategoryExpenses or /RME or /re\n retrieves expenses for "
"current month for every category\n\n"
"category\n\n\n"
"/AddExpenseToCurrentMonth or /AECM or /ae\n adds expense to current month\n\n"
"/AddCurrentMonthIncome or ACMI or ai\n adds income to current month\n\n"
"/FormatDefinedFile or /FDF\n restores correct formating for whole document"
" defined by date, takes up to 5 minutes, do not use frequently\n")
def add_current_month_expense_input_category(message,
worksheet_list,
month=None):
"""
:param month: Not used yet. #TODO maybe I want later to have the possibility to add expenses for old months sheets
:param message:
:param worksheet_list:
"""
category_num = validate_input(message)
try:
worksheet = worksheet_list[category_num]
if worksheet in worksheet_list and int(message.text) > 0:
bot.send_message(
message.chat.id, 'You have chosen: ' + str(message.text) +
") " + (worksheet_list[category_num]).title)
msg = bot.reply_to(
message, 'Please enter tag and price as example "Bread: 50": ')
bot.register_next_step_handler(
msg, add_current_month_expense_input_string, worksheet)
else:
bot.send_message(
message.chat.id,
'Category with number: ' + str(message.text) + ' '
'not found! Please retry')
msg = bot.reply_to(
message,
'Please choose the expense category number from message above: '
)
bot.register_next_step_handler(
msg, add_current_month_expense_input_category, worksheet_list)
except Exception as e:
bot.reply_to(
message, 'ERROR!\nCategory ' + str(message.text) +
' not found!\nTry once more!')
msg = bot.reply_to(
message,
'Please choose the expense category number from message above: ')
bot.register_next_step_handler(
msg, add_current_month_expense_input_category, worksheet_list)
logger.error(str(e))
def check_month_expenses_per_category(message,
worksheet_list,
selected_month=None):
"""
returns Sum spent per category per month
:param selected_month:
:param message:
:param worksheet_list:
"""
category_num = validate_input(message)
try:
worksheet = worksheet_list[category_num]
cols_count = worksheet.col_count
bot.send_message(
message.chat.id,
f" {worksheet.title} expenses for {selected_month} are {worksheet.col_values(cols_count)[0]}."
)
except Exception as e:
bot.reply_to(
message,
"Please choose the expense category number from message above: ")
def add_current_month_expense_input_string(message, worksheet):
"""
This function represents the process of adding expense to current month
"""
input_data = validate_input(message)
if len(input_data) == 2:
read_row_str = add_input_data_to_google_sheet(message, input_data,
worksheet)
bot.send_message(
message.chat.id,
f"{read_row_str} has been added to {worksheet.title} worksheet into"
f" {formatted_data} file.")
else:
bot.send_message(
message.chat.id,
f"You have entered *** {message.text} *** - wrong input format (Expense:price) or not 2"
"parameters in the input")
msg = bot.reply_to(
message, f"Please enter the expense for {str(formatted_data)} and "
f"{str(worksheet.title)} category in format expense:price")
bot.register_next_step_handler(msg,
add_current_month_expense_input_string,
worksheet)
def current_month_balance(message):
"""
Method that returns 3 different data to User: income, expense and balance
:param message:
:return: Current income, expense and balance
"""
sh = open_google_sheets(formatted_data)
worksheet = sh.worksheet("Balance")
bot.send_message(
message.chat.id,
"Current month income is: " + worksheet.acell('B15').value)
bot.send_message(
message.chat.id,
"Current month expenses are: " + worksheet.acell('D15').value)
bot.send_message(
message.chat.id,
"Current month balance is: " + worksheet.acell('F1').value)