FastapiAdmin/backend/app/api/v1/module_platform/invoice/service.py

"""发票管理 Service"""
import random
from datetime import date, datetime, timedelta

from app.api.v1.module_platform.order.crud import OrderCRUD
from app.core.base_schema import AuthSchema
from app.core.exceptions import CustomException
from app.core.logger import logger

from .crud import InvoiceCRUD
from .schema import (
    InvoiceApplySchema,
    InvoiceCreateSchema,
    InvoiceOutSchema,
    InvoiceQueryParam,
    InvoiceUpdateSchema,
)


def _generate_invoice_no() -> str:
    """生成发票编号"""
    today = date.today().strftime("%Y%m%d")
    suffix = str(random.randint(100000, 999999))
    return f"INV{today}{suffix}"


class InvoiceTenantService:
    """租户端发票服务"""

    @staticmethod
    async def apply(auth: AuthSchema, data: InvoiceApplySchema, tenant_id: int) -> InvoiceOutSchema:
        # 校验：专票必填字段
        if data.invoice_type == "vat_special":
            if not data.tax_no:
                raise CustomException(msg="增值税专用发票必须填写纳税人识别号")
            if not data.bank_info:
                raise CustomException(msg="增值税专用发票必须填写开户行及账号")
            if not data.address_info:
                raise CustomException(msg="增值税专用发票必须填写注册地址及电话")

        # 校验：订单存在且已支付
        order = await OrderCRUD(auth).get(id=data.order_id)
        if not order:
            raise CustomException(msg="订单不存在")
        if order.status != 1:
            raise CustomException(msg="仅已支付订单可申请开票")

        # 校验：30 天内
        if order.created_time and datetime.now() - order.created_time > timedelta(days=30):
            raise CustomException(msg="订单支付超过 30 天，不可申请开票")

        crud = InvoiceCRUD(auth)
        existing = await crud.get_by_order_id(data.order_id)
        if existing:
            raise CustomException(msg="该订单已申请过发票")

        tax_rate = 0  # 默认税率0%，未来可从套餐配置读取
        tax_amount = int(order.amount * tax_rate / 100)
        invoice = await crud.create(
            InvoiceCreateSchema(
                invoice_no=_generate_invoice_no(),
                order_id=data.order_id,
                tenant_id=tenant_id,
                invoice_type=data.invoice_type,
                title=data.title,
                tax_no=data.tax_no,
                bank_info=data.bank_info,
                address_info=data.address_info,
                amount=order.amount,
                tax_amount=tax_amount,
                description=data.description,
            )
        )
        logger.info(f"发票申请成功: invoice_no={invoice.invoice_no}, order_id={data.order_id}")
        return InvoiceOutSchema.model_validate(invoice)

    @staticmethod
    async def list_my(auth: AuthSchema, tenant_id: int, params: InvoiceQueryParam) -> dict:
        search: dict = {"tenant_id": tenant_id}
        if params.invoice_type:
            search["invoice_type"] = params.invoice_type
        if params.status is not None:
            search["status"] = params.status
        return await InvoiceCRUD(auth).page(
            offset=(params.page_no - 1) * params.page_size,
            limit=params.page_size,
            order_by=[{"created_time": "desc"}],
            search=search,
            out_schema=InvoiceOutSchema,
        )


class InvoicePlatformService:
    """平台端发票服务"""

    @staticmethod
    async def list_all(auth: AuthSchema, params: InvoiceQueryParam) -> dict:
        search: dict = {}
        if params.invoice_type:
            search["invoice_type"] = params.invoice_type
        if params.status is not None:
            search["status"] = params.status
        if params.tenant_id:
            search["tenant_id"] = params.tenant_id
        return await InvoiceCRUD(auth).page(
            offset=(params.page_no - 1) * params.page_size,
            limit=params.page_size,
            order_by=[{"created_time": "desc"}],
            search=search,
            out_schema=InvoiceOutSchema,
        )

    @staticmethod
    async def issue(auth: AuthSchema, invoice_id: int, pdf_url: str, api_response: str) -> InvoiceOutSchema:
        crud = InvoiceCRUD(auth)
        invoice = await crud.get(id=invoice_id)
        if not invoice:
            raise CustomException(msg="发票不存在")
        if invoice.status != 0:
            raise CustomException(msg="仅待开票状态可操作")

        # 第三方开票 API 调用（暂为存根，对接百望云/票通时替换）
        api_ok, pdf_url_result, api_response_result = await InvoicePlatformService._call_third_party_api(invoice)

        if not api_ok:
            invoice = await crud.update(invoice_id, InvoiceUpdateSchema(status=2, api_response=api_response_result))
            logger.warning(f"第三方开票API调用失败: invoice_no={invoice.invoice_no}, resp={api_response_result}")
            raise CustomException(msg=f"开票失败: {api_response_result}")

        invoice = await crud.update(
            invoice_id,
            InvoiceUpdateSchema(
                status=1,
                pdf_url=pdf_url or pdf_url_result,
                api_response=api_response or api_response_result,
            ),
        )
        logger.info(f"发票开具成功: invoice_no={invoice.invoice_no}")
        return InvoiceOutSchema.model_validate(invoice)

    @staticmethod
    async def _call_third_party_api(invoice) -> tuple[bool, str, str]:
        """第三方开票 API 存根

        对接百望云/票通等电子发票平台时替换此方法。
        开发模式下直接模拟成功，返回 PDF 路径。

        返回:
            (success: bool, pdf_url: str, api_response: str)
        """
        pdf_url = f"/static/invoice/{invoice.tenant_id}/{invoice.invoice_no}.pdf"
        api_response = '{"code":"SUCCESS","msg":"模拟开票成功","invoice_no":"' + invoice.invoice_no + '"}'
        return True, pdf_url, api_response

    @staticmethod
    async def void(auth: AuthSchema, invoice_id: int, data) -> InvoiceOutSchema:
        crud = InvoiceCRUD(auth)
        invoice = await crud.get(id=invoice_id)
        if not invoice:
            raise CustomException(msg="发票不存在")
        if invoice.status != 1:
            code_text = {0: "待开票", 2: "开票失败", 3: "已作废"}.get(invoice.status, "未知")
            raise CustomException(msg=f"仅已开票状态可作废，当前状态: {code_text}")
        description = getattr(data, "description", "") or ""
        invoice = await crud.update(invoice_id, InvoiceUpdateSchema(status=3, description=description))
        logger.info(f"发票作废: invoice_no={invoice.invoice_no}")
        return InvoiceOutSchema.model_validate(invoice)