from typing import TYPE_CHECKING, List, Optional

from sqlmodel import Field, Relationship, SQLModel

if TYPE_CHECKING:
    from app.models.route import Route
    from app.models.user import User


class OrganizationBase(SQLModel):
    name: str = Field(unique=True)


class Organization(OrganizationBase, table=True):
    __tablename__ = "organization"

    id: Optional[int] = Field(default=None, primary_key=True)

    users: List["User"] = Relationship(back_populates="organization")
    routes: List["Route"] = Relationship(back_populates="organization")