You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
89 lines
4.0 KiB
Python
89 lines
4.0 KiB
Python
from dataclasses import dataclass
|
|
|
|
from fastapi import Query
|
|
from pydantic import BaseModel, ConfigDict, Field
|
|
|
|
from app.common.enums import InvoiceTypeEnum, QueueEnum
|
|
from app.core.base_params import BaseQueryParam
|
|
from app.core.base_schema import BaseSchema, TenantBySchema, UserBySchema
|
|
|
|
|
|
class InvoiceCreateSchema(BaseModel):
|
|
"""创建发票（内部使用）"""
|
|
|
|
invoice_no: str = Field(..., description="发票号码")
|
|
order_id: int = Field(..., description="关联订单 ID")
|
|
tenant_id: int = Field(..., description="租户 ID")
|
|
invoice_type: InvoiceTypeEnum = Field(..., description="发票类型")
|
|
title: str = Field(..., max_length=200, description="发票抬头")
|
|
tax_no: str | None = Field(default=None, max_length=50, description="纳税人识别号")
|
|
bank_info: str | None = Field(default=None, description="开户行及账号")
|
|
address_info: str | None = Field(default=None, description="注册地址及电话")
|
|
amount: int = Field(..., ge=0, description="发票金额(分)")
|
|
tax_amount: int = Field(default=0, ge=0, description="税额(分)")
|
|
status: int = Field(default=0, ge=0, le=3, description="状态(0:待开票 1:已开票 2:开票失败 3:已作废)")
|
|
description: str | None = Field(default=None, description="备注")
|
|
|
|
|
|
class InvoiceUpdateSchema(BaseModel):
|
|
"""更新发票（内部使用）"""
|
|
|
|
status: int | None = Field(default=None, ge=0, le=3, description="状态(0:待开票 1:已开票 2:开票失败 3:已作废)")
|
|
pdf_url: str | None = Field(default=None, max_length=500, description="发票 PDF 下载地址")
|
|
oss_license_pdf_url: str | None = Field(default=None, max_length=500, description="开源授权函 PDF 下载地址")
|
|
api_response: str | None = Field(default=None, description="第三方 API 响应")
|
|
description: str | None = Field(default=None, description="备注")
|
|
|
|
|
|
class InvoiceApplySchema(BaseModel):
|
|
"""申请开票"""
|
|
|
|
order_id: int = Field(..., description="订单 ID")
|
|
invoice_type: InvoiceTypeEnum = Field(..., description="发票类型")
|
|
title: str = Field(..., max_length=200, description="发票抬头")
|
|
tax_no: str | None = Field(default=None, max_length=50, description="纳税人识别号")
|
|
bank_info: str | None = Field(default=None, description="开户行及账号")
|
|
address_info: str | None = Field(default=None, description="注册地址及电话")
|
|
description: str | None = Field(default=None, description="备注")
|
|
|
|
|
|
class InvoiceIssueSchema(BaseModel):
|
|
"""超管开票"""
|
|
|
|
api_response: str | None = Field(default=None, description="第三方 API 响应（手动填入）")
|
|
pdf_url: str | None = Field(default=None, max_length=500, description="发票 PDF 下载地址")
|
|
oss_license_pdf_url: str | None = Field(default=None, max_length=500, description="开源授权函 PDF 下载地址")
|
|
|
|
|
|
class InvoiceVoidSchema(BaseModel):
|
|
"""作废发票"""
|
|
|
|
description: str | None = Field(default=None, description="作废原因")
|
|
|
|
|
|
class InvoiceOutSchema(InvoiceCreateSchema, BaseSchema, UserBySchema, TenantBySchema):
|
|
"""发票响应"""
|
|
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
pdf_url: str | None = Field(default=None, description="发票 PDF 下载地址")
|
|
oss_license_pdf_url: str | None = Field(default=None, description="开源授权函 PDF 下载地址")
|
|
api_response: str | None = Field(default=None, description="第三方 API 响应")
|
|
|
|
|
|
@dataclass
|
|
class InvoiceQueryParam(BaseQueryParam):
|
|
"""发票查询参数"""
|
|
|
|
invoice_type: InvoiceTypeEnum | None = Query(None, description="发票类型")
|
|
status: int | None = Query(None, description="状态")
|
|
tenant_id: int | None = Query(None, description="租户ID")
|
|
|
|
def __post_init__(self) -> None:
|
|
if self.invoice_type:
|
|
self.invoice_type = (QueueEnum.eq.value, self.invoice_type)
|
|
if self.status is not None:
|
|
self.status = (QueueEnum.eq.value, self.status)
|
|
if self.tenant_id is not None:
|
|
self.tenant_id = (QueueEnum.eq.value, self.tenant_id)
|