1

video-gen-api/app/api/v1/admin.py

@@ -2210,6 +2210,8 @@ async def admin_list_generation_records(
     status: str | None = Query(None),
     engine_id: str | None = Query(None),
     include_media_references: bool | None = Query(None),
+    start_date: str | None = Query(None, description="创建时间起始，格式 YYYY-MM-DD"),
+    end_date: str | None = Query(None, description="创建时间结束，格式 YYYY-MM-DD"),
     page: int = Query(1, ge=1),
     page_size: int = Query(20, ge=1, le=500),
     admin: User = Depends(get_admin_user),
@@ -2232,6 +2234,10 @@ async def admin_list_generation_records(
         query = query.where(GenerationRecord.engine_id == engine_id)
     if include_media_references is not None:
         query = query.where(GenerationRecord.include_media_references.is_(include_media_references))
+    if start_date:
+        query = query.where(GenerationRecord.created_at >= datetime.strptime(start_date, "%Y-%m-%d").replace(tzinfo=CST))
+    if end_date:
+        query = query.where(GenerationRecord.created_at < (datetime.strptime(end_date, "%Y-%m-%d") + timedelta(days=1)).replace(tzinfo=CST))
 
     # Count total
     count_query = (
@@ -2247,6 +2253,10 @@ async def admin_list_generation_records(
         count_query = count_query.where(GenerationRecord.engine_id == engine_id)
     if include_media_references is not None:
         count_query = count_query.where(GenerationRecord.include_media_references.is_(include_media_references))
+    if start_date:
+        count_query = count_query.where(GenerationRecord.created_at >= datetime.strptime(start_date, "%Y-%m-%d").replace(tzinfo=CST))
+    if end_date:
+        count_query = count_query.where(GenerationRecord.created_at < (datetime.strptime(end_date, "%Y-%m-%d") + timedelta(days=1)).replace(tzinfo=CST))
     total_result = await db.execute(count_query)
     total = total_result.scalar() or 0