예제 #1
0
async def process_callback_button1(callback_query: types.CallbackQuery):
    global product
    product.category = callback_query.data
    DBMS.add_product(product)
    await bot.send_message(
        callback_query.message.chat.id,
        f"Товар {product.codename} на сумму {product.price} "
        f"добавлен в категорию {product.category}.")
#%% Если категория данного товара нам уже известа,
# метод product_exist добавит новый расход по данному товару в БД,
# и вернёт товар с его категорией (category='Продукты').
raw_message = 'Бананы - 150'
product = DBMS.product_exist(raw_message)

print(product)


#%% Если категория уже в БД, но продукт - нет.
raw_message = 'Клубника - 200'
product = DBMS.product_exist(raw_message)
if not product.category:
    product.category = 'Продукты'
DBMS.add_product(product)


#%% Вывод всех категорий при помощи метода all_categories.
print('Все категории:')
[print(category) for category in DBMS.all_categories()]


#%% Вывод всех расходов по категориям.
print(DBMS.db.all_categories_costs())


##############
### Доходы ###
##############