from aiogram.fsm.state import State, StatesGroup


class MenuState(StatesGroup):
    MENU = State()
    PROFILE = State()
    SUBSCRIPTIONS = State()


class SetProfileNameState(StatesGroup):
    WAITING_FOR_NAME = State()


class SetProfilePhoneState(StatesGroup):
    WAITING_FOR_PHONE = State()


class SetProfileBirthdayState(StatesGroup):
    WAITING_FOR_DATE = State()


class SetProfileGiftPaymentDataState(StatesGroup):
    WAITING_FOR_PHONE = State()
    WAITING_FOR_BANK = State()


class AddSubscriptionState(StatesGroup):
    WAITING_FOR_PHONE = State()
    WAITING_FOR_DATE = State()
    WAITING_FOR_NAME = State()
    WAITING_FOR_POOL_DECISION = State()
    WAITING_FOR_CONFIRMATION = State()


class AddSubscriptionPoolState(StatesGroup):
    WAITING_FOR_DESCRIPTION = State()
    WAITING_FOR_PAYMENT_PHONE = State()
    WAITING_FOR_PAYMENT_BANK = State()