app/models/user.py

import uuid
from datetime import datetime, timezone
from typing import Optional, List, TYPE_CHECKING
from sqlmodel import Field, SQLModel, Relationship
from .enums import RoleEnum, SpecialtyEnum
from .membership import Membership

if TYPE_CHECKING:
    from .procedure import Procedure
    from .team import Team

class User(SQLModel, table=True):
    """
    Represents a row in the `users` table.
    """
    __tablename__ = "users"

    id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
    email: str = Field(unique=True, index=True, nullable=False)
    full_name: str = Field(nullable=False)
    role: RoleEnum = Field(nullable=False)
    specialty: Optional[SpecialtyEnum] = Field(default=None)
    avatar_url: Optional[str] = Field(default=None)
    
    is_active: bool = Field(default=True)
    created_at: datetime = Field(default_factory=lambda: datetime.now(timezone.utc))

    # Relationships
    procedures: List["Procedure"] = Relationship(back_populates="doctor")
    teams: List["Team"] = Relationship(back_populates="members", link_model=Membership)