1

2026-08-10 16:29:26 +08:00
parent e364c46ce4
commit d1e8eb7316
18 changed files with 1335 additions and 819 deletions
@@ -0,0 +1,128 @@
+import logging
+from datetime import datetime, timezone
+
+from fastapi import HTTPException, status
+from sqlalchemy import select, update
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from app.models.invoice_header import InvoiceHeader
+from app.schemas.invoice import InvoiceHeaderCreate, InvoiceHeaderUpdate
+from app.utils.id import generate_id
+
+logger = logging.getLogger("videogen")
+
+
+async def create_header(db: AsyncSession, user_id: str, data: InvoiceHeaderCreate) -> InvoiceHeader:
+    """创建发票抬头。"""
+    now = datetime.now(timezone.utc)
+    header = InvoiceHeader(
+        id=generate_id(),
+        user_id=user_id,
+        type=data.type,
+        name=data.name,
+        tax_no=data.tax_no,
+        register_address=data.register_address,
+        register_phone=data.register_phone,
+        bank_name=data.bank_name,
+        bank_account=data.bank_account,
+        email=data.email,
+        is_default=data.is_default,
+        created_at=now,
+        updated_at=now,
+    )
+
+    # 如果设为默认，先将其他抬头取消默认
+    if data.is_default:
+        await db.execute(
+            update(InvoiceHeader)
+            .where(InvoiceHeader.user_id == user_id)
+            .values(is_default=False, updated_at=now)
+        )
+
+    db.add(header)
+    await db.flush()
+    return header
+
+
+async def get_user_headers(db: AsyncSession, user_id: str) -> list[InvoiceHeader]:
+    """获取用户的所有发票抬头。"""
+    result = await db.execute(
+        select(InvoiceHeader)
+        .where(InvoiceHeader.user_id == user_id)
+        .order_by(InvoiceHeader.is_default.desc(), InvoiceHeader.created_at.desc())
+    )
+    return list(result.scalars().all())
+
+
+async def get_header_by_id(db: AsyncSession, header_id: str, user_id: str) -> InvoiceHeader | None:
+    """获取指定发票抬头（仅限本人）。"""
+    result = await db.execute(
+        select(InvoiceHeader).where(
+            InvoiceHeader.id == header_id,
+            InvoiceHeader.user_id == user_id,
+        )
+    )
+    return result.scalar_one_or_none()
+
+
+async def update_header(
+    db: AsyncSession, header_id: str, user_id: str, data: InvoiceHeaderUpdate
+) -> InvoiceHeader:
+    """更新发票抬头。"""
+    header = await get_header_by_id(db, header_id, user_id)
+    if not header:
+        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="发票抬头不存在")
+
+    update_data = {}
+    for field, value in data.model_dump(exclude_unset=True).items():
+        update_data[field] = value
+
+    if update_data:
+        update_data["updated_at"] = datetime.now(timezone.utc)
+        await db.execute(
+            update(InvoiceHeader)
+            .where(InvoiceHeader.id == header_id)
+            .values(**update_data)
+        )
+
+    # 如果设为默认，先将其他抬头取消默认
+    if data.is_default:
+        now = datetime.now(timezone.utc)
+        await db.execute(
+            update(InvoiceHeader)
+            .where(InvoiceHeader.user_id == user_id, InvoiceHeader.id != header_id)
+            .values(is_default=False, updated_at=now)
+        )
+
+    await db.refresh(header)
+    return header
+
+
+async def delete_header(db: AsyncSession, header_id: str, user_id: str) -> None:
+    """删除发票抬头。"""
+    header = await get_header_by_id(db, header_id, user_id)
+    if not header:
+        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="发票抬头不存在")
+
+    await db.delete(header)
+    await db.flush()
+
+
+async def set_default_header(db: AsyncSession, header_id: str, user_id: str) -> InvoiceHeader:
+    """设置默认发票抬头。"""
+    header = await get_header_by_id(db, header_id, user_id)
+    if not header:
+        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="发票抬头不存在")
+
+    now = datetime.now(timezone.utc)
+    # 先取消其他默认
+    await db.execute(
+        update(InvoiceHeader)
+        .where(InvoiceHeader.user_id == user_id, InvoiceHeader.id != header_id)
+        .values(is_default=False, updated_at=now)
+    )
+    # 设置当前为默认
+    header.is_default = True
+    header.updated_at = now
+    await db.flush()
+    return header