Merge branch 'main' of https://gitee.com/wg123/video-gen

video-gen-api/app/models/contact_request.py

@@ -1,4 +1,4 @@
-from sqlalchemy import Boolean, ForeignKey, String, Text
+from sqlalchemy import Boolean, ForeignKey, String, Text, UniqueConstraint, Index
 from sqlalchemy.orm import Mapped, mapped_column
 
 from app.models.base import Base, TimestampMixin
@@ -14,4 +14,11 @@ class ContactRequest(Base, TimestampMixin):
     industry: Mapped[str] = mapped_column(String(64))
     name: Mapped[str] = mapped_column(String(64))
     message: Mapped[str | None] = mapped_column(Text, nullable=True)
-    is_handled: Mapped[bool] = mapped_column(Boolean, default=False)
+    is_handled: Mapped[bool] = mapped_column(Boolean, default=False)
+    submit_date: Mapped[str] = mapped_column(String(10), index=True)
+
+    __table_args__ = (
+        UniqueConstraint('user_id', 'submit_date', name='uq_contact_user_date'),
+        Index('idx_contact_handled_created', 'is_handled', 'created_at'),
+        Index('idx_contact_user_date_created', 'user_id', 'submit_date', 'created_at'),
+    )

video-gen-api/app/models/generation_record.py

@@ -1,6 +1,6 @@
 from datetime import datetime
 
-from sqlalchemy import DateTime, ForeignKey, Integer, String, Text, Float
+from sqlalchemy import DateTime, ForeignKey, Integer, String, Text, Float, Index
 from sqlalchemy.orm import Mapped, mapped_column
 
 from app.models.base import Base, TimestampMixin, SoftDeleteMixin
@@ -45,3 +45,8 @@ class GenerationRecord(Base, TimestampMixin, SoftDeleteMixin):
     )
     error_message: Mapped[str | None] = mapped_column(Text, nullable=True)
     idempotency_key: Mapped[str | None] = mapped_column(String(64), nullable=True, index=True)
+
+    __table_args__ = (
+        Index('idx_genrec_user_status_created', 'user_id', 'status', 'created_at'),
+        Index('idx_genrec_project_status', 'project_id', 'status'),
+    )

video-gen-api/app/models/menu_config.py

@@ -1,4 +1,4 @@
-from sqlalchemy import Boolean, Integer, String, ForeignKey
+from sqlalchemy import Boolean, Integer, String, ForeignKey, Index
 from sqlalchemy.orm import Mapped, mapped_column
 
 from app.models.base import Base, TimestampMixin
@@ -13,7 +13,11 @@ class MenuConfig(Base, TimestampMixin):
     icon: Mapped[str] = mapped_column(String(64), default="")
     sort_order: Mapped[int] = mapped_column(Integer, default=0)
     is_active: Mapped[bool] = mapped_column(Boolean, default=True)
-    parent_id: Mapped[str | None] = mapped_column(String(32), ForeignKey("menu_configs.id"), nullable=True)
-    menu_type: Mapped[str] = mapped_column(String(16), default="page")  # page / group
-    menu_target: Mapped[str] = mapped_column(String(16), default="frontend")  # frontend / admin / both
-    is_default: Mapped[bool] = mapped_column(Boolean, default=False)  # default show for new users
+    parent_id: Mapped[str | None] = mapped_column(String(32), ForeignKey("menu_configs.id"), nullable=True, index=True)
+    menu_type: Mapped[str] = mapped_column(String(16), default="page")
+    menu_target: Mapped[str] = mapped_column(String(16), default="frontend")
+    is_default: Mapped[bool] = mapped_column(Boolean, default=False)
+
+    __table_args__ = (
+        Index('idx_menu_target_active_sort', 'menu_target', 'is_active', 'sort_order'),
+    )

video-gen-api/app/models/operation_log.py

@@ -1,4 +1,4 @@
-from sqlalchemy import Integer, String, Text
+from sqlalchemy import Integer, String, Text, Index
 from sqlalchemy.orm import Mapped, mapped_column
 
 from app.models.base import Base, TimestampMixin
@@ -10,8 +10,13 @@ class OperationLog(Base, TimestampMixin):
     id: Mapped[str] = mapped_column(String(32), primary_key=True)
     user_id: Mapped[str] = mapped_column(String(32), index=True)
     username: Mapped[str] = mapped_column(String(64))
-    action: Mapped[str] = mapped_column(String(128))  # e.g. "创建用户", "修改菜单"
-    method: Mapped[str] = mapped_column(String(10))  # POST/PUT/DELETE
+    action: Mapped[str] = mapped_column(String(128))
+    method: Mapped[str] = mapped_column(String(10))
     path: Mapped[str] = mapped_column(String(256))
-    detail: Mapped[str | None] = mapped_column(Text, nullable=True)  # JSON detail
+    detail: Mapped[str | None] = mapped_column(Text, nullable=True)
     ip: Mapped[str | None] = mapped_column(String(64), nullable=True)
+
+    __table_args__ = (
+        Index('idx_oplog_user_created', 'user_id', 'created_at'),
+        Index('idx_oplog_created', 'created_at'),
+    )

video-gen-api/app/models/payment_order.py

@@ -1,6 +1,6 @@
 from datetime import datetime
 
-from sqlalchemy import DateTime, Float, ForeignKey, Integer, String
+from sqlalchemy import DateTime, Float, ForeignKey, Integer, String, Index
 from sqlalchemy.orm import Mapped, mapped_column
 
 from app.models.base import Base, TimestampMixin
@@ -22,9 +22,13 @@ class PaymentOrder(Base, TimestampMixin):
         DateTime(timezone=True), nullable=True
     )
     trade_no: Mapped[str | None] = mapped_column(String(128), nullable=True)
-    # Refund fields
     refund_trade_no: Mapped[str | None] = mapped_column(String(128), nullable=True)
     refunded_at: Mapped[datetime | None] = mapped_column(
         DateTime(timezone=True), nullable=True
     )
     refund_amount: Mapped[float | None] = mapped_column(Float, nullable=True)
+
+    __table_args__ = (
+        Index('idx_payorder_user_status_created', 'user_id', 'status', 'created_at'),
+        Index('idx_payorder_status_created', 'status', 'created_at'),
+    )

video-gen-api/app/models/upload_task.py

@@ -1,4 +1,4 @@
-from sqlalchemy import String, Text, Integer
+from sqlalchemy import String, Text, Integer, Index
 from sqlalchemy.orm import Mapped, mapped_column
 
 from app.models.base import Base, TimestampMixin, SoftDeleteMixin
@@ -31,3 +31,8 @@ class UploadTask(Base, TimestampMixin, SoftDeleteMixin):
     other_info: Mapped[str | None] = mapped_column(
         String(500), nullable=True, comment="其他信息"
     )
+
+    __table_args__ = (
+        Index('idx_upload_user_status_created', 'user_id', 'status', 'created_at'),
+        Index('idx_upload_oauth_status', 'oauth_id', 'status'),
+    )