"""User and UserTenant models.""" from __future__ import annotations import uuid from datetime import datetime from typing import Any from sqlalchemy import String, Boolean, DateTime, ForeignKey, func, UniqueConstraint from sqlalchemy.dialects.postgresql import UUID as PGUUID, JSONB from sqlalchemy.orm import Mapped, mapped_column, relationship from app.core.db import Base, TenantMixin class User(Base, TenantMixin): """User entity — belongs to a tenant, can be member of multiple tenants.""" __tablename__ = "users" __table_args__ = ( UniqueConstraint("tenant_id", "email", name="uq_users_tenant_email"), ) id: Mapped[uuid.UUID] = mapped_column( PGUUID(as_uuid=True), primary_key=True, default=uuid.uuid4 ) email: Mapped[str] = mapped_column(String(255), nullable=False, index=True) name: Mapped[str] = mapped_column(String(200), nullable=False) password_hash: Mapped[str] = mapped_column(String(255), nullable=False) role: Mapped[str] = mapped_column(String(50), nullable=False, default="viewer") is_active: Mapped[bool] = mapped_column(Boolean, default=True, nullable=False) preferences: Mapped[dict[str, Any]] = mapped_column(JSONB, default=dict, nullable=False) class UserTenant(Base): """N:M association — user membership in tenants.""" __tablename__ = "user_tenants" user_id: Mapped[uuid.UUID] = mapped_column( PGUUID(as_uuid=True), ForeignKey("users.id", ondelete="CASCADE"), primary_key=True ) tenant_id: Mapped[uuid.UUID] = mapped_column( PGUUID(as_uuid=True), ForeignKey("tenants.id", ondelete="CASCADE"), primary_key=True ) is_default: Mapped[bool] = mapped_column(Boolean, default=False, nullable=False) created_at: Mapped[datetime] = mapped_column( DateTime(timezone=True), nullable=False, server_default=func.now() )