1

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

@@ -36,6 +36,7 @@ from app.api.v1.private_portrait import router as private_portrait_router
 from app.api.v1.private_portrait_virtual import router as private_portrait_virtual_router
 from app.api.v1.upload_resource import router as upload_resource_router
 from app.api.v1.invoices import router as invoices_router
+from app.api.v1.invoice_headers import router as invoice_headers_router
 
 api_router = APIRouter()
 api_router.include_router(auth_router)
@@ -74,3 +75,4 @@ api_router.include_router(private_portrait_router)
 api_router.include_router(private_portrait_virtual_router)
 api_router.include_router(upload_resource_router)
 api_router.include_router(invoices_router)
+api_router.include_router(invoice_headers_router)

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
 

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

@@ -0,0 +1,96 @@
+import logging
+
+from fastapi import APIRouter, Depends, HTTPException, status
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from app.dependencies import get_db, get_current_user
+from app.models.user import User
+from app.schemas.invoice import InvoiceHeaderCreate, InvoiceHeaderOut, InvoiceHeaderUpdate
+from app.services.invoice_header import (
+    create_header,
+    delete_header,
+    get_user_headers,
+    set_default_header,
+    update_header,
+)
+
+logger = logging.getLogger("videogen")
+
+router = APIRouter(prefix="/invoice-headers", tags=["invoice-headers"])
+
+
+def _header_to_out(header) -> dict:
+    return {
+        "id": header.id,
+        "userId": header.user_id,
+        "type": header.type,
+        "name": header.name,
+        "taxNo": header.tax_no,
+        "registerAddress": header.register_address,
+        "registerPhone": header.register_phone,
+        "bankName": header.bank_name,
+        "bankAccount": header.bank_account,
+        "email": header.email,
+        "isDefault": header.is_default,
+        "createdAt": header.created_at.isoformat() if header.created_at else None,
+        "updatedAt": header.updated_at.isoformat() if header.updated_at else None,
+    }
+
+
+@router.post("", response_model=InvoiceHeaderOut)
+async def create_invoice_header(
+    req: InvoiceHeaderCreate,
+    current_user: User = Depends(get_current_user),
+    db: AsyncSession = Depends(get_db),
+):
+    """创建发票抬头。"""
+    header = await create_header(db, current_user.id, req)
+    await db.commit()
+    return _header_to_out(header)
+
+
+@router.get("")
+async def list_invoice_headers(
+    current_user: User = Depends(get_current_user),
+    db: AsyncSession = Depends(get_db),
+):
+    """获取当前用户的所有发票抬头。"""
+    headers = await get_user_headers(db, current_user.id)
+    return {"items": [_header_to_out(h) for h in headers]}
+
+
+@router.put("/{header_id}", response_model=InvoiceHeaderOut)
+async def update_invoice_header(
+    header_id: str,
+    req: InvoiceHeaderUpdate,
+    current_user: User = Depends(get_current_user),
+    db: AsyncSession = Depends(get_db),
+):
+    """更新发票抬头。"""
+    header = await update_header(db, header_id, current_user.id, req)
+    await db.commit()
+    return _header_to_out(header)
+
+
+@router.delete("/{header_id}")
+async def delete_invoice_header(
+    header_id: str,
+    current_user: User = Depends(get_current_user),
+    db: AsyncSession = Depends(get_db),
+):
+    """删除发票抬头。"""
+    await delete_header(db, header_id, current_user.id)
+    await db.commit()
+    return {"success": True}
+
+
+@router.put("/{header_id}/set-default", response_model=InvoiceHeaderOut)
+async def set_default_invoice_header(
+    header_id: str,
+    current_user: User = Depends(get_current_user),
+    db: AsyncSession = Depends(get_db),
+):
+    """设置默认发票抬头。"""
+    header = await set_default_header(db, header_id, current_user.id)
+    await db.commit()
+    return _header_to_out(header)