40 lines
1.4 KiB
Python
40 lines
1.4 KiB
Python
|
from datetime import datetime
|
||
|
|
|
||
|
|
from sqlalchemy import Boolean, DateTime, ForeignKey, String
|
||
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
||
|
|
|
||
|
|
from app.models.base import Base, utcnow
|
||
|
|
|
||
|
|
|
||
|
|
class User(Base):
|
||
|
|
__tablename__ = "users"
|
||
|
|
|
||
|
|
id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
|
||
|
|
# Entra object id; null for dev-mode/seeded users.
|
||
|
|
entra_oid: Mapped[str | None] = mapped_column(String(64), unique=True, nullable=True)
|
||
|
|
email: Mapped[str] = mapped_column(String(255), unique=True, index=True)
|
||
|
|
display_name: Mapped[str] = mapped_column(String(255))
|
||
|
|
employee_id: Mapped[str | None] = mapped_column(String(64), nullable=True)
|
||
|
|
is_active: Mapped[bool] = mapped_column(Boolean, default=True)
|
||
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, default=utcnow)
|
||
|
|
last_login_at: Mapped[datetime | None] = mapped_column(DateTime, nullable=True)
|
||
|
|
|
||
|
|
role_rows: Mapped[list["UserRole"]] = relationship(
|
||
|
|
back_populates="user", cascade="all, delete-orphan", lazy="selectin"
|
||
|
|
)
|
||
|
|
|
||
|
|
@property
|
||
|
|
def roles(self) -> list[str]:
|
||
|
|
return sorted(r.role for r in self.role_rows)
|
||
|
|
|
||
|
|
|
||
|
|
class UserRole(Base):
|
||
|
|
__tablename__ = "user_roles"
|
||
|
|
|
||
|
|
user_id: Mapped[int] = mapped_column(
|
||
|
|
ForeignKey("users.id", ondelete="CASCADE"), primary_key=True
|
||
|
|
)
|
||
|
|
role: Mapped[str] = mapped_column(String(40), primary_key=True)
|
||
|
|
|
||
|
|
user: Mapped[User] = relationship(back_populates="role_rows")
|