"""
This module contains the models for the consumer logic.
"""

from pydantic import Field

from common.src.models.sql import BaseColumns
from .enums import ColumnAliases


class Columns(BaseColumns):
    """Columns model for the SQL queries and their output."""

    label_id: int = Field(..., alias=ColumnAliases.LABEL_ID)
    available_periods: list[int] = Field(..., alias=ColumnAliases.AVAILABLE_PERIODS)