import enum import uuid from sqlalchemy import Boolean, Enum, String from sqlalchemy.dialects.postgresql import UUID from sqlalchemy.orm import Mapped, mapped_column, relationship from app.models.base import Base, TimestampMixin class UserRole(str, enum.Enum): USER = "USER" ADMIN = "ADMIN" class User(TimestampMixin, Base): __tablename__ = "users" id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4) email: Mapped[str] = mapped_column(String(320), unique=True, index=True, nullable=False) full_name: Mapped[str | None] = mapped_column(String(255), nullable=True) hashed_password: Mapped[str] = mapped_column(String(255), nullable=False) role: Mapped[UserRole] = mapped_column( Enum(UserRole, name="user_role"), nullable=False, default=UserRole.USER, server_default=UserRole.USER.value, ) is_active: Mapped[bool] = mapped_column(Boolean, nullable=False, default=True, server_default="true") pipeline_dialogs = relationship( "PipelineDialog", back_populates="user", cascade="all, delete-orphan", passive_deletes=True, lazy="selectin", ) actions = relationship("Action", passive_deletes=True, lazy="selectin") capabilities = relationship("Capability", passive_deletes=True, lazy="selectin")