57 lines
2.4 KiB
Python
57 lines
2.4 KiB
Python
from datetime import datetime
|
|
|
|
from sqlalchemy import Boolean, DateTime, Integer, String, JSON
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
|
|
|
from app.enums.user import FrontendUserKind
|
|
from app.models.base import Base, TimestampMixin
|
|
|
|
|
|
class User(Base, TimestampMixin):
|
|
__tablename__ = "users"
|
|
__allow_unmapped__ = True
|
|
|
|
id: Mapped[str] = mapped_column(String(32), primary_key=True)
|
|
username: Mapped[str] = mapped_column(String(64), unique=True, index=True)
|
|
email: Mapped[str | None] = mapped_column(String(255), unique=True, nullable=True)
|
|
phone: Mapped[str | None] = mapped_column(String(20), unique=True, nullable=True)
|
|
hashed_password: Mapped[str | None] = mapped_column(String(128), nullable=True)
|
|
avatar: Mapped[str | None] = mapped_column(String(512), nullable=True)
|
|
is_active: Mapped[bool] = mapped_column(Boolean, default=True)
|
|
is_admin: Mapped[bool] = mapped_column(Boolean, default=False)
|
|
user_type: Mapped[str] = mapped_column(String(16), default="frontend", index=True)
|
|
frontend_user_kind: Mapped[str] = mapped_column(
|
|
String(16),
|
|
default=FrontendUserKind.EXTERNAL.value,
|
|
server_default=FrontendUserKind.EXTERNAL.value,
|
|
index=True,
|
|
nullable=False,
|
|
)
|
|
team_id: Mapped[str | None] = mapped_column(String(32), nullable=True, index=True)
|
|
last_login_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)
|
|
password_set_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)
|
|
first_membership_paid_at: Mapped[datetime | None] = mapped_column(
|
|
DateTime(timezone=True), nullable=True, index=True
|
|
)
|
|
allowed_menus: Mapped[list | None] = mapped_column(JSON, nullable=True)
|
|
private_portrait_asset_limit: Mapped[int] = mapped_column(
|
|
Integer, default=50, server_default="50", nullable=False
|
|
)
|
|
|
|
# Token 版本号 — 每次登录/退出时递增，用于实现单设备登录（全局互斥）
|
|
token_version: Mapped[int] = mapped_column(
|
|
Integer, default=0, server_default="0", nullable=False
|
|
)
|
|
|
|
@property
|
|
def credits(self) -> float:
|
|
return float(getattr(self, "_credits_snapshot", 0.0) or 0.0)
|
|
|
|
@credits.setter
|
|
def credits(self, value: float | int) -> None:
|
|
self._credits_snapshot = round(float(value or 0.0), 2)
|
|
|
|
@property
|
|
def must_set_password(self) -> bool:
|
|
return self.user_type == "frontend" and not self.hashed_password
|