"""States for prediction process."""

from aiogram.fsm.state import StatesGroup, State


class PredictionState(StatesGroup):
    """States for match prediction process."""

    waiting_for_prediction_type = State()
    waiting_for_team1_score = State()
    waiting_for_team2_score = State()
    waiting_for_result_prediction = State()


class PredictionAdminState(StatesGroup):
    """States for admin match management."""

    waiting_for_team1 = State()
    waiting_for_team2 = State()
    waiting_for_team1_flag = State()
    waiting_for_team2_flag = State()
    waiting_for_match_time = State()
    waiting_for_result_team1 = State()
    waiting_for_result_team2 = State()
    waiting_for_broadcast_content = State()