"""User and UserTenant models.""" from __future__ import annotations import uuid from datetime import datetime from typing import Any from sqlalchemy import Boolean, DateTime, ForeignKey, String, UniqueConstraint, func from sqlalchemy.dialects.postgresql import JSONB from sqlalchemy.dialects.postgresql import UUID as PGUUID from sqlalchemy.orm import Mapped, mapped_column 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() )