cbh/api/events/models.py

from datetime import datetime
from pydantic import Field

from cbh.api.account.models import AccountShorten
from cbh.api.events.dto import EventType
from cbh.core.database import MongoBaseModel, MongoBaseShortenModel


class EventModel(MongoBaseModel):
    """
    Event model.
    """

    reason: str | None = None
    type: EventType

    startDate: datetime
    endDate: datetime

    coach: AccountShorten

    isActive: bool = True
    datetimeInserted: datetime = Field(default_factory=datetime.now)


class EventShorten(MongoBaseShortenModel):
    """
    Event shorten model.
    """

    startDate: datetime
    endDate: datetime

    isActive: bool