100 lines
3.2 KiB
Python
100 lines
3.2 KiB
Python
from fastapi import APIRouter, BackgroundTasks, UploadFile, File, Form
|
||
from models.request import (
|
||
GetTokenRequest,
|
||
FetchInnerFlowRequest,
|
||
CheckParseStatusRequest,
|
||
GetBankStatementRequest,
|
||
DeleteFilesRequest,
|
||
)
|
||
from services.token_service import TokenService
|
||
from services.file_service import FileService
|
||
from services.statement_service import StatementService
|
||
from utils.error_simulator import ErrorSimulator
|
||
from typing import List
|
||
|
||
# 创建路由器
|
||
router = APIRouter()
|
||
|
||
# 初始化服务实例
|
||
token_service = TokenService()
|
||
file_service = FileService()
|
||
statement_service = StatementService()
|
||
|
||
|
||
# ==================== 接口1：获取Token ====================
|
||
@router.post("/account/common/getToken")
|
||
async def get_token(request: GetTokenRequest):
|
||
"""创建项目并获取访问Token
|
||
|
||
如果 projectNo 包含 error_XXXX 标记，将返回对应的错误响应
|
||
"""
|
||
# 检测错误标记
|
||
error_code = ErrorSimulator.detect_error_marker(request.projectNo)
|
||
if error_code:
|
||
return ErrorSimulator.build_error_response(error_code)
|
||
|
||
# 正常流程
|
||
return token_service.create_token(request)
|
||
|
||
|
||
# ==================== 接口2：上传文件 ====================
|
||
@router.post("/watson/api/project/remoteUploadSplitFile")
|
||
async def upload_file(
|
||
background_tasks: BackgroundTasks,
|
||
groupId: int = Form(..., description="项目ID"),
|
||
file: UploadFile = File(..., description="流水文件"),
|
||
):
|
||
"""上传流水文件
|
||
|
||
文件将立即返回，并在后台延迟4秒完成解析
|
||
"""
|
||
return await file_service.upload_file(groupId, file, background_tasks)
|
||
|
||
|
||
# ==================== 接口3：拉取行内流水 ====================
|
||
@router.post("/watson/api/project/getJZFileOrZjrcuFile")
|
||
async def fetch_inner_flow(request: FetchInnerFlowRequest):
|
||
"""拉取行内流水
|
||
|
||
如果 customerNo 包含 error_XXXX 标记，将返回对应的错误响应
|
||
"""
|
||
# 检测错误标记
|
||
error_code = ErrorSimulator.detect_error_marker(request.customerNo)
|
||
if error_code:
|
||
return ErrorSimulator.build_error_response(error_code)
|
||
|
||
# 正常流程
|
||
return file_service.fetch_inner_flow(request)
|
||
|
||
|
||
# ==================== 接口4：检查文件解析状态 ====================
|
||
@router.post("/watson/api/project/upload/getpendings")
|
||
async def check_parse_status(request: CheckParseStatusRequest):
|
||
"""检查文件解析状态
|
||
|
||
返回文件是否还在解析中（parsing字段）
|
||
"""
|
||
return file_service.check_parse_status(
|
||
request.groupId, request.inprogressList
|
||
)
|
||
|
||
|
||
# ==================== 接口5：删除文件 ====================
|
||
@router.post("/watson/api/project/batchDeleteUploadFile")
|
||
async def delete_files(request: DeleteFilesRequest):
|
||
"""批量删除上传的文件
|
||
|
||
根据logIds列表删除对应的文件记录
|
||
"""
|
||
return file_service.delete_files(request.groupId, request.logIds, request.userId)
|
||
|
||
|
||
# ==================== 接口6：获取银行流水 ====================
|
||
@router.post("/watson/api/project/getBSByLogId")
|
||
async def get_bank_statement(request: GetBankStatementRequest):
|
||
"""获取银行流水列表
|
||
|
||
支持分页查询（pageNow, pageSize）
|
||
"""
|
||
return statement_service.get_bank_statement(request)
|