model.py 2.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  1. from sqlalchemy import Integer, String, Text
  2. from sqlalchemy.orm import Mapped, mapped_column
  3. from app.config.setting import settings
  4. from app.core.base_model import ModelMixin, UserMixin
  5. def get_log_text_column_type():
  6. """
  7. 根据数据库类型选择适合存储大文本的列类型。
  8. MySQL 使用 LONGTEXT,PostgreSQL 使用 TEXT,其它数据库回退到 SQLAlchemy 的 Text。
  9. 返回:
  10. - type: SQLAlchemy 列类型(可用于 mapped_column)。
  11. """
  12. db_type = settings.DATABASE_TYPE
  13. if db_type == "mysql":
  14. from sqlalchemy.dialects.mysql import LONGTEXT
  15. return LONGTEXT
  16. elif db_type == "postgres":
  17. from sqlalchemy.dialects.postgresql import TEXT
  18. return TEXT
  19. else:
  20. return Text
  21. class OperationLogModel(ModelMixin, UserMixin):
  22. """
  23. 系统日志模型
  24. 日志类型:
  25. - 1: 登录日志
  26. - 2: 操作日志
  27. """
  28. __tablename__: str = "sys_log"
  29. __table_args__: dict[str, str] = {"comment": "系统日志表"}
  30. __loader_options__: list[str] = ["created_by", "updated_by"]
  31. type: Mapped[int] = mapped_column(Integer, comment="日志类型(1登录日志 2操作日志)")
  32. request_path: Mapped[str] = mapped_column(String(255), comment="请求路径")
  33. request_method: Mapped[str] = mapped_column(String(10), comment="请求方式")
  34. request_payload: Mapped[str | None] = mapped_column(get_log_text_column_type(), comment="请求体")
  35. request_ip: Mapped[str | None] = mapped_column(String(50), comment="请求IP地址")
  36. login_location: Mapped[str | None] = mapped_column(String(255), comment="登录位置")
  37. request_os: Mapped[str | None] = mapped_column(String(64), nullable=True, comment="操作系统")
  38. request_browser: Mapped[str | None] = mapped_column(String(64), nullable=True, comment="浏览器")
  39. response_code: Mapped[int] = mapped_column(Integer, comment="响应状态码")
  40. response_json: Mapped[str | None] = mapped_column(get_log_text_column_type(), nullable=True, comment="响应体")
  41. process_time: Mapped[str | None] = mapped_column(String(20), nullable=True, comment="处理时间")