from aiogram.fsm.state import State, StatesGroup


class BudgetSetup(StatesGroup):
    waiting_name = State()
    waiting_amount = State()
    waiting_end_date = State()


class DatedExpense(StatesGroup):
    waiting_date = State()
    waiting_expense = State()