model.py 1.5 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647
  1. import enum
  2. from datetime import date, datetime, time
  3. from sqlalchemy import (
  4. BIGINT,
  5. JSON,
  6. Boolean,
  7. Date,
  8. DateTime,
  9. Float,
  10. Integer,
  11. String,
  12. Text,
  13. Time,
  14. )
  15. from sqlalchemy.orm import Mapped, mapped_column
  16. from app.core.base_model import ModelMixin, UserMixin
  17. class StatusEnum(enum.Enum):
  18. """状态枚举"""
  19. ACTIVE = "active"
  20. INACTIVE = "inactive"
  21. class DemoModel(ModelMixin, UserMixin):
  22. """
  23. 示例表 - 涵盖大多数常用数据类型
  24. """
  25. __tablename__: str = "gen_demo"
  26. __table_args__: dict[str, str] = {"comment": "示例表"}
  27. __loader_options__: list[str] = ["created_by", "updated_by"]
  28. # 字符串类型
  29. name: Mapped[str] = mapped_column(String(64), nullable=False, comment="名称")
  30. a: Mapped[int | None] = mapped_column(Integer, nullable=True, comment="整数")
  31. b: Mapped[int | None] = mapped_column(BIGINT, nullable=True, comment="大整数")
  32. c: Mapped[float | None] = mapped_column(Float, nullable=True, comment="浮点数")
  33. d: Mapped[bool] = mapped_column(Boolean, default=True, nullable=False, comment="布尔型")
  34. e: Mapped[date | None] = mapped_column(Date, nullable=True, comment="日期")
  35. f: Mapped[time | None] = mapped_column(Time, nullable=True, comment="时间")
  36. g: Mapped[datetime | None] = mapped_column(DateTime, nullable=True, comment="日期时间")
  37. h: Mapped[str | None] = mapped_column(Text, nullable=True, comment="长文本")
  38. i: Mapped[dict | None] = mapped_column(JSON, nullable=True, comment="元数据(JSON格式)")