feat(models,utils): implement data models and utility classes

This commit is contained in:
wkc
2026-03-03 09:02:33 +08:00
parent aa17a14c4e
commit e6bc2d64dd
9 changed files with 357 additions and 0 deletions

View File

@@ -0,0 +1,187 @@
from pydantic import BaseModel, Field
from typing import Optional, List, Dict, Any
# ==================== Token相关模型 ====================
class TokenData(BaseModel):
"""Token数据"""
token: str = Field(..., description="token")
projectId: int = Field(..., description="见知项目Id")
projectNo: str = Field(..., description="项目编号")
entityName: str = Field(..., description="项目名称")
analysisType: int = Field(0, description="分析类型")
class GetTokenResponse(BaseModel):
"""获取Token响应"""
code: str = Field("200", description="返回码")
data: Optional[TokenData] = Field(None, description="返回数据")
message: str = Field("create.token.success", description="返回消息")
status: str = Field("200", description="状态")
successResponse: bool = Field(True, description="是否成功响应")
# ==================== 文件上传相关模型 ====================
class AccountInfo(BaseModel):
"""账户信息"""
bank: str = Field(..., description="银行")
accountName: str = Field(..., description="账户名称")
accountNo: str = Field(..., description="账号")
currency: str = Field(..., description="币种")
class UploadLogItem(BaseModel):
"""上传日志项"""
accountNoList: List[str] = Field(default=[], description="账号列表")
bankName: str = Field(..., description="银行名称")
dataTypeInfo: List[str] = Field(default=[], description="数据类型信息")
downloadFileName: str = Field(..., description="下载文件名")
enterpriseNameList: List[str] = Field(default=[], description="企业名称列表")
filePackageId: str = Field(..., description="文件包ID")
fileSize: int = Field(..., description="文件大小")
fileUploadBy: int = Field(..., description="上传者ID")
fileUploadByUserName: str = Field(..., description="上传者用户名")
fileUploadTime: str = Field(..., description="上传时间")
leId: int = Field(..., description="企业ID")
logId: int = Field(..., description="日志ID")
logMeta: str = Field(..., description="日志元数据")
logType: str = Field(..., description="日志类型")
loginLeId: int = Field(..., description="登录企业ID")
realBankName: str = Field(..., description="真实银行名称")
rows: int = Field(0, description="行数")
source: str = Field(..., description="来源")
status: int = Field(-5, description="状态值")
templateName: str = Field(..., description="模板名称")
totalRecords: int = Field(0, description="总记录数")
trxDateEndId: int = Field(..., description="交易结束日期ID")
trxDateStartId: int = Field(..., description="交易开始日期ID")
uploadFileName: str = Field(..., description="上传文件名")
uploadStatusDesc: str = Field(..., description="上传状态描述")
class UploadFileResponse(BaseModel):
"""上传文件响应"""
code: str = Field("200", description="返回码")
data: Optional[Dict[str, Any]] = Field(None, description="返回数据")
status: str = Field("200", description="状态")
successResponse: bool = Field(True, description="是否成功响应")
# ==================== 检查解析状态相关模型 ====================
class PendingItem(BaseModel):
"""待处理项"""
accountNoList: List[str] = Field(default=[], description="账号列表")
bankName: str = Field(..., description="银行名称")
dataTypeInfo: List[str] = Field(default=[], description="数据类型信息")
downloadFileName: str = Field(..., description="下载文件名")
enterpriseNameList: List[str] = Field(default=[], description="企业名称列表")
filePackageId: str = Field(..., description="文件包ID")
fileSize: int = Field(..., description="文件大小")
fileUploadBy: int = Field(..., description="上传者ID")
fileUploadByUserName: str = Field(..., description="上传者用户名")
fileUploadTime: str = Field(..., description="上传时间")
isSplit: int = Field(0, description="是否分割")
leId: int = Field(..., description="企业ID")
logId: int = Field(..., description="日志ID")
logMeta: str = Field(..., description="日志元数据")
logType: str = Field(..., description="日志类型")
loginLeId: int = Field(..., description="登录企业ID")
lostHeader: List[str] = Field(default=[], description="丢失的头部")
realBankName: str = Field(..., description="真实银行名称")
rows: int = Field(0, description="行数")
source: str = Field(..., description="来源")
status: int = Field(-5, description="状态值")
templateName: str = Field(..., description="模板名称")
totalRecords: int = Field(0, description="总记录数")
trxDateEndId: int = Field(..., description="交易结束日期ID")
trxDateStartId: int = Field(..., description="交易开始日期ID")
uploadFileName: str = Field(..., description="上传文件名")
uploadStatusDesc: str = Field(..., description="上传状态描述")
class CheckParseStatusResponse(BaseModel):
"""检查解析状态响应"""
code: str = Field("200", description="返回码")
data: Optional[Dict[str, Any]] = Field(None, description="返回数据包含parsing和pendingList")
status: str = Field("200", description="状态")
successResponse: bool = Field(True, description="是否成功响应")
# ==================== 银行流水相关模型 ====================
class BankStatementItem(BaseModel):
"""银行流水项"""
accountId: int = Field(0, description="账号ID")
accountMaskNo: str = Field(..., description="账号")
accountingDate: str = Field(..., description="记账日期")
accountingDateId: int = Field(..., description="记账日期ID")
archivingFlag: int = Field(0, description="归档标志")
attachments: int = Field(0, description="附件数")
balanceAmount: float = Field(..., description="余额")
bank: str = Field(..., description="银行")
bankComments: str = Field("", description="银行注释")
bankStatementId: int = Field(..., description="流水ID")
bankTrxNumber: str = Field(..., description="银行交易号")
batchId: int = Field(..., description="批次ID")
cashType: str = Field("1", description="现金类型")
commentsNum: int = Field(0, description="评论数")
crAmount: float = Field(0, description="贷方金额")
cretNo: str = Field(..., description="证件号")
currency: str = Field("CNY", description="币种")
customerAccountMaskNo: str = Field(..., description="客户账号")
customerBank: str = Field("", description="客户银行")
customerId: int = Field(-1, description="客户ID")
customerName: str = Field(..., description="客户名称")
customerReference: str = Field("", description="客户参考")
downPaymentFlag: int = Field(0, description="首付标志")
drAmount: float = Field(0, description="借方金额")
exceptionType: str = Field("", description="异常类型")
groupId: int = Field(0, description="项目ID")
internalFlag: int = Field(0, description="内部标志")
leId: int = Field(..., description="企业ID")
leName: str = Field(..., description="企业名称")
overrideBsId: int = Field(0, description="覆盖流水ID")
paymentMethod: str = Field("", description="支付方式")
sourceCatalogId: int = Field(0, description="来源目录ID")
split: int = Field(0, description="分割")
subBankstatementId: int = Field(0, description="子流水ID")
toDoFlag: int = Field(0, description="待办标志")
transAmount: float = Field(..., description="交易金额")
transFlag: str = Field("P", description="交易标志")
transTypeId: int = Field(0, description="交易类型ID")
transformAmount: int = Field(0, description="转换金额")
transformCrAmount: int = Field(0, description="转换贷方金额")
transformDrAmount: int = Field(0, description="转换借方金额")
transfromBalanceAmount: int = Field(0, description="转换余额")
trxBalance: int = Field(0, description="交易余额")
trxDate: str = Field(..., description="交易日期")
userMemo: str = Field(..., description="用户备注")
class GetBankStatementResponse(BaseModel):
"""获取银行流水响应"""
code: str = Field("200", description="返回码")
data: Optional[Dict[str, Any]] = Field(None, description="返回数据包含bankStatementList和totalCount")
status: str = Field("200", description="状态")
successResponse: bool = Field(True, description="是否成功响应")
# ==================== 其他响应模型 ====================
class FetchInnerFlowResponse(BaseModel):
"""拉取行内流水响应"""
code: str = Field("200", description="返回码")
data: Optional[Dict[str, Any]] = Field(None, description="返回数据")
status: str = Field("200", description="状态")
successResponse: bool = Field(True, description="是否成功响应")
class DeleteFilesResponse(BaseModel):
"""删除文件响应"""
code: str = Field("200", description="返回码")
data: Optional[Dict[str, str]] = Field(None, description="返回数据")
status: str = Field("200", description="状态")
successResponse: bool = Field(True, description="是否成功响应")