Compare commits
10 Commits
afe1ae8353
...
92a57fa5d0
| Author | SHA1 | Date | |
|---|---|---|---|
| 92a57fa5d0 | |||
| d921a2e7ba | |||
| a54693a7bf | |||
| 21c87d49d7 | |||
| 3d7021dc6b | |||
| d6fcd38cd5 | |||
| 5c55f4acde | |||
| 7d4170a3c3 | |||
| f6af4eace8 | |||
| 8edb6badc4 |
171
.env.dev
Normal file
171
.env.dev
Normal file
@ -0,0 +1,171 @@
|
|||||||
|
# -------- 应用配置 --------
|
||||||
|
# 应用运行环境
|
||||||
|
APP_ENV = 'dev'
|
||||||
|
|
||||||
|
# 应用名称
|
||||||
|
APP_NAME = 'Cutoms-System'
|
||||||
|
|
||||||
|
# 应用代理路径
|
||||||
|
APP_ROOT_PATH = ''
|
||||||
|
|
||||||
|
# 应用主机
|
||||||
|
APP_HOST = '0.0.0.0'
|
||||||
|
|
||||||
|
# 应用端口
|
||||||
|
APP_PORT = 8082
|
||||||
|
|
||||||
|
# 应用版本
|
||||||
|
APP_VERSION= '1.0.0'
|
||||||
|
|
||||||
|
# 应用是否开启热重载
|
||||||
|
APP_RELOAD = true
|
||||||
|
|
||||||
|
# 应用是否开启IP归属区域查询
|
||||||
|
APP_IP_LOCATION_QUERY = false
|
||||||
|
|
||||||
|
# 应用是否允许账号同时登录
|
||||||
|
APP_SAME_TIME_LOGIN = true
|
||||||
|
|
||||||
|
# -------JWT配置------------
|
||||||
|
# JWT 签名密钥
|
||||||
|
JWT_SECRET_KEY=b01c66dc2c58dc6a0aabfe2144256be36226de378bf87f72c0c795dda67f4d55
|
||||||
|
|
||||||
|
# JWT 签名算法
|
||||||
|
JWT_ALGORITHM=HS256
|
||||||
|
|
||||||
|
# JWT 盐值
|
||||||
|
JWT_SALT=jwt_salt
|
||||||
|
|
||||||
|
# JWT 令牌有效期(分钟)
|
||||||
|
JWT_EXPIRE_MINUTES=1440
|
||||||
|
|
||||||
|
# JWT 令牌在 Redis 中的缓存有效期(分钟)
|
||||||
|
JWT_REDIS_EXPIRE_MINUTES=30
|
||||||
|
|
||||||
|
# -------- 数据库配置 --------
|
||||||
|
# 数据库类型,默认为'mysql'
|
||||||
|
DB_TYPE = 'mysql'
|
||||||
|
|
||||||
|
# 数据库主机
|
||||||
|
DB_HOST = '127.0.0.1'
|
||||||
|
|
||||||
|
# 数据库端口
|
||||||
|
DB_PORT = 3306
|
||||||
|
|
||||||
|
# 数据库用户名
|
||||||
|
DB_USERNAME = 'root'
|
||||||
|
|
||||||
|
# 数据库密码
|
||||||
|
DB_PASSWORD = 'mysqlroot'
|
||||||
|
|
||||||
|
# 数据库名称
|
||||||
|
DB_DATABASE = 'cutoms_system'
|
||||||
|
|
||||||
|
# 是否开启日志
|
||||||
|
DB_ECHO = true
|
||||||
|
|
||||||
|
# 数据库日志级别,默认为 10(DEBUG)
|
||||||
|
DB_LOG_LEVEL = 10
|
||||||
|
|
||||||
|
# 允许溢出连接池大小的最大连接数
|
||||||
|
DB_MAX_OVERFLOW = 10
|
||||||
|
|
||||||
|
# 连接池大小,0表示连接数无限制
|
||||||
|
DB_POOL_SIZE = 50
|
||||||
|
|
||||||
|
# 连接回收时间(单位:秒)
|
||||||
|
DB_POOL_RECYCLE = 3600
|
||||||
|
|
||||||
|
# 连接池中没有线程可用时,最多等待的时间(单位:秒)
|
||||||
|
DB_POOL_TIMEOUT = 30
|
||||||
|
|
||||||
|
# -------- Redis配置 --------
|
||||||
|
|
||||||
|
# Redis主机
|
||||||
|
REDIS_HOST = '127.0.0.1'
|
||||||
|
|
||||||
|
# Redis端口
|
||||||
|
REDIS_PORT = 6379
|
||||||
|
|
||||||
|
# Redis用户名
|
||||||
|
REDIS_USERNAME = ''
|
||||||
|
|
||||||
|
# Redis密码
|
||||||
|
REDIS_PASSWORD = ''
|
||||||
|
|
||||||
|
# Redis数据库
|
||||||
|
REDIS_DATABASE = 2
|
||||||
|
|
||||||
|
# ======================
|
||||||
|
# 上传配置
|
||||||
|
# ======================
|
||||||
|
|
||||||
|
# 文件上传的 URL 前缀,默认为 '/profile'。
|
||||||
|
# 例如:`/profile/example.jpg`。
|
||||||
|
UPLOAD_PREFIX=/profile
|
||||||
|
|
||||||
|
# 文件上传的存储路径,默认为 'data/upload_path'。
|
||||||
|
# 上传的文件将存储在此目录中,如果目录不存在,会自动创建。
|
||||||
|
UPLOAD_PATH=data/upload_path
|
||||||
|
|
||||||
|
# 上传机器的标识,默认为 'A'。
|
||||||
|
# 用于区分不同的上传机器或节点,在多机部署时可以使用此字段。
|
||||||
|
UPLOAD_MACHINE=A
|
||||||
|
|
||||||
|
# 默认允许上传的文件扩展名列表,使用逗号分隔。
|
||||||
|
# 包含常见的图片、文档、压缩文件、视频和 PDF 格式。
|
||||||
|
# 可以根据需求扩展或修改此列表。
|
||||||
|
DEFAULT_ALLOWED_EXTENSION=bmp,gif,jpg,jpeg,png,doc,docx,xls,xlsx,ppt,pptx,html,htm,txt,rar,zip,gz,bz2,mp4,avi,rmvb,pdf
|
||||||
|
|
||||||
|
# 文件下载的存储路径,默认为 'data/download_path'。
|
||||||
|
# 下载的文件将存储在此目录中,如果目录不存在,会自动创建。
|
||||||
|
DOWNLOAD_PATH=data/download_path
|
||||||
|
|
||||||
|
|
||||||
|
# ======================
|
||||||
|
# 邮件配置
|
||||||
|
# ======================
|
||||||
|
|
||||||
|
# 邮件发送者的用户名,默认为空。
|
||||||
|
EMAIL_USERNAME=
|
||||||
|
|
||||||
|
# 邮件发送者的密码,默认为空。
|
||||||
|
EMAIL_PASSWORD=
|
||||||
|
|
||||||
|
# 邮件服务器地址,默认为 "smtp.qq.com"。
|
||||||
|
# 如果是其他邮件服务商,请修改为对应的 SMTP 服务器地址。
|
||||||
|
EMIAL_HOST=smtp.qq.com
|
||||||
|
|
||||||
|
# 邮件服务器端口,默认为 465。
|
||||||
|
# 如果是其他邮件服务商,请根据其要求修改端口号。
|
||||||
|
EMAIL_PORT=587
|
||||||
|
|
||||||
|
|
||||||
|
# ======================
|
||||||
|
# 地图配置
|
||||||
|
# ======================
|
||||||
|
|
||||||
|
# 百度地图的 AK 密钥,用于获取地图数据。
|
||||||
|
# 请在百度地图官网申请获取 AK 密钥。
|
||||||
|
AK=
|
||||||
|
# 百度地图sk密钥,用于获取地图数据。
|
||||||
|
# 请在百度地图官网申请获取 SK 密钥。
|
||||||
|
SK=
|
||||||
|
|
||||||
|
# ======================
|
||||||
|
# ElasticSearch配置
|
||||||
|
# ======================
|
||||||
|
# ElasticSearch的URL,默认为 "http://localhost:9200"。
|
||||||
|
ES_HOST=localhost
|
||||||
|
|
||||||
|
# ElasticSearch的端口,默认为 9200。
|
||||||
|
ES_PORT=9200
|
||||||
|
|
||||||
|
# ElasticSearch的索引名称,默认为 "cutoms_system"。
|
||||||
|
ES_INDEX=product_codes
|
||||||
|
|
||||||
|
# ElasticSearch的用户名。
|
||||||
|
ES_USER=elastic
|
||||||
|
|
||||||
|
# ElasticSearch的密码。
|
||||||
|
ES_PASSWORD=changeme
|
||||||
@ -9,6 +9,7 @@ from functools import wraps
|
|||||||
|
|
||||||
from fastapi import Request
|
from fastapi import Request
|
||||||
|
|
||||||
|
from config.constant import RedisKeyConfig
|
||||||
from controller.login import LoginController
|
from controller.login import LoginController
|
||||||
from exceptions.exception import PermissionException
|
from exceptions.exception import PermissionException
|
||||||
|
|
||||||
@ -40,3 +41,29 @@ class Auth:
|
|||||||
raise PermissionException(message="该用户无此接口权限!")
|
raise PermissionException(message="该用户无此接口权限!")
|
||||||
|
|
||||||
return wrapper
|
return wrapper
|
||||||
|
|
||||||
|
|
||||||
|
async def hasAuth(request: Request, permission: str) -> bool:
|
||||||
|
"""
|
||||||
|
判断是有拥有某项权限
|
||||||
|
"""
|
||||||
|
token = request.headers.get('Authorization') # 直接使用 request 对象
|
||||||
|
current_user = await LoginController.get_current_user(request, token)
|
||||||
|
permissions = current_user.get('permissions')
|
||||||
|
if permission in permissions:
|
||||||
|
return True
|
||||||
|
else:
|
||||||
|
return False
|
||||||
|
|
||||||
|
|
||||||
|
async def hasAdmin(request: Request, department_id: str) -> bool:
|
||||||
|
"""
|
||||||
|
判断是否有管理员权限
|
||||||
|
"""
|
||||||
|
permissions = []
|
||||||
|
if ids := await request.app.state.redis.get(f'{RedisKeyConfig.SYSTEM_CONFIG.key}:permission_departments'):
|
||||||
|
permissions = eval(ids)
|
||||||
|
if department_id in permissions:
|
||||||
|
return True
|
||||||
|
else:
|
||||||
|
return False
|
||||||
|
|||||||
1304
api/code.py
1304
api/code.py
File diff suppressed because it is too large
Load Diff
@ -125,7 +125,7 @@ async def get_config_list(request: Request,
|
|||||||
type: Optional[str] = Query(default=None, description="系统内置"),
|
type: Optional[str] = Query(default=None, description="系统内置"),
|
||||||
):
|
):
|
||||||
filterArgs = {
|
filterArgs = {
|
||||||
f'{k}__contains': v for k, v in {
|
f'{k}__icontains': v for k, v in {
|
||||||
'name': name,
|
'name': name,
|
||||||
'key': key,
|
'key': key,
|
||||||
'type': type,
|
'type': type,
|
||||||
|
|||||||
@ -79,8 +79,9 @@ async def delete_department_list(request: Request, params: DeleteListParams,
|
|||||||
if department := await Department.get_or_none(id=item, del_flag=1):
|
if department := await Department.get_or_none(id=item, del_flag=1):
|
||||||
if item in sub_departments:
|
if item in sub_departments:
|
||||||
await delete_department_recursive(department_id=department.id)
|
await delete_department_recursive(department_id=department.id)
|
||||||
if await request.app.state.redis.get(f'{RedisKeyConfig.USER_INFO.key}:*'):
|
userInfos = await request.app.state.redis.keys(f'{RedisKeyConfig.USER_INFO.key}*')
|
||||||
await request.app.state.redis.delete(f'{RedisKeyConfig.USER_INFO.key}:*')
|
if userInfos:
|
||||||
|
await request.app.state.redis.delete(*userInfos)
|
||||||
return Response.success(msg="删除成功!")
|
return Response.success(msg="删除成功!")
|
||||||
|
|
||||||
|
|
||||||
@ -116,6 +117,9 @@ async def update_department(request: Request, params: AddDepartmentParams, id: s
|
|||||||
department.sort = params.sort
|
department.sort = params.sort
|
||||||
department.status = params.status
|
department.status = params.status
|
||||||
await department.save()
|
await department.save()
|
||||||
|
userInfos = await request.app.state.redis.keys(f'{RedisKeyConfig.USER_INFO.key}*')
|
||||||
|
if userInfos:
|
||||||
|
await request.app.state.redis.delete(*userInfos)
|
||||||
return Response.success(msg="修改成功!")
|
return Response.success(msg="修改成功!")
|
||||||
else:
|
else:
|
||||||
return Response.error(msg="修改失败,部门不存在!")
|
return Response.error(msg="修改失败,部门不存在!")
|
||||||
@ -167,7 +171,7 @@ async def get_department_list(
|
|||||||
current_user: dict = Depends(LoginController.get_current_user)
|
current_user: dict = Depends(LoginController.get_current_user)
|
||||||
):
|
):
|
||||||
filterArgs = {
|
filterArgs = {
|
||||||
f'{k}__contains': v for k, v in {
|
f'{k}__icontains': v for k, v in {
|
||||||
'name': name,
|
'name': name,
|
||||||
'principal': principal,
|
'principal': principal,
|
||||||
'phone': phone,
|
'phone': phone,
|
||||||
|
|||||||
19
api/file.py
19
api/file.py
@ -7,7 +7,6 @@
|
|||||||
# @Comment : 本程序
|
# @Comment : 本程序
|
||||||
import os
|
import os
|
||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
|
|
||||||
from fastapi import APIRouter, UploadFile, File, Path, Depends, Request, Query
|
from fastapi import APIRouter, UploadFile, File, Path, Depends, Request, Query
|
||||||
from fastapi.responses import FileResponse, JSONResponse
|
from fastapi.responses import FileResponse, JSONResponse
|
||||||
|
|
||||||
@ -40,7 +39,7 @@ async def upload_file(
|
|||||||
raise ModelValidatorException(message="文件类型不支持")
|
raise ModelValidatorException(message="文件类型不支持")
|
||||||
|
|
||||||
# 2. 生成唯一的文件名
|
# 2. 生成唯一的文件名
|
||||||
timestamp = datetime.now().strftime("%Y%m%d%H%M%S")
|
timestamp = datetime.now().strftime("%Y%m%d%H%M%S%f")
|
||||||
unique_filename = f"{current_user.get('id')}_{timestamp}.{file_extension}"
|
unique_filename = f"{current_user.get('id')}_{timestamp}.{file_extension}"
|
||||||
|
|
||||||
# 3. 保存文件到服务器
|
# 3. 保存文件到服务器
|
||||||
@ -61,7 +60,7 @@ async def upload_file(
|
|||||||
relative_path=relative_path,
|
relative_path=relative_path,
|
||||||
uploader_id=current_user.get("id"),
|
uploader_id=current_user.get("id"),
|
||||||
)
|
)
|
||||||
result = await file_record.first().values(
|
result = await FileModel.get_or_none(id=file_record.id).values(
|
||||||
id="id",
|
id="id",
|
||||||
name="name",
|
name="name",
|
||||||
size="size",
|
size="size",
|
||||||
@ -86,7 +85,7 @@ async def download_file(
|
|||||||
id: str = Path(..., description="文件ID"),
|
id: str = Path(..., description="文件ID"),
|
||||||
):
|
):
|
||||||
# 1. 查询文件记录
|
# 1. 查询文件记录
|
||||||
file_record = await FileModel.get_or_none(id=id)
|
file_record = await FileModel.get_or_none(id=id, del_flag=1)
|
||||||
if not file_record:
|
if not file_record:
|
||||||
raise ServiceException(message="文件不存在!")
|
raise ServiceException(message="文件不存在!")
|
||||||
|
|
||||||
@ -110,10 +109,10 @@ async def get_file_info(
|
|||||||
current_user: dict = Depends(LoginController.get_current_user),
|
current_user: dict = Depends(LoginController.get_current_user),
|
||||||
):
|
):
|
||||||
# 1. 查询文件记录
|
# 1. 查询文件记录
|
||||||
file_record = await FileModel.get_or_none(id=id)
|
file_record = await FileModel.get_or_none(id=id, del_flag=1)
|
||||||
if not file_record:
|
if not file_record:
|
||||||
raise ServiceException(message="文件不存在!")
|
raise ServiceException(message="文件不存在!")
|
||||||
result = await file_record.first().values(
|
result = await FileModel.get_or_none(id=id, del_flag=1).values(
|
||||||
id="id",
|
id="id",
|
||||||
name="name",
|
name="name",
|
||||||
size="size",
|
size="size",
|
||||||
@ -139,11 +138,11 @@ async def delete_file(
|
|||||||
id: str = Path(..., description="文件ID"),
|
id: str = Path(..., description="文件ID"),
|
||||||
current_user: dict = Depends(LoginController.get_current_user), ):
|
current_user: dict = Depends(LoginController.get_current_user), ):
|
||||||
# 1. 查询文件记录
|
# 1. 查询文件记录
|
||||||
file_record = await FileModel.get_or_none(id=id)
|
file_record = await FileModel.get_or_none(id=id, del_flag=1)
|
||||||
if not file_record:
|
if not file_record:
|
||||||
raise ServiceException(message="文件不存在!")
|
raise ServiceException(message="文件不存在!")
|
||||||
if await Upload.check_file_exists(file_record.absolute_path):
|
if Upload.check_file_exists(file_record.absolute_path):
|
||||||
await Upload.delete_file(file_record.absolute_path)
|
Upload.delete_file(file_record.absolute_path)
|
||||||
await file_record.delete()
|
await file_record.delete()
|
||||||
return Response.success()
|
return Response.success()
|
||||||
|
|
||||||
@ -164,7 +163,7 @@ async def get_file_list(
|
|||||||
current_user: dict = Depends(LoginController.get_current_user), ):
|
current_user: dict = Depends(LoginController.get_current_user), ):
|
||||||
# 1. 查询文件记录
|
# 1. 查询文件记录
|
||||||
filterArgs = {
|
filterArgs = {
|
||||||
f'{k}__contains': v for k, v in {
|
f'{k}__icontains': v for k, v in {
|
||||||
'name': name,
|
'name': name,
|
||||||
'file_type': file_type,
|
'file_type': file_type,
|
||||||
'uploader__id': uploader_id,
|
'uploader__id': uploader_id,
|
||||||
|
|||||||
741
api/hts.py
Normal file
741
api/hts.py
Normal file
@ -0,0 +1,741 @@
|
|||||||
|
# _*_ coding : UTF-8 _*_
|
||||||
|
# @Time : 2025/03/24 02:29:03
|
||||||
|
# @UpdateTime : 2025/03/24 02:29:03
|
||||||
|
# @Author : sonder
|
||||||
|
# @File : version.py
|
||||||
|
# @Comment : 本程序用于生成版本信息增删改查接口
|
||||||
|
import json
|
||||||
|
import uuid
|
||||||
|
from datetime import datetime
|
||||||
|
from typing import Optional
|
||||||
|
|
||||||
|
from fastapi import APIRouter, Depends, Path, Request, Query
|
||||||
|
from fastapi.responses import JSONResponse
|
||||||
|
|
||||||
|
from annotation.auth import Auth
|
||||||
|
from annotation.log import Log
|
||||||
|
from config.constant import BusinessType
|
||||||
|
from controller.login import LoginController
|
||||||
|
from models import Version, HtsClass, HtsItem, File
|
||||||
|
from schemas.common import BaseResponse, DeleteListParams
|
||||||
|
from schemas.hts import AddHTSClassParams, UpdateHTSClassParams, GetHTSClassInfoResponse, GetHTSClassListResponse, \
|
||||||
|
ImportHtsItemParams
|
||||||
|
from schemas.hts import AddHtsItemParams, UpdateHtsItemParams, GetHtsItemInfoResponse, GetHtsItemListResponse
|
||||||
|
from schemas.hts import AddVersionParams, UpdateVersionParams, GetVersionInfoResponse, GetVersionListResponse
|
||||||
|
from utils.response import Response
|
||||||
|
|
||||||
|
htsAPI = APIRouter(
|
||||||
|
prefix="/hts",
|
||||||
|
dependencies=[Depends(LoginController.get_current_user)],
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@htsAPI.post("/version/add", response_class=JSONResponse, response_model=BaseResponse, summary="新增税率编码版本")
|
||||||
|
@Log(title="新增税率编码版本", business_type=BusinessType.INSERT)
|
||||||
|
@Auth(permission_list=["version:btn:add"])
|
||||||
|
async def add_version(request: Request, params: AddVersionParams):
|
||||||
|
if await Version.get_or_none(
|
||||||
|
version=params.version,
|
||||||
|
|
||||||
|
date=params.date,
|
||||||
|
|
||||||
|
url=params.url,
|
||||||
|
|
||||||
|
del_flag=1
|
||||||
|
):
|
||||||
|
return Response.error(msg="税率编码版本已存在!")
|
||||||
|
version = await Version.create(
|
||||||
|
|
||||||
|
version=params.version,
|
||||||
|
|
||||||
|
date=params.date,
|
||||||
|
|
||||||
|
url=params.url,
|
||||||
|
)
|
||||||
|
if version:
|
||||||
|
return Response.success(msg="新增成功!")
|
||||||
|
else:
|
||||||
|
return Response.error(msg="新增失败")
|
||||||
|
|
||||||
|
|
||||||
|
@htsAPI.delete("/version/delete/{id}", response_class=JSONResponse, response_model=BaseResponse,
|
||||||
|
summary="删除税率编码版本")
|
||||||
|
@htsAPI.post("/version/delete/{id}", response_class=JSONResponse, response_model=BaseResponse,
|
||||||
|
summary="删除税率编码版本")
|
||||||
|
@Log(title="删除税率编码版本", business_type=BusinessType.DELETE)
|
||||||
|
@Auth(permission_list=["version:btn:delete"])
|
||||||
|
async def delete_version(request: Request, id: str = Path(description="税率编码版本ID")):
|
||||||
|
if version := await Version.get_or_none(id=id, del_flag=1):
|
||||||
|
version.del_flag = 0
|
||||||
|
await version.save()
|
||||||
|
return Response.success(msg="删除成功")
|
||||||
|
else:
|
||||||
|
return Response.error(msg="税率编码版本不存在!")
|
||||||
|
|
||||||
|
|
||||||
|
@htsAPI.delete("/version/deleteList", response_class=JSONResponse, response_model=BaseResponse,
|
||||||
|
summary="批量删除税率编码版本")
|
||||||
|
@htsAPI.post("/version/deleteList", response_class=JSONResponse, response_model=BaseResponse,
|
||||||
|
summary="批量删除税率编码版本")
|
||||||
|
@Log(title="批量删除税率编码版本", business_type=BusinessType.DELETE)
|
||||||
|
@Auth(permission_list=["version:btn:delete"])
|
||||||
|
async def delete_version_list(request: Request, params: DeleteListParams):
|
||||||
|
for id in set(params.ids):
|
||||||
|
if version := await Version.get_or_none(id=id, del_flag=1):
|
||||||
|
version.del_flag = 0
|
||||||
|
await version.save()
|
||||||
|
return Response.success(msg="删除成功")
|
||||||
|
|
||||||
|
|
||||||
|
@htsAPI.put("/version/update/{id}", response_class=JSONResponse, response_model=BaseResponse,
|
||||||
|
summary="修改税率编码版本")
|
||||||
|
@htsAPI.post("/version/update/{id}", response_class=JSONResponse, response_model=BaseResponse,
|
||||||
|
summary="修改税率编码版本")
|
||||||
|
@Log(title="修改税率编码版本", business_type=BusinessType.UPDATE)
|
||||||
|
@Auth(permission_list=["version:btn:update"])
|
||||||
|
async def update_version(request: Request, params: UpdateVersionParams, id: str = Path(description="税率编码版本ID")):
|
||||||
|
if version := await Version.get_or_none(id=id, del_flag=1):
|
||||||
|
print(params)
|
||||||
|
version.version = params.version
|
||||||
|
version.date = params.date
|
||||||
|
version.url = params.url
|
||||||
|
await version.save()
|
||||||
|
return Response.success(msg="修改成功")
|
||||||
|
else:
|
||||||
|
return Response.error(msg="税率编码版本不存在")
|
||||||
|
|
||||||
|
|
||||||
|
@htsAPI.get("/version/info/{id}", response_class=JSONResponse, response_model=GetVersionInfoResponse,
|
||||||
|
summary="获取税率编码版本信息")
|
||||||
|
@Log(title="获取税率编码版本信息", business_type=BusinessType.SELECT)
|
||||||
|
@Auth(permission_list=["version:btn:info"])
|
||||||
|
async def get_version_info(request: Request, id: str = Path(description="税率编码版本ID")):
|
||||||
|
if version := await Version.get_or_none(id=id, del_flag=1):
|
||||||
|
data = {
|
||||||
|
|
||||||
|
"id": version.id,
|
||||||
|
|
||||||
|
"create_time": version.create_time,
|
||||||
|
|
||||||
|
"update_time": version.update_time,
|
||||||
|
|
||||||
|
"version": version.version,
|
||||||
|
|
||||||
|
"date": version.date,
|
||||||
|
|
||||||
|
"url": version.url,
|
||||||
|
|
||||||
|
}
|
||||||
|
return Response.success(data=data)
|
||||||
|
else:
|
||||||
|
return Response.error(msg="税率编码版本不存在")
|
||||||
|
|
||||||
|
|
||||||
|
@htsAPI.get("/version/list", response_class=JSONResponse, response_model=GetVersionListResponse,
|
||||||
|
summary="获取税率编码版本列表")
|
||||||
|
@Log(title="获取税率编码版本列表", business_type=BusinessType.SELECT)
|
||||||
|
@Auth(permission_list=["version:btn:list"])
|
||||||
|
async def get_version_list(
|
||||||
|
request: Request,
|
||||||
|
|
||||||
|
page: int = Query(default=1, description="当前页码"),
|
||||||
|
|
||||||
|
pageSize: int = Query(default=10, description="每页数量"),
|
||||||
|
|
||||||
|
version: Optional[str] = Query(default=None, description="版本号"),
|
||||||
|
|
||||||
|
date: Optional[str] = Query(default=None, description="版本日期"),
|
||||||
|
|
||||||
|
url: Optional[str] = Query(default=None, description="下载地址"),
|
||||||
|
|
||||||
|
):
|
||||||
|
filterArgs = {
|
||||||
|
|
||||||
|
"version__icontains": version,
|
||||||
|
|
||||||
|
"date__icontains": date,
|
||||||
|
|
||||||
|
"url__icontains": url,
|
||||||
|
|
||||||
|
}
|
||||||
|
filterArgs = {k: v for k, v in filterArgs.items() if v is not None}
|
||||||
|
total = await Version.filter(**filterArgs, del_flag=1).count()
|
||||||
|
data = await Version.filter(**filterArgs, del_flag=1).offset((page - 1) * pageSize).limit(pageSize).values(
|
||||||
|
id="id",
|
||||||
|
create_time="create_time",
|
||||||
|
update_time="update_time",
|
||||||
|
version="version",
|
||||||
|
date="date",
|
||||||
|
url="url",
|
||||||
|
)
|
||||||
|
return Response.success(data={
|
||||||
|
"total": total,
|
||||||
|
"result": data,
|
||||||
|
"page": page,
|
||||||
|
"pageSize": pageSize,
|
||||||
|
})
|
||||||
|
|
||||||
|
|
||||||
|
@htsAPI.post("/class/add", response_class=JSONResponse, response_model=BaseResponse, summary="新增编码类别")
|
||||||
|
@Log(title="新增编码类别", business_type=BusinessType.INSERT)
|
||||||
|
@Auth(permission_list=["htsclass:btn:add"])
|
||||||
|
async def add_htsclass(request: Request, params: AddHTSClassParams):
|
||||||
|
if await HtsClass.get_or_none(
|
||||||
|
|
||||||
|
class_name=params.class_name,
|
||||||
|
|
||||||
|
class_description=params.class_description,
|
||||||
|
|
||||||
|
chapter_name=params.chapter_name,
|
||||||
|
|
||||||
|
chapter_description=params.chapter_description,
|
||||||
|
|
||||||
|
del_flag=1
|
||||||
|
):
|
||||||
|
return Response.error(msg="编码类别已存在!")
|
||||||
|
htsclass = await HtsClass.create(
|
||||||
|
|
||||||
|
class_name=params.class_name,
|
||||||
|
|
||||||
|
class_description=params.class_description,
|
||||||
|
|
||||||
|
chapter_name=params.chapter_name,
|
||||||
|
|
||||||
|
chapter_description=params.chapter_description,
|
||||||
|
|
||||||
|
)
|
||||||
|
if htsclass:
|
||||||
|
return Response.success(msg="新增成功!")
|
||||||
|
else:
|
||||||
|
return Response.error(msg="新增失败")
|
||||||
|
|
||||||
|
|
||||||
|
@htsAPI.delete("/class/delete/{id}", response_class=JSONResponse, response_model=BaseResponse, summary="删除编码类别")
|
||||||
|
@htsAPI.post("/class/delete/{id}", response_class=JSONResponse, response_model=BaseResponse, summary="删除编码类别")
|
||||||
|
@Log(title="删除编码类别", business_type=BusinessType.DELETE)
|
||||||
|
@Auth(permission_list=["htsclass:btn:delete"])
|
||||||
|
async def delete_htsclass(request: Request, id: str = Path(description="编码类别ID")):
|
||||||
|
if htsclass := await HtsClass.get_or_none(id=id, del_flag=1):
|
||||||
|
htsclass.del_flag = 0
|
||||||
|
await htsclass.save()
|
||||||
|
return Response.success(msg="删除成功")
|
||||||
|
else:
|
||||||
|
return Response.error(msg="编码类别不存在!")
|
||||||
|
|
||||||
|
|
||||||
|
@htsAPI.delete("/class/deleteList", response_class=JSONResponse, response_model=BaseResponse,
|
||||||
|
summary="批量删除编码类别")
|
||||||
|
@htsAPI.post("/class/deleteList", response_class=JSONResponse, response_model=BaseResponse, summary="批量删除编码类别")
|
||||||
|
@Log(title="批量删除编码类别", business_type=BusinessType.DELETE)
|
||||||
|
@Auth(permission_list=["htsclass:btn:delete"])
|
||||||
|
async def delete_htsclass_list(request: Request, params: DeleteListParams):
|
||||||
|
for id in set(params.ids):
|
||||||
|
if htsclass := await HtsClass.get_or_none(id=id, del_flag=1):
|
||||||
|
htsclass.del_flag = 0
|
||||||
|
await htsclass.save()
|
||||||
|
return Response.success(msg="删除成功")
|
||||||
|
|
||||||
|
|
||||||
|
@htsAPI.put("/class/update/{id}", response_class=JSONResponse, response_model=BaseResponse, summary="修改编码类别")
|
||||||
|
@htsAPI.post("/class/update/{id}", response_class=JSONResponse, response_model=BaseResponse, summary="修改编码类别")
|
||||||
|
@Log(title="修改编码类别", business_type=BusinessType.UPDATE)
|
||||||
|
@Auth(permission_list=["htsclass:btn:update"])
|
||||||
|
async def update_htsclass(request: Request, params: UpdateHTSClassParams, id: str = Path(description="编码类别ID")):
|
||||||
|
if htsclass := await HtsClass.get_or_none(id=id, del_flag=1):
|
||||||
|
|
||||||
|
htsclass.class_name = params.class_name
|
||||||
|
|
||||||
|
htsclass.class_description = params.class_description
|
||||||
|
|
||||||
|
htsclass.chapter_name = params.chapter_name
|
||||||
|
|
||||||
|
htsclass.chapter_description = params.chapter_description
|
||||||
|
|
||||||
|
await htsclass.save()
|
||||||
|
return Response.success(msg="修改成功")
|
||||||
|
else:
|
||||||
|
return Response.error(msg="编码类别不存在")
|
||||||
|
|
||||||
|
|
||||||
|
@htsAPI.get("/class/info/{id}", response_class=JSONResponse, response_model=GetHTSClassInfoResponse,
|
||||||
|
summary="获取编码类别信息")
|
||||||
|
@Log(title="获取编码类别信息", business_type=BusinessType.SELECT)
|
||||||
|
@Auth(permission_list=["htsclass:btn:info"])
|
||||||
|
async def get_htsclass_info(request: Request, id: str = Path(description="编码类别ID")):
|
||||||
|
if htsclass := await HtsClass.get_or_none(id=id, del_flag=1):
|
||||||
|
data = {
|
||||||
|
|
||||||
|
"id": htsclass.id,
|
||||||
|
|
||||||
|
"create_time": htsclass.create_time,
|
||||||
|
|
||||||
|
"update_time": htsclass.update_time,
|
||||||
|
|
||||||
|
"class_name": htsclass.class_name,
|
||||||
|
|
||||||
|
"class_description": htsclass.class_description,
|
||||||
|
|
||||||
|
"chapter_name": htsclass.chapter_name,
|
||||||
|
|
||||||
|
"chapter_description": htsclass.chapter_description,
|
||||||
|
|
||||||
|
}
|
||||||
|
return Response.success(data=data)
|
||||||
|
else:
|
||||||
|
return Response.error(msg="编码类别不存在")
|
||||||
|
|
||||||
|
|
||||||
|
@htsAPI.get("/class/list", response_class=JSONResponse, response_model=GetHTSClassListResponse,
|
||||||
|
summary="获取编码类别列表")
|
||||||
|
@Log(title="获取编码类别列表", business_type=BusinessType.SELECT)
|
||||||
|
@Auth(permission_list=["htsclass:btn:list"])
|
||||||
|
async def get_htsclass_list(
|
||||||
|
request: Request,
|
||||||
|
|
||||||
|
page: int = Query(default=1, description="当前页码"),
|
||||||
|
|
||||||
|
pageSize: int = Query(default=10, description="每页数量"),
|
||||||
|
|
||||||
|
class_name: Optional[str] = Query(default=None, description="类名"),
|
||||||
|
|
||||||
|
class_description: Optional[str] = Query(default=None, description="类描述"),
|
||||||
|
|
||||||
|
chapter_name: Optional[str] = Query(default=None, description="章节"),
|
||||||
|
|
||||||
|
chapter_description: Optional[str] = Query(default=None, description="章节描述"),
|
||||||
|
|
||||||
|
):
|
||||||
|
filterArgs = {
|
||||||
|
|
||||||
|
"class_name__icontains": class_name,
|
||||||
|
|
||||||
|
"class_description__icontains": class_description,
|
||||||
|
|
||||||
|
"chapter_name__icontains": chapter_name,
|
||||||
|
|
||||||
|
"chapter_description__icontains": chapter_description,
|
||||||
|
|
||||||
|
}
|
||||||
|
filterArgs = {k: v for k, v in filterArgs.items() if v is not None}
|
||||||
|
total = await HtsClass.filter(**filterArgs, del_flag=1).count()
|
||||||
|
data = await HtsClass.filter(**filterArgs, del_flag=1).offset((page - 1) * pageSize).limit(pageSize).values(
|
||||||
|
|
||||||
|
id="id",
|
||||||
|
|
||||||
|
create_time="create_time",
|
||||||
|
|
||||||
|
update_time="update_time",
|
||||||
|
|
||||||
|
class_name="class_name",
|
||||||
|
|
||||||
|
class_description="class_description",
|
||||||
|
|
||||||
|
chapter_name="chapter_name",
|
||||||
|
|
||||||
|
chapter_description="chapter_description",
|
||||||
|
|
||||||
|
)
|
||||||
|
return Response.success(data={
|
||||||
|
"total": total,
|
||||||
|
"result": data,
|
||||||
|
"page": page,
|
||||||
|
"pageSize": pageSize,
|
||||||
|
})
|
||||||
|
|
||||||
|
|
||||||
|
@htsAPI.post("/item/add", response_class=JSONResponse, response_model=BaseResponse, summary="新增编码项")
|
||||||
|
@Log(title="新增编码项", business_type=BusinessType.INSERT)
|
||||||
|
@Auth(permission_list=["htsitem:btn:add"])
|
||||||
|
async def add_htsitem(request: Request, params: AddHtsItemParams):
|
||||||
|
if await HtsItem.get_or_none(
|
||||||
|
|
||||||
|
parent_id=params.parent_id,
|
||||||
|
|
||||||
|
htsno=params.htsno,
|
||||||
|
|
||||||
|
indent=params.indent,
|
||||||
|
|
||||||
|
description=params.description,
|
||||||
|
|
||||||
|
units=params.units,
|
||||||
|
|
||||||
|
general=params.general,
|
||||||
|
|
||||||
|
special=params.special,
|
||||||
|
|
||||||
|
other=params.other,
|
||||||
|
|
||||||
|
quota_quantity=params.quota_quantity,
|
||||||
|
|
||||||
|
additional_duties=params.additional_duties,
|
||||||
|
|
||||||
|
footnotes=params.footnotes,
|
||||||
|
|
||||||
|
class_id=params.class_id,
|
||||||
|
|
||||||
|
version_id=params.version_id,
|
||||||
|
|
||||||
|
del_flag=1
|
||||||
|
):
|
||||||
|
return Response.error(msg="编码项目已存在!")
|
||||||
|
htsitem = await HtsItem.create(
|
||||||
|
|
||||||
|
parent_id=params.parent_id,
|
||||||
|
|
||||||
|
htsno=params.htsno,
|
||||||
|
|
||||||
|
indent=params.indent,
|
||||||
|
|
||||||
|
description=params.description,
|
||||||
|
|
||||||
|
units=params.units,
|
||||||
|
|
||||||
|
general=params.general,
|
||||||
|
|
||||||
|
special=params.special,
|
||||||
|
|
||||||
|
other=params.other,
|
||||||
|
|
||||||
|
quota_quantity=params.quota_quantity,
|
||||||
|
|
||||||
|
additional_duties=params.additional_duties,
|
||||||
|
|
||||||
|
footnotes=params.footnotes,
|
||||||
|
|
||||||
|
class_id=params.class_id,
|
||||||
|
|
||||||
|
version_id=params.version_id,
|
||||||
|
|
||||||
|
)
|
||||||
|
if htsitem:
|
||||||
|
return Response.success(msg="新增成功!")
|
||||||
|
else:
|
||||||
|
return Response.error(msg="新增失败")
|
||||||
|
|
||||||
|
|
||||||
|
@htsAPI.delete("/item/delete/{id}", response_class=JSONResponse, response_model=BaseResponse, summary="删除编码项")
|
||||||
|
@htsAPI.post("/item/delete/{id}", response_class=JSONResponse, response_model=BaseResponse, summary="删除编码项")
|
||||||
|
@Log(title="删除编码项", business_type=BusinessType.DELETE)
|
||||||
|
@Auth(permission_list=["htsitem:btn:delete"])
|
||||||
|
async def delete_htsitem(request: Request, id: str = Path(description="编码项目ID")):
|
||||||
|
if htsitem := await HtsItem.get_or_none(id=id, del_flag=1):
|
||||||
|
htsitem.del_flag = 0
|
||||||
|
await htsitem.save()
|
||||||
|
return Response.success(msg="删除成功")
|
||||||
|
else:
|
||||||
|
return Response.error(msg="编码项目不存在!")
|
||||||
|
|
||||||
|
|
||||||
|
@htsAPI.delete("/item/deleteList", response_class=JSONResponse, response_model=BaseResponse, summary="批量删除编码项")
|
||||||
|
@htsAPI.post("/item/deleteList", response_class=JSONResponse, response_model=BaseResponse, summary="批量删除编码项")
|
||||||
|
@Log(title="批量删除编码项", business_type=BusinessType.DELETE)
|
||||||
|
@Auth(permission_list=["htsitem:btn:delete"])
|
||||||
|
async def delete_htsitem_list(request: Request, params: DeleteListParams):
|
||||||
|
for id in set(params.ids):
|
||||||
|
if htsitem := await HtsItem.get_or_none(id=id, del_flag=1):
|
||||||
|
htsitem.del_flag = 0
|
||||||
|
await htsitem.save()
|
||||||
|
return Response.success(msg="删除成功")
|
||||||
|
|
||||||
|
|
||||||
|
@htsAPI.put("/item/update/{id}", response_class=JSONResponse, response_model=BaseResponse, summary="修改编码项")
|
||||||
|
@htsAPI.post("/item/update/{id}", response_class=JSONResponse, response_model=BaseResponse, summary="修改编码项")
|
||||||
|
@Log(title="修改编码项", business_type=BusinessType.UPDATE)
|
||||||
|
@Auth(permission_list=["htsitem:btn:update"])
|
||||||
|
async def update_htsitem(request: Request, params: UpdateHtsItemParams, id: str = Path(description="编码项目ID")):
|
||||||
|
if htsitem := await HtsItem.get_or_none(id=id, del_flag=1):
|
||||||
|
|
||||||
|
htsitem.parent_id = params.parent_id
|
||||||
|
|
||||||
|
htsitem.htsno = params.htsno
|
||||||
|
|
||||||
|
htsitem.indent = params.indent
|
||||||
|
|
||||||
|
htsitem.description = params.description
|
||||||
|
|
||||||
|
htsitem.units = params.units
|
||||||
|
|
||||||
|
htsitem.general = params.general
|
||||||
|
|
||||||
|
htsitem.special = params.special
|
||||||
|
|
||||||
|
htsitem.other = params.other
|
||||||
|
|
||||||
|
htsitem.quota_quantity = params.quota_quantity
|
||||||
|
|
||||||
|
htsitem.additional_duties = params.additional_duties
|
||||||
|
|
||||||
|
htsitem.footnotes = params.footnotes
|
||||||
|
if class_ := await HtsClass.get_or_none(id=params.class_id, del_flag=1):
|
||||||
|
htsitem.class_ = class_
|
||||||
|
if version := await Version.get_or_none(id=params.version_id, del_flag=1):
|
||||||
|
htsitem.version = version
|
||||||
|
htsitem.version_id = params.version_id
|
||||||
|
|
||||||
|
await htsitem.save()
|
||||||
|
return Response.success(msg="修改成功")
|
||||||
|
else:
|
||||||
|
return Response.error(msg="编码项目不存在")
|
||||||
|
|
||||||
|
|
||||||
|
@htsAPI.get("/item/info/{id}", response_class=JSONResponse, response_model=GetHtsItemInfoResponse,
|
||||||
|
summary="获取编码项信息")
|
||||||
|
@Log(title="获取编码项信息", business_type=BusinessType.SELECT)
|
||||||
|
@Auth(permission_list=["htsitem:btn:info"])
|
||||||
|
async def get_htsitem_info(request: Request, id: str = Path(description="编码项目ID")):
|
||||||
|
if htsitem := await HtsItem.get_or_none(id=id, del_flag=1):
|
||||||
|
data = {
|
||||||
|
|
||||||
|
"id": htsitem.id,
|
||||||
|
|
||||||
|
"create_time": htsitem.create_time,
|
||||||
|
|
||||||
|
"update_time": htsitem.update_time,
|
||||||
|
|
||||||
|
"parent_id": htsitem.parent_id,
|
||||||
|
|
||||||
|
"htsno": htsitem.htsno,
|
||||||
|
|
||||||
|
"indent": htsitem.indent,
|
||||||
|
|
||||||
|
"description": htsitem.description,
|
||||||
|
|
||||||
|
"units": htsitem.units,
|
||||||
|
|
||||||
|
"general": htsitem.general,
|
||||||
|
|
||||||
|
"special": htsitem.special,
|
||||||
|
|
||||||
|
"other": htsitem.other,
|
||||||
|
|
||||||
|
"quota_quantity": htsitem.quota_quantity,
|
||||||
|
|
||||||
|
"additional_duties": htsitem.additional_duties,
|
||||||
|
|
||||||
|
"footnotes": htsitem.footnotes,
|
||||||
|
|
||||||
|
"class_id": htsitem.class_id,
|
||||||
|
|
||||||
|
"version_id": htsitem.version_id,
|
||||||
|
|
||||||
|
}
|
||||||
|
return Response.success(data=data)
|
||||||
|
else:
|
||||||
|
return Response.error(msg="编码项目不存在")
|
||||||
|
|
||||||
|
|
||||||
|
@htsAPI.get("/item/list", response_class=JSONResponse, response_model=GetHtsItemListResponse, summary="获取编码项列表")
|
||||||
|
@Log(title="获取编码项列表", business_type=BusinessType.SELECT)
|
||||||
|
@Auth(permission_list=["htsitem:btn:list"])
|
||||||
|
async def get_htsitem_list(
|
||||||
|
request: Request,
|
||||||
|
|
||||||
|
page: int = Query(default=1, description="当前页码"),
|
||||||
|
|
||||||
|
pageSize: int = Query(default=10, description="每页数量"),
|
||||||
|
|
||||||
|
htsno: Optional[str] = Query(default=None, description="编码"),
|
||||||
|
|
||||||
|
description: Optional[str] = Query(default=None, description="描述"),
|
||||||
|
|
||||||
|
units: Optional[str] = Query(default=None, description="单位列表"),
|
||||||
|
|
||||||
|
general: Optional[str] = Query(default=None, description="通用税率"),
|
||||||
|
|
||||||
|
special: Optional[str] = Query(default=None, description="特殊税率,适用于特定国家或地区"),
|
||||||
|
|
||||||
|
other: Optional[str] = Query(default=None, description="其他税率"),
|
||||||
|
|
||||||
|
quota_quantity: Optional[str] = Query(default=None, description="配额数量"),
|
||||||
|
|
||||||
|
additional_duties: Optional[str] = Query(default=None, description="附加税"),
|
||||||
|
|
||||||
|
footnotes: Optional[str] = Query(default=None, description="脚注列表"),
|
||||||
|
|
||||||
|
class_id: Optional[str] = Query(default=None, description="所属类"),
|
||||||
|
|
||||||
|
version_id: Optional[str] = Query(default=None, description="所属版本"),
|
||||||
|
|
||||||
|
):
|
||||||
|
filterArgs = {
|
||||||
|
|
||||||
|
"htsno__icontains": htsno,
|
||||||
|
|
||||||
|
"description__icontains": description,
|
||||||
|
|
||||||
|
"units__icontains": units,
|
||||||
|
|
||||||
|
"general__icontains": general,
|
||||||
|
|
||||||
|
"special__icontains": special,
|
||||||
|
|
||||||
|
"other__icontains": other,
|
||||||
|
|
||||||
|
"quota_quantity__icontains": quota_quantity,
|
||||||
|
|
||||||
|
"additional_duties__icontains": additional_duties,
|
||||||
|
|
||||||
|
"footnotes__icontains": footnotes,
|
||||||
|
|
||||||
|
"class__id": class_id,
|
||||||
|
|
||||||
|
"version_id": version_id,
|
||||||
|
|
||||||
|
}
|
||||||
|
filterArgs = {k: v for k, v in filterArgs.items() if v is not None}
|
||||||
|
total = await HtsItem.filter(**filterArgs, del_flag=1).count()
|
||||||
|
data = await HtsItem.filter(**filterArgs, del_flag=1).offset((page - 1) * pageSize).limit(pageSize).values(
|
||||||
|
|
||||||
|
id="id",
|
||||||
|
|
||||||
|
create_time="create_time",
|
||||||
|
|
||||||
|
update_time="update_time",
|
||||||
|
|
||||||
|
parent_id="parent_id",
|
||||||
|
|
||||||
|
htsno="htsno",
|
||||||
|
|
||||||
|
indent="indent",
|
||||||
|
|
||||||
|
description="description",
|
||||||
|
|
||||||
|
units="units",
|
||||||
|
|
||||||
|
general="general",
|
||||||
|
|
||||||
|
special="special",
|
||||||
|
|
||||||
|
other="other",
|
||||||
|
|
||||||
|
quota_quantity="quota_quantity",
|
||||||
|
|
||||||
|
additional_duties="additional_duties",
|
||||||
|
|
||||||
|
footnotes="footnotes",
|
||||||
|
|
||||||
|
class_id="class__id",
|
||||||
|
|
||||||
|
version_id="version_id",
|
||||||
|
|
||||||
|
)
|
||||||
|
return Response.success(data={
|
||||||
|
"total": total,
|
||||||
|
"result": data,
|
||||||
|
"page": page,
|
||||||
|
"pageSize": pageSize,
|
||||||
|
})
|
||||||
|
|
||||||
|
|
||||||
|
@htsAPI.post("/item/import", response_class=JSONResponse, response_model=BaseResponse, summary="导入编码项")
|
||||||
|
@Log(title="导入编码项", business_type=BusinessType.INSERT)
|
||||||
|
@Auth(permission_list=["htsitem:btn:import"])
|
||||||
|
async def add_htsitem(request: Request, params: ImportHtsItemParams):
|
||||||
|
version = await Version.get_or_none(id=params.version_id, del_flag=1)
|
||||||
|
if not version:
|
||||||
|
return Response.error(msg="版本不存在")
|
||||||
|
class_ = await HtsClass.get_or_none(id=params.class_id, del_flag=1)
|
||||||
|
if not class_:
|
||||||
|
return Response.error(msg="编码类不存在")
|
||||||
|
|
||||||
|
file = await File.get_or_none(id=params.file_id, del_flag=1)
|
||||||
|
if not file:
|
||||||
|
return Response.error(msg="文件不存在")
|
||||||
|
if await HtsItem.get_or_none(class_=params.class_id, version=params.version_id):
|
||||||
|
return Response.error(msg="编码项已存在")
|
||||||
|
|
||||||
|
def build_hierarchy_with_ids(data):
|
||||||
|
hierarchy = []
|
||||||
|
stack = []
|
||||||
|
|
||||||
|
for item in data:
|
||||||
|
current_level = int(item['indent'])
|
||||||
|
current_item = {
|
||||||
|
'id': str(uuid.uuid4()), # 生成唯一的 UUID
|
||||||
|
'indent': item['indent'],
|
||||||
|
'htsno': item['htsno'].replace(".", "").strip(),
|
||||||
|
'description': item['description'],
|
||||||
|
'parent_id': None, # 默认没有父级,
|
||||||
|
"units": json.dumps(item["units"], ensure_ascii=False),
|
||||||
|
"general": item["general"],
|
||||||
|
"special": item["special"],
|
||||||
|
"other": item["other"],
|
||||||
|
"footnotes": json.dumps(item["footnotes"], ensure_ascii=False),
|
||||||
|
"quota_quantity": item["quotaQuantity"],
|
||||||
|
"additional_duties": item["additionalDuties"],
|
||||||
|
"addiitional_duties": item["addiitionalDuties"]
|
||||||
|
}
|
||||||
|
|
||||||
|
# 找到当前项的父级
|
||||||
|
while stack and int(stack[-1]['indent']) >= current_level:
|
||||||
|
stack.pop()
|
||||||
|
|
||||||
|
if stack:
|
||||||
|
# 设置当前项的 parent_id 为父级的 id
|
||||||
|
current_item['parent_id'] = stack[-1]['item']['id']
|
||||||
|
|
||||||
|
# 将当前项添加到层级结构中
|
||||||
|
hierarchy.append(current_item)
|
||||||
|
|
||||||
|
# 将当前项压入栈中
|
||||||
|
stack.append({'indent': current_level, 'item': current_item})
|
||||||
|
|
||||||
|
return hierarchy
|
||||||
|
|
||||||
|
with open(file.absolute_path, 'r', encoding='utf-8') as r:
|
||||||
|
data = json.load(r)
|
||||||
|
for x in data:
|
||||||
|
x["indent"] = str(int(x["indent"]) + 1)
|
||||||
|
data.append({
|
||||||
|
"htsno": str(class_.chapter_name).replace("Chapter", "").strip() if int(
|
||||||
|
str(class_.chapter_name).replace("Chapter", "").strip(" ")) > 10 else "0" +
|
||||||
|
str(class_.chapter_name).replace(
|
||||||
|
"Chapter", "").strip(),
|
||||||
|
"indent": "0",
|
||||||
|
"description": class_.class_description,
|
||||||
|
"units": [],
|
||||||
|
"general": "",
|
||||||
|
"special": "",
|
||||||
|
"other": "",
|
||||||
|
"footnotes": [],
|
||||||
|
"quotaQuantity": "",
|
||||||
|
"additionalDuties": "",
|
||||||
|
"addiitionalDuties": ""
|
||||||
|
})
|
||||||
|
hierarchy = build_hierarchy_with_ids(data)
|
||||||
|
# 批量插入查询结果
|
||||||
|
create_tasks = [
|
||||||
|
HtsItem(
|
||||||
|
id=item["id"],
|
||||||
|
|
||||||
|
parent_id=item["parent_id"],
|
||||||
|
|
||||||
|
htsno=item["htsno"],
|
||||||
|
|
||||||
|
indent=item["indent"],
|
||||||
|
|
||||||
|
description=item["description"],
|
||||||
|
|
||||||
|
units=item["units"],
|
||||||
|
|
||||||
|
general=item["general"],
|
||||||
|
|
||||||
|
special=item["special"],
|
||||||
|
|
||||||
|
other=item["other"],
|
||||||
|
|
||||||
|
quota_quantity=item["quota_quantity"],
|
||||||
|
|
||||||
|
additional_duties=item["additional_duties"],
|
||||||
|
|
||||||
|
footnotes=item["footnotes"],
|
||||||
|
|
||||||
|
class_=class_,
|
||||||
|
|
||||||
|
version=version
|
||||||
|
)
|
||||||
|
for item in hierarchy
|
||||||
|
]
|
||||||
|
await HtsItem.bulk_create(create_tasks)
|
||||||
|
return Response.success()
|
||||||
|
|
||||||
@ -123,7 +123,7 @@ async def get_locale_list(request: Request,
|
|||||||
code: Optional[str] = Query(default=None, description="国际化类型代码"),
|
code: Optional[str] = Query(default=None, description="国际化类型代码"),
|
||||||
):
|
):
|
||||||
filterArgs = {
|
filterArgs = {
|
||||||
f'{k}__contains': v for k, v in {
|
f'{k}__icontains': v for k, v in {
|
||||||
'name': name,
|
'name': name,
|
||||||
'code': code
|
'code': code
|
||||||
}.items() if v
|
}.items() if v
|
||||||
@ -242,7 +242,7 @@ async def get_i18n_list(request: Request,
|
|||||||
translation: Optional[str] = Query(default=None, description="国际化内容翻译内容"),
|
translation: Optional[str] = Query(default=None, description="国际化内容翻译内容"),
|
||||||
):
|
):
|
||||||
filterArgs = {
|
filterArgs = {
|
||||||
f'{k}__contains': v for k, v in {
|
f'{k}__icontains': v for k, v in {
|
||||||
'key': key,
|
'key': key,
|
||||||
'locale_id': locale_id,
|
'locale_id': locale_id,
|
||||||
'translation': translation
|
'translation': translation
|
||||||
|
|||||||
25
api/log.py
25
api/log.py
@ -12,7 +12,7 @@ from fastapi import APIRouter, Depends, Path, Query, Request
|
|||||||
from fastapi.encoders import jsonable_encoder
|
from fastapi.encoders import jsonable_encoder
|
||||||
from fastapi.responses import JSONResponse
|
from fastapi.responses import JSONResponse
|
||||||
|
|
||||||
from annotation.auth import Auth
|
from annotation.auth import Auth, hasAuth
|
||||||
from annotation.log import Log
|
from annotation.log import Log
|
||||||
from config.constant import BusinessType, RedisKeyConfig
|
from config.constant import BusinessType, RedisKeyConfig
|
||||||
from controller.login import LoginController
|
from controller.login import LoginController
|
||||||
@ -41,11 +41,10 @@ async def get_login_log(request: Request,
|
|||||||
current_user: dict = Depends(LoginController.get_current_user),
|
current_user: dict = Depends(LoginController.get_current_user),
|
||||||
):
|
):
|
||||||
sub_departments = current_user.get("sub_departments")
|
sub_departments = current_user.get("sub_departments")
|
||||||
|
user_id = current_user.get("id")
|
||||||
online_user_list = await LoginController.get_online_user(request, sub_departments)
|
online_user_list = await LoginController.get_online_user(request, sub_departments)
|
||||||
online_user_list = list(
|
|
||||||
filter(lambda x: x["department_id"] in sub_departments, jsonable_encoder(online_user_list)))
|
|
||||||
filterArgs = {
|
filterArgs = {
|
||||||
f'{k}__contains': v for k, v in {
|
f'{k}__icontains': v for k, v in {
|
||||||
'username': username,
|
'username': username,
|
||||||
'nickname': nickname,
|
'nickname': nickname,
|
||||||
}.items() if v
|
}.items() if v
|
||||||
@ -56,10 +55,20 @@ async def get_login_log(request: Request,
|
|||||||
startTime = datetime.fromtimestamp(float(startTime) / 1000)
|
startTime = datetime.fromtimestamp(float(startTime) / 1000)
|
||||||
endTime = datetime.fromtimestamp(float(endTime) / 1000)
|
endTime = datetime.fromtimestamp(float(endTime) / 1000)
|
||||||
filterArgs['login_time__range'] = [startTime, endTime]
|
filterArgs['login_time__range'] = [startTime, endTime]
|
||||||
|
if await hasAuth(request, "login:btn:admin"):
|
||||||
|
online_user_list = list(
|
||||||
|
filter(lambda x: x["department_id"] in sub_departments, jsonable_encoder(online_user_list)))
|
||||||
if not department_id:
|
if not department_id:
|
||||||
filterArgs['user__department__id__in'] = sub_departments
|
filterArgs['user__department__id__in'] = sub_departments
|
||||||
else:
|
else:
|
||||||
filterArgs['user__department__id'] = department_id
|
filterArgs['user__department__id'] = department_id
|
||||||
|
else:
|
||||||
|
online_user_list = list(
|
||||||
|
filter(lambda x: x["user_id"] == user_id, jsonable_encoder(online_user_list)))
|
||||||
|
if department_id:
|
||||||
|
filterArgs['user__department__id'] = department_id
|
||||||
|
else:
|
||||||
|
filterArgs["user__department__id"] = current_user.get("department_id")
|
||||||
result = await LoginLog.filter(**filterArgs, user__del_flag=1, del_flag=1).offset(
|
result = await LoginLog.filter(**filterArgs, user__del_flag=1, del_flag=1).offset(
|
||||||
(page - 1) * pageSize).limit(pageSize).values(
|
(page - 1) * pageSize).limit(pageSize).values(
|
||||||
id="id",
|
id="id",
|
||||||
@ -171,8 +180,9 @@ async def get_operation_log(request: Request,
|
|||||||
current_user: dict = Depends(LoginController.get_current_user),
|
current_user: dict = Depends(LoginController.get_current_user),
|
||||||
):
|
):
|
||||||
sub_departments = current_user.get("sub_departments")
|
sub_departments = current_user.get("sub_departments")
|
||||||
|
user_id = current_user.get("id")
|
||||||
filterArgs = {
|
filterArgs = {
|
||||||
f'{k}__contains': v for k, v in {
|
f'{k}__icontains': v for k, v in {
|
||||||
'operation_name': name,
|
'operation_name': name,
|
||||||
'operation_type': type,
|
'operation_type': type,
|
||||||
'operator__username': username,
|
'operator__username': username,
|
||||||
@ -185,10 +195,15 @@ async def get_operation_log(request: Request,
|
|||||||
startTime = datetime.fromtimestamp(float(startTime) / 1000)
|
startTime = datetime.fromtimestamp(float(startTime) / 1000)
|
||||||
endTime = datetime.fromtimestamp(float(endTime) / 1000)
|
endTime = datetime.fromtimestamp(float(endTime) / 1000)
|
||||||
filterArgs['operation_time__range'] = [startTime, endTime]
|
filterArgs['operation_time__range'] = [startTime, endTime]
|
||||||
|
if await hasAuth(request, "operation:btn:admin"):
|
||||||
if not department_id:
|
if not department_id:
|
||||||
filterArgs['department__id__in'] = sub_departments
|
filterArgs['department__id__in'] = sub_departments
|
||||||
else:
|
else:
|
||||||
filterArgs['department__id'] = department_id
|
filterArgs['department__id'] = department_id
|
||||||
|
else:
|
||||||
|
filterArgs['operator__id'] = user_id
|
||||||
|
if department_id:
|
||||||
|
filterArgs['department__id'] = department_id
|
||||||
result = await OperationLog.filter(**filterArgs, operator__del_flag=1, del_flag=1).offset(
|
result = await OperationLog.filter(**filterArgs, operator__del_flag=1, del_flag=1).offset(
|
||||||
(page - 1) * pageSize).limit(
|
(page - 1) * pageSize).limit(
|
||||||
pageSize).values(
|
pageSize).values(
|
||||||
|
|||||||
143
api/login.py
143
api/login.py
@ -5,6 +5,7 @@
|
|||||||
# @File : login.py
|
# @File : login.py
|
||||||
# @Software : PyCharm
|
# @Software : PyCharm
|
||||||
# @Comment : 本程序
|
# @Comment : 本程序
|
||||||
|
import json
|
||||||
import uuid
|
import uuid
|
||||||
from datetime import timedelta, datetime
|
from datetime import timedelta, datetime
|
||||||
|
|
||||||
@ -18,7 +19,7 @@ from config.constant import BusinessType
|
|||||||
from config.constant import RedisKeyConfig
|
from config.constant import RedisKeyConfig
|
||||||
from controller.login import CustomOAuth2PasswordRequestForm, LoginController
|
from controller.login import CustomOAuth2PasswordRequestForm, LoginController
|
||||||
from controller.query import QueryController
|
from controller.query import QueryController
|
||||||
from models import Department, User, Role, UserRole
|
from models import Department, User, Role, UserRole, LoginLog, OperationLog, QueryCodeLog, QueryCode, HtsClass, HtsItem,Version
|
||||||
from schemas.common import BaseResponse
|
from schemas.common import BaseResponse
|
||||||
from schemas.login import LoginParams, GetUserInfoResponse, LoginResponse, GetCaptchaResponse, GetEmailCodeParams, \
|
from schemas.login import LoginParams, GetUserInfoResponse, LoginResponse, GetCaptchaResponse, GetEmailCodeParams, \
|
||||||
ResetPasswordParams
|
ResetPasswordParams
|
||||||
@ -271,3 +272,143 @@ async def logout(request: Request, status: bool = Depends(LoginController.logout
|
|||||||
if status:
|
if status:
|
||||||
return Response.success(data="退出成功!")
|
return Response.success(data="退出成功!")
|
||||||
return Response.error(data="登出失败!")
|
return Response.error(data="登出失败!")
|
||||||
|
|
||||||
|
|
||||||
|
@loginAPI.post("/unsubscribe", response_class=JSONResponse, response_model=BaseResponse, summary="用户注销")
|
||||||
|
@Log(title="用户注销", business_type=BusinessType.FORCE)
|
||||||
|
async def unsubscribe(request: Request, current_user: dict = Depends(LoginController.get_current_user)):
|
||||||
|
id = current_user.get("id")
|
||||||
|
session_id = current_user.get("session_id")
|
||||||
|
if user := await User.get_or_none(id=id, del_flag=1):
|
||||||
|
user.del_flag = 0
|
||||||
|
await user.save()
|
||||||
|
redis_token = await request.app.state.redis.get(f'{RedisKeyConfig.ACCESS_TOKEN.key}:{session_id}')
|
||||||
|
if redis_token:
|
||||||
|
await request.app.state.redis.delete(f'{RedisKeyConfig.ACCESS_TOKEN.key}:{session_id}')
|
||||||
|
# 移除用户角色
|
||||||
|
await UserRole.filter(user_id=user.id, del_flag=1).update(del_flag=0)
|
||||||
|
# 移除用户登录日志
|
||||||
|
await LoginLog.filter(user_id=user.id, del_flag=1).update(del_flag=0)
|
||||||
|
# 移除用户操作日志
|
||||||
|
await OperationLog.filter(user_id=user.id, del_flag=1).update(del_flag=0)
|
||||||
|
# 移除编码查询日志
|
||||||
|
logs = await QueryCodeLog.filter(operator__id=user.id, del_flag=1).values("id")
|
||||||
|
for log in logs:
|
||||||
|
await QueryCode.filter(session_id=log["id"], del_flag=1).update(del_flag=0)
|
||||||
|
await QueryCodeLog.filter(id=log["id"], del_flag=1).update(del_flag=0)
|
||||||
|
# 更新用户信息缓存
|
||||||
|
if await request.app.state.redis.get(f'{RedisKeyConfig.USER_INFO.key}:{id}'):
|
||||||
|
await request.app.state.redis.delete(f'{RedisKeyConfig.USER_INFO.key}:{id}')
|
||||||
|
# 更新用户路由缓存
|
||||||
|
if await request.app.state.redis.get(f'{RedisKeyConfig.USER_ROUTES.key}:{id}'):
|
||||||
|
await request.app.state.redis.delete(f'{RedisKeyConfig.USER_ROUTES.key}:{id}')
|
||||||
|
return Response.success(data="注销成功!")
|
||||||
|
else:
|
||||||
|
return Response.error(data="注销失败!")
|
||||||
|
|
||||||
|
|
||||||
|
# @loginAPI.get("/test")
|
||||||
|
# async def test(request: Request):
|
||||||
|
# values = await HtsClass.filter(del_flag=1).values("chapter_name","id")
|
||||||
|
# version=await Version.get(id="4b2101cc-9bc3-4084-af16-eeb27b24b682")
|
||||||
|
# for item in values:
|
||||||
|
# hts = str(item["chapter_name"]).replace("Chapter", "").strip()
|
||||||
|
# class_=await HtsClass.get(id=item["id"])
|
||||||
|
# if hts == "77":
|
||||||
|
# continue
|
||||||
|
#
|
||||||
|
# def build_hierarchy_with_ids(data):
|
||||||
|
# hierarchy = []
|
||||||
|
# stack = []
|
||||||
|
#
|
||||||
|
# for item in data:
|
||||||
|
# current_level = int(item['indent'])
|
||||||
|
# current_item = {
|
||||||
|
# 'id': str(uuid.uuid4()), # 生成唯一的 UUID
|
||||||
|
# 'indent': item['indent'],
|
||||||
|
# 'htsno': item['htsno'].replace(".", "").strip(),
|
||||||
|
# 'description': item['description'],
|
||||||
|
# 'parent_id': None, # 默认没有父级,
|
||||||
|
# "units": json.dumps(item["units"], ensure_ascii=False),
|
||||||
|
# "general": item["general"],
|
||||||
|
# "special": item["special"],
|
||||||
|
# "other": item["other"],
|
||||||
|
# "footnotes": json.dumps(item["footnotes"], ensure_ascii=False),
|
||||||
|
# "quota_quantity": item["quotaQuantity"],
|
||||||
|
# "additional_duties": item["additionalDuties"],
|
||||||
|
# "addiitional_duties": item["addiitionalDuties"]
|
||||||
|
# }
|
||||||
|
#
|
||||||
|
# # 找到当前项的父级
|
||||||
|
# while stack and int(stack[-1]['indent']) >= current_level:
|
||||||
|
# stack.pop()
|
||||||
|
#
|
||||||
|
# if stack:
|
||||||
|
# # 设置当前项的 parent_id 为父级的 id
|
||||||
|
# current_item['parent_id'] = stack[-1]['item']['id']
|
||||||
|
#
|
||||||
|
# # 将当前项添加到层级结构中
|
||||||
|
# hierarchy.append(current_item)
|
||||||
|
#
|
||||||
|
# # 将当前项压入栈中
|
||||||
|
# stack.append({'indent': current_level, 'item': current_item})
|
||||||
|
#
|
||||||
|
# return hierarchy
|
||||||
|
#
|
||||||
|
# with open(f"E:/PythonCodes/cutoms-system-backend/test/{hts}.json", 'r', encoding='utf-8') as r:
|
||||||
|
# data = json.load(r)
|
||||||
|
# for x in data:
|
||||||
|
# x["indent"] = str(int(x["indent"]) + 1)
|
||||||
|
# data.append({
|
||||||
|
# "htsno": str(item["chapter_name"]).replace("Chapter", "").strip() if int(
|
||||||
|
# str(item["chapter_name"]).replace("Chapter", "").strip(" ")) > 10 else "0" +
|
||||||
|
# str(item["chapter_name"]).replace(
|
||||||
|
# "Chapter", "").strip(),
|
||||||
|
# "indent": "0",
|
||||||
|
# "description": class_.class_description,
|
||||||
|
# "units": [],
|
||||||
|
# "general": "",
|
||||||
|
# "special": "",
|
||||||
|
# "other": "",
|
||||||
|
# "footnotes": [],
|
||||||
|
# "quotaQuantity": "",
|
||||||
|
# "additionalDuties": "",
|
||||||
|
# "addiitionalDuties": ""
|
||||||
|
# })
|
||||||
|
# hierarchy = build_hierarchy_with_ids(data)
|
||||||
|
# # 批量插入查询结果
|
||||||
|
# create_tasks = [
|
||||||
|
# HtsItem(
|
||||||
|
# id=item["id"],
|
||||||
|
#
|
||||||
|
# parent_id=item["parent_id"],
|
||||||
|
#
|
||||||
|
# htsno=item["htsno"],
|
||||||
|
#
|
||||||
|
# indent=item["indent"],
|
||||||
|
#
|
||||||
|
# description=item["description"],
|
||||||
|
#
|
||||||
|
# units=item["units"],
|
||||||
|
#
|
||||||
|
# general=item["general"],
|
||||||
|
#
|
||||||
|
# special=item["special"],
|
||||||
|
#
|
||||||
|
# other=item["other"],
|
||||||
|
#
|
||||||
|
# quota_quantity=item["quota_quantity"],
|
||||||
|
#
|
||||||
|
# additional_duties=item["additional_duties"],
|
||||||
|
#
|
||||||
|
# footnotes=item["footnotes"],
|
||||||
|
#
|
||||||
|
# class_=class_,
|
||||||
|
#
|
||||||
|
# version=version
|
||||||
|
# )
|
||||||
|
# for item in hierarchy
|
||||||
|
# ]
|
||||||
|
# await HtsItem.bulk_create(create_tasks)
|
||||||
|
# print(class_.chapter_name)
|
||||||
|
|
||||||
|
|||||||
@ -10,11 +10,11 @@ from typing import Optional
|
|||||||
from fastapi import APIRouter, Depends, Path, Query, Request
|
from fastapi import APIRouter, Depends, Path, Query, Request
|
||||||
from fastapi.responses import JSONResponse
|
from fastapi.responses import JSONResponse
|
||||||
|
|
||||||
from annotation.auth import Auth
|
from annotation.auth import Auth, hasAdmin
|
||||||
from annotation.log import Log
|
from annotation.log import Log
|
||||||
from config.constant import BusinessType, RedisKeyConfig
|
from config.constant import BusinessType, RedisKeyConfig
|
||||||
from controller.login import LoginController
|
from controller.login import LoginController
|
||||||
from models import Permission
|
from models import Permission, RolePermission
|
||||||
from schemas.common import BaseResponse
|
from schemas.common import BaseResponse
|
||||||
from schemas.permission import AddPermissionParams, GetPermissionInfoResponse, GetPermissionListResponse
|
from schemas.permission import AddPermissionParams, GetPermissionInfoResponse, GetPermissionListResponse
|
||||||
from utils.response import Response
|
from utils.response import Response
|
||||||
@ -51,6 +51,7 @@ async def add_permission(request: Request, params: AddPermissionParams):
|
|||||||
leave_transition=params.leave_transition,
|
leave_transition=params.leave_transition,
|
||||||
fixed_tag=params.fixed_tag,
|
fixed_tag=params.fixed_tag,
|
||||||
hidden_tag=params.hidden_tag,
|
hidden_tag=params.hidden_tag,
|
||||||
|
is_admin=params.is_admin
|
||||||
)
|
)
|
||||||
if permission:
|
if permission:
|
||||||
# 更新用户信息缓存
|
# 更新用户信息缓存
|
||||||
@ -72,8 +73,8 @@ async def add_permission(request: Request, params: AddPermissionParams):
|
|||||||
@Auth(permission_list=["permission:btn:delete"])
|
@Auth(permission_list=["permission:btn:delete"])
|
||||||
async def delete_permission(request: Request, id: str = Path(description="权限ID")):
|
async def delete_permission(request: Request, id: str = Path(description="权限ID")):
|
||||||
if permission := await Permission.get_or_none(id=id, del_flag=1):
|
if permission := await Permission.get_or_none(id=id, del_flag=1):
|
||||||
permission.del_flag = 0
|
# 移除角色权限
|
||||||
await permission.save()
|
await delete_permission_recursive(permission_id=permission.id)
|
||||||
# 更新用户信息缓存
|
# 更新用户信息缓存
|
||||||
userInfos = await request.app.state.redis.keys(f'{RedisKeyConfig.USER_INFO.key}*')
|
userInfos = await request.app.state.redis.keys(f'{RedisKeyConfig.USER_INFO.key}*')
|
||||||
if userInfos:
|
if userInfos:
|
||||||
@ -87,6 +88,20 @@ async def delete_permission(request: Request, id: str = Path(description="权限
|
|||||||
return Response.error(msg="删除权限失败,权限不存在!")
|
return Response.error(msg="删除权限失败,权限不存在!")
|
||||||
|
|
||||||
|
|
||||||
|
async def delete_permission_recursive(permission_id: str):
|
||||||
|
"""
|
||||||
|
递归删除权限及其附属权限
|
||||||
|
:param permission_id: 权限ID
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
await Permission.filter(id=permission_id, del_flag=1).update(del_flag=0)
|
||||||
|
await RolePermission.filter(permission_id=permission_id, del_flag=1).update(del_flag=0)
|
||||||
|
sub_permissions = await Permission.filter(parent_id=permission_id, del_flag=1).all()
|
||||||
|
for sub_department in sub_permissions:
|
||||||
|
await delete_permission_recursive(sub_department.id)
|
||||||
|
return True
|
||||||
|
|
||||||
|
|
||||||
@permissionAPI.put("/update/{id}", response_model=BaseResponse, response_class=JSONResponse, summary="更新权限")
|
@permissionAPI.put("/update/{id}", response_model=BaseResponse, response_class=JSONResponse, summary="更新权限")
|
||||||
@permissionAPI.post("/update/{id}", response_model=BaseResponse, response_class=JSONResponse, summary="更新权限")
|
@permissionAPI.post("/update/{id}", response_model=BaseResponse, response_class=JSONResponse, summary="更新权限")
|
||||||
@Log(title="更新权限", business_type=BusinessType.UPDATE)
|
@Log(title="更新权限", business_type=BusinessType.UPDATE)
|
||||||
@ -114,6 +129,7 @@ async def update_permission(request: Request, params: AddPermissionParams, id: s
|
|||||||
permission.leave_transition = params.leave_transition
|
permission.leave_transition = params.leave_transition
|
||||||
permission.fixed_tag = params.fixed_tag
|
permission.fixed_tag = params.fixed_tag
|
||||||
permission.hidden_tag = params.hidden_tag
|
permission.hidden_tag = params.hidden_tag
|
||||||
|
permission.is_admin = params.is_admin
|
||||||
await permission.save()
|
await permission.save()
|
||||||
# 更新用户信息缓存
|
# 更新用户信息缓存
|
||||||
userInfos = await request.app.state.redis.keys(f'{RedisKeyConfig.USER_INFO.key}*')
|
userInfos = await request.app.state.redis.keys(f'{RedisKeyConfig.USER_INFO.key}*')
|
||||||
@ -161,6 +177,7 @@ async def get_permission(request: Request, id: str = Path(description="权限ID"
|
|||||||
fixed_tag="fixed_tag",
|
fixed_tag="fixed_tag",
|
||||||
show_link="show_link",
|
show_link="show_link",
|
||||||
show_parent="show_parent",
|
show_parent="show_parent",
|
||||||
|
is_admin="is_admin"
|
||||||
)
|
)
|
||||||
return Response.success(msg="查询权限详情成功!", data=permission)
|
return Response.success(msg="查询权限详情成功!", data=permission)
|
||||||
else:
|
else:
|
||||||
@ -195,10 +212,12 @@ async def get_permission_list(
|
|||||||
enterTransition: Optional[str] = Query(default=None, description="进场动画"),
|
enterTransition: Optional[str] = Query(default=None, description="进场动画"),
|
||||||
leaveTransition: Optional[str] = Query(default=None, description="离场动画"),
|
leaveTransition: Optional[str] = Query(default=None, description="离场动画"),
|
||||||
fixedTag: Optional[bool] = Query(default=None, description="固定标签页"),
|
fixedTag: Optional[bool] = Query(default=None, description="固定标签页"),
|
||||||
hiddenTag: Optional[bool] = Query(default=None, description="隐藏标签页")
|
hiddenTag: Optional[bool] = Query(default=None, description="隐藏标签页"),
|
||||||
|
isAdmin: Optional[bool] = Query(default=None, description="是否为管理专属页面"),
|
||||||
|
current_user: dict = Depends(LoginController.get_current_user),
|
||||||
):
|
):
|
||||||
filterArgs = {
|
filterArgs = {
|
||||||
f'{k}__contains': v for k, v in {
|
f'{k}__icontains': v for k, v in {
|
||||||
"id": id,
|
"id": id,
|
||||||
"name": name,
|
"name": name,
|
||||||
"parent_id": parentId,
|
"parent_id": parentId,
|
||||||
@ -219,9 +238,13 @@ async def get_permission_list(
|
|||||||
"enter_transition": enterTransition,
|
"enter_transition": enterTransition,
|
||||||
"leave_transition": leaveTransition,
|
"leave_transition": leaveTransition,
|
||||||
"fixed_tag": fixedTag,
|
"fixed_tag": fixedTag,
|
||||||
"hidden_tag": hiddenTag
|
"hidden_tag": hiddenTag,
|
||||||
|
"is_admin": isAdmin
|
||||||
}.items() if v
|
}.items() if v
|
||||||
}
|
}
|
||||||
|
department_id = current_user.get("department_id", "")
|
||||||
|
if not await hasAdmin(request, department_id):
|
||||||
|
filterArgs["is_admin"] = False
|
||||||
total = await Permission.filter(**filterArgs, del_flag=1).count()
|
total = await Permission.filter(**filterArgs, del_flag=1).count()
|
||||||
result = await Permission.filter(**filterArgs, del_flag=1).offset((page - 1) * pageSize).limit(pageSize).order_by(
|
result = await Permission.filter(**filterArgs, del_flag=1).offset((page - 1) * pageSize).limit(pageSize).order_by(
|
||||||
'rank').values(
|
'rank').values(
|
||||||
@ -250,7 +273,8 @@ async def get_permission_list(
|
|||||||
hidden_tag="hidden_tag",
|
hidden_tag="hidden_tag",
|
||||||
fixed_tag="fixed_tag",
|
fixed_tag="fixed_tag",
|
||||||
show_link="show_link",
|
show_link="show_link",
|
||||||
show_parent="show_parent"
|
show_parent="show_parent",
|
||||||
|
is_admin="is_admin"
|
||||||
)
|
)
|
||||||
return Response.success(data={
|
return Response.success(data={
|
||||||
"total": total,
|
"total": total,
|
||||||
|
|||||||
28
api/role.py
28
api/role.py
@ -10,7 +10,7 @@ from typing import Optional
|
|||||||
from fastapi import APIRouter, Depends, Path, Query, Request
|
from fastapi import APIRouter, Depends, Path, Query, Request
|
||||||
from fastapi.responses import JSONResponse
|
from fastapi.responses import JSONResponse
|
||||||
|
|
||||||
from annotation.auth import Auth
|
from annotation.auth import Auth, hasAuth, hasAdmin
|
||||||
from annotation.log import Log
|
from annotation.log import Log
|
||||||
from config.constant import BusinessType, RedisKeyConfig
|
from config.constant import BusinessType, RedisKeyConfig
|
||||||
from controller.login import LoginController
|
from controller.login import LoginController
|
||||||
@ -187,15 +187,23 @@ async def get_role_list(
|
|||||||
current_user: dict = Depends(LoginController.get_current_user)
|
current_user: dict = Depends(LoginController.get_current_user)
|
||||||
):
|
):
|
||||||
filterArgs = {
|
filterArgs = {
|
||||||
f'{k}__contains': v for k, v in {
|
f'{k}__icontains': v for k, v in {
|
||||||
"name": name,
|
"name": name,
|
||||||
"code": code,
|
"code": code,
|
||||||
"description": description,
|
"description": description,
|
||||||
"status": status
|
"status": status
|
||||||
}.items() if v
|
}.items() if v
|
||||||
}
|
}
|
||||||
|
if await hasAuth(request, "role:btn:admin"):
|
||||||
if not department_id:
|
if not department_id:
|
||||||
filterArgs["department__id__in"] = current_user.get("sub_departments")
|
filterArgs["department__id__in"] = current_user.get("sub_departments")
|
||||||
|
else:
|
||||||
|
filterArgs["department__id"] = current_user.get("department_id")
|
||||||
|
else:
|
||||||
|
if department_id:
|
||||||
|
filterArgs["department__id"] = department_id
|
||||||
|
else:
|
||||||
|
filterArgs["department__id"] = current_user.get("department_id")
|
||||||
total = await Role.filter(**filterArgs, del_flag=1).count()
|
total = await Role.filter(**filterArgs, del_flag=1).count()
|
||||||
data = await Role.filter(**filterArgs, del_flag=1).offset(
|
data = await Role.filter(**filterArgs, del_flag=1).offset(
|
||||||
(page - 1) * pageSize).limit(
|
(page - 1) * pageSize).limit(
|
||||||
@ -231,6 +239,11 @@ async def add_role_permission(request: Request, params: AddRolePermissionParams,
|
|||||||
id: str = Path(..., description="角色ID"),
|
id: str = Path(..., description="角色ID"),
|
||||||
current_user: dict = Depends(LoginController.get_current_user)):
|
current_user: dict = Depends(LoginController.get_current_user)):
|
||||||
sub_departments = current_user.get("sub_departments")
|
sub_departments = current_user.get("sub_departments")
|
||||||
|
if await hasAdmin(request, current_user.get("department_id")):
|
||||||
|
department_permissions = await Permission.filter(del_flag=1).values("id")
|
||||||
|
else:
|
||||||
|
department_permissions = await Permission.filter(is_admin=False, del_flag=1).values("id")
|
||||||
|
department_permissions = filterKeyValues(department_permissions, "id")
|
||||||
if role := await Role.get_or_none(id=id, del_flag=1, department__id__in=sub_departments):
|
if role := await Role.get_or_none(id=id, del_flag=1, department__id__in=sub_departments):
|
||||||
# 已有角色权限
|
# 已有角色权限
|
||||||
rolePermissions = await RolePermission.filter(role_id=id, del_flag=1).values("permission_id")
|
rolePermissions = await RolePermission.filter(role_id=id, del_flag=1).values("permission_id")
|
||||||
@ -239,6 +252,8 @@ async def add_role_permission(request: Request, params: AddRolePermissionParams,
|
|||||||
add_list = set(params.permission_ids).difference(set(rolePermissions))
|
add_list = set(params.permission_ids).difference(set(rolePermissions))
|
||||||
# 循环添加角色权限
|
# 循环添加角色权限
|
||||||
for item in add_list:
|
for item in add_list:
|
||||||
|
if item not in department_permissions:
|
||||||
|
continue
|
||||||
permission = await Permission.get_or_none(id=item, del_flag=1)
|
permission = await Permission.get_or_none(id=item, del_flag=1)
|
||||||
if permission:
|
if permission:
|
||||||
await RolePermission.create(
|
await RolePermission.create(
|
||||||
@ -290,10 +305,15 @@ async def update_role_permission(request: Request, params: AddRolePermissionPara
|
|||||||
id: str = Path(..., description="角色ID"),
|
id: str = Path(..., description="角色ID"),
|
||||||
current_user: dict = Depends(LoginController.get_current_user)):
|
current_user: dict = Depends(LoginController.get_current_user)):
|
||||||
sub_departments = current_user.get("sub_departments")
|
sub_departments = current_user.get("sub_departments")
|
||||||
|
if await hasAdmin(request, current_user.get("department_id")):
|
||||||
|
department_permissions = await Permission.filter(del_flag=1).values("id")
|
||||||
|
else:
|
||||||
|
department_permissions = await Permission.filter(is_admin=False, del_flag=1).values("id")
|
||||||
|
department_permissions = await filterKeyValues(department_permissions, key="id", convert_type=str)
|
||||||
if role := await Role.get_or_none(id=id, del_flag=1, department__id__in=sub_departments):
|
if role := await Role.get_or_none(id=id, del_flag=1, department__id__in=sub_departments):
|
||||||
# 已有角色权限
|
# 已有角色权限
|
||||||
rolePermissions = await RolePermission.filter(role_id=role.id, del_flag=1).values("permission_id")
|
rolePermissions = await RolePermission.filter(role_id=role.id, del_flag=1).values("permission_id")
|
||||||
rolePermissions = await filterKeyValues(rolePermissions, "permission_id")
|
rolePermissions = await filterKeyValues(rolePermissions, key="permission_id", convert_type=str)
|
||||||
# 利用集合筛选出角色权限中不存在的权限
|
# 利用集合筛选出角色权限中不存在的权限
|
||||||
delete_list = set(rolePermissions).difference(set(params.permission_ids))
|
delete_list = set(rolePermissions).difference(set(params.permission_ids))
|
||||||
# 利用集合筛选出角色权限中新增的权限
|
# 利用集合筛选出角色权限中新增的权限
|
||||||
@ -303,6 +323,8 @@ async def update_role_permission(request: Request, params: AddRolePermissionPara
|
|||||||
await RolePermission.filter(role_id=id, permission_id=item, del_flag=1).update(del_flag=0)
|
await RolePermission.filter(role_id=id, permission_id=item, del_flag=1).update(del_flag=0)
|
||||||
# 循环添加角色权限
|
# 循环添加角色权限
|
||||||
for item in add_list:
|
for item in add_list:
|
||||||
|
if item not in department_permissions:
|
||||||
|
continue
|
||||||
await RolePermission.create(role_id=id, permission_id=item)
|
await RolePermission.create(role_id=id, permission_id=item)
|
||||||
# 更新用户信息缓存
|
# 更新用户信息缓存
|
||||||
userInfos = await request.app.state.redis.keys(f'{RedisKeyConfig.USER_INFO.key}*')
|
userInfos = await request.app.state.redis.keys(f'{RedisKeyConfig.USER_INFO.key}*')
|
||||||
|
|||||||
85
api/user.py
85
api/user.py
@ -10,7 +10,7 @@ import os
|
|||||||
from datetime import datetime, timedelta
|
from datetime import datetime, timedelta
|
||||||
from typing import Optional
|
from typing import Optional
|
||||||
|
|
||||||
from fastapi import APIRouter, Depends, Path, Query, UploadFile, File, Request
|
from fastapi import APIRouter, Depends, Path, Query, UploadFile, File, Request, Form
|
||||||
from fastapi.responses import JSONResponse
|
from fastapi.responses import JSONResponse
|
||||||
|
|
||||||
from annotation.auth import Auth
|
from annotation.auth import Auth
|
||||||
@ -29,7 +29,7 @@ from schemas.department import GetDepartmentListResponse
|
|||||||
from schemas.file import UploadFileResponse
|
from schemas.file import UploadFileResponse
|
||||||
from schemas.user import AddUserParams, GetUserListResponse, GetUserInfoResponse, UpdateUserParams, \
|
from schemas.user import AddUserParams, GetUserListResponse, GetUserInfoResponse, UpdateUserParams, \
|
||||||
AddUserRoleParams, GetUserRoleInfoResponse, UpdateUserRoleParams, GetUserPermissionListResponse, \
|
AddUserRoleParams, GetUserRoleInfoResponse, UpdateUserRoleParams, GetUserPermissionListResponse, \
|
||||||
ResetPasswordParams, GetUserStatisticsResponse
|
ResetPasswordParams, GetUserStatisticsResponse, UpdateBaseUserInfoParams
|
||||||
from utils.common import filterKeyValues
|
from utils.common import filterKeyValues
|
||||||
from utils.password import Password
|
from utils.password import Password
|
||||||
from utils.response import Response
|
from utils.response import Response
|
||||||
@ -184,7 +184,7 @@ async def get_user_list(
|
|||||||
):
|
):
|
||||||
sub_departments = current_user.get("sub_departments")
|
sub_departments = current_user.get("sub_departments")
|
||||||
filterArgs = {
|
filterArgs = {
|
||||||
f'{k}__contains': v for k, v in {
|
f'{k}__icontains': v for k, v in {
|
||||||
'username': username,
|
'username': username,
|
||||||
'nickname': nickname,
|
'nickname': nickname,
|
||||||
'phone': phone,
|
'phone': phone,
|
||||||
@ -438,6 +438,85 @@ async def reset_user_password(request: Request, params: ResetPasswordParams, id:
|
|||||||
return Response.failure(msg="用户不存在!")
|
return Response.failure(msg="用户不存在!")
|
||||||
|
|
||||||
|
|
||||||
|
@userAPI.put("/updateBaseUserInfo", response_model=BaseResponse, response_class=JSONResponse,
|
||||||
|
summary="更新基础个人信息")
|
||||||
|
@userAPI.post("/updateBaseUserInfo", response_model=BaseResponse, response_class=JSONResponse,
|
||||||
|
summary="更新基础个人信息")
|
||||||
|
@Log(title="更新基础个人信息", business_type=BusinessType.UPDATE)
|
||||||
|
async def update_base_userinfo(request: Request, params: UpdateBaseUserInfoParams,
|
||||||
|
current_user: dict = Depends(LoginController.get_current_user)):
|
||||||
|
user = await User.get_or_none(id=current_user.get("id"), del_flag=1)
|
||||||
|
if user:
|
||||||
|
user.nickname = params.name
|
||||||
|
user.gender = params.gender
|
||||||
|
await user.save()
|
||||||
|
if await request.app.state.redis.get(f'{RedisKeyConfig.USER_INFO.key}:{user.id}'):
|
||||||
|
await request.app.state.redis.delete(f'{RedisKeyConfig.USER_INFO.key}:{user.id}')
|
||||||
|
return Response.success(msg="更新成功!")
|
||||||
|
return Response.error(msg="更新失败!")
|
||||||
|
|
||||||
|
|
||||||
|
@userAPI.put("/updatePassword", response_class=JSONResponse, response_model=BaseResponse, summary="用户更新密码")
|
||||||
|
@userAPI.post("/updatePassword", response_class=JSONResponse, response_model=BaseResponse, summary="用户更新密码")
|
||||||
|
@Log(title="用户更新密码", business_type=BusinessType.UPDATE)
|
||||||
|
async def update_user_password(request: Request, oldPassword: str = Form(description="用户旧密码"),
|
||||||
|
newPassword: str = Form(description="用户新密码"),
|
||||||
|
current_user: dict = Depends(LoginController.get_current_user)):
|
||||||
|
if user := await User.get_or_none(id=current_user.get("id"), del_flag=1):
|
||||||
|
password = await Password.get_password_hash(oldPassword)
|
||||||
|
if user.password != password:
|
||||||
|
return Response.error(msg="旧密码错误!")
|
||||||
|
newPassword = await Password.get_password_hash(newPassword)
|
||||||
|
user.password = newPassword
|
||||||
|
await user.save()
|
||||||
|
if await request.app.state.redis.get(f'{RedisKeyConfig.USER_INFO.key}:{user.id}'):
|
||||||
|
await request.app.state.redis.delete(f'{RedisKeyConfig.USER_INFO.key}:{user.id}')
|
||||||
|
return Response.success(msg="更新成功!")
|
||||||
|
return Response.error(msg="更新失败!")
|
||||||
|
|
||||||
|
|
||||||
|
@userAPI.put("/updatePhone", response_class=JSONResponse, response_model=BaseResponse, summary="用户更新手机号")
|
||||||
|
@userAPI.post("/updatePhone", response_class=JSONResponse, response_model=BaseResponse, summary="用户更新手机号")
|
||||||
|
@Log(title="用户更新手机号", business_type=BusinessType.UPDATE)
|
||||||
|
async def update_user_phone(request: Request, password: str = Form(description="用户密码"),
|
||||||
|
phone: str = Form(description="用户手机号"),
|
||||||
|
current_user: dict = Depends(LoginController.get_current_user)):
|
||||||
|
if user := await User.get_or_none(id=current_user.get("id"), del_flag=1):
|
||||||
|
password = await Password.get_password_hash(password)
|
||||||
|
if user.password != password:
|
||||||
|
return Response.error("更改失败,请正确输入旧密码")
|
||||||
|
phoneStatus = await User.filter(phone=phone, del_flag=1).count()
|
||||||
|
if phoneStatus:
|
||||||
|
return Response.error(f"更改失败,手机号:{phone}已绑定其他账号!")
|
||||||
|
user.phone = phone
|
||||||
|
await user.save()
|
||||||
|
if await request.app.state.redis.get(f'{RedisKeyConfig.USER_INFO.key}:{user.id}'):
|
||||||
|
await request.app.state.redis.delete(f'{RedisKeyConfig.USER_INFO.key}:{user.id}')
|
||||||
|
return Response.success(msg="更新成功!")
|
||||||
|
return Response.error(msg="更新失败!")
|
||||||
|
|
||||||
|
|
||||||
|
@userAPI.put("/updateEmail", response_class=JSONResponse, response_model=BaseResponse, summary="用户更新邮箱")
|
||||||
|
@userAPI.post("/updateEmail", response_class=JSONResponse, response_model=BaseResponse, summary="用户更新邮箱")
|
||||||
|
@Log(title="用户更新邮箱", business_type=BusinessType.UPDATE)
|
||||||
|
async def update_user_email(request: Request, password: str = Form(description="用户密码"),
|
||||||
|
email: str = Form(description="用户邮箱"),
|
||||||
|
current_user: dict = Depends(LoginController.get_current_user)):
|
||||||
|
if user := await User.get_or_none(id=current_user.get("id"), del_flag=1):
|
||||||
|
password = await Password.get_password_hash(password)
|
||||||
|
if user.password != password:
|
||||||
|
return Response.error("更改失败,请正确输入旧密码")
|
||||||
|
emailStatus = await User.filter(email=email, del_flag=1).count()
|
||||||
|
if emailStatus:
|
||||||
|
return Response.error(f"更改失败,邮箱:{email}已绑定其他账号!")
|
||||||
|
user.email = email
|
||||||
|
await user.save()
|
||||||
|
if await request.app.state.redis.get(f'{RedisKeyConfig.USER_INFO.key}:{user.id}'):
|
||||||
|
await request.app.state.redis.delete(f'{RedisKeyConfig.USER_INFO.key}:{user.id}')
|
||||||
|
return Response.success(msg="更新成功!")
|
||||||
|
return Response.error(msg="更新失败!")
|
||||||
|
|
||||||
|
|
||||||
@userAPI.get("/statistics", response_model=GetUserStatisticsResponse, response_class=JSONResponse,
|
@userAPI.get("/statistics", response_model=GetUserStatisticsResponse, response_class=JSONResponse,
|
||||||
summary="获取用户查询统计")
|
summary="获取用户查询统计")
|
||||||
@Log(title="获取用户查询统计", business_type=BusinessType.SELECT)
|
@Log(title="获取用户查询统计", business_type=BusinessType.SELECT)
|
||||||
|
|||||||
2
app.py
2
app.py
@ -24,6 +24,7 @@ from api.role import roleAPI
|
|||||||
from api.server import serverAPI
|
from api.server import serverAPI
|
||||||
from api.user import userAPI
|
from api.user import userAPI
|
||||||
from api.code import codeAPI
|
from api.code import codeAPI
|
||||||
|
from api.hts import htsAPI
|
||||||
from config.database import init_db, close_db
|
from config.database import init_db, close_db
|
||||||
from config.env import AppConfig
|
from config.env import AppConfig
|
||||||
from config.get_ElasticSearch import ElasticSearch
|
from config.get_ElasticSearch import ElasticSearch
|
||||||
@ -93,6 +94,7 @@ api_list = [
|
|||||||
{'api': i18nAPI, 'tags': ['国际化管理']},
|
{'api': i18nAPI, 'tags': ['国际化管理']},
|
||||||
{'api': configApi, 'tags': ['配置管理']},
|
{'api': configApi, 'tags': ['配置管理']},
|
||||||
{'api': codeAPI, 'tags': ['编码管理']},
|
{'api': codeAPI, 'tags': ['编码管理']},
|
||||||
|
{'api': htsAPI, 'tags': ['海关税率管理']},
|
||||||
]
|
]
|
||||||
|
|
||||||
for api in api_list:
|
for api in api_list:
|
||||||
|
|||||||
13
assets/templates/上传模版.csv
Normal file
13
assets/templates/上传模版.csv
Normal file
@ -0,0 +1,13 @@
|
|||||||
|
code,description
|
||||||
|
01012100,Live purebred breeding horses
|
||||||
|
01012900,Live horses other than purebred breeding horses
|
||||||
|
01013000,Live asses
|
||||||
|
01019030,Mules and hinnies imported for immediate slaughter
|
||||||
|
01019040,Mules and hinnies not imported for immediate slaughter
|
||||||
|
01022100,Live purebred breeding cattle
|
||||||
|
01022920,Cows imported specially for dairy purposes
|
||||||
|
01022940,Live cattle other than purebred or those imported for dairy purposes
|
||||||
|
01023100,Live purebred breeding buffalo
|
||||||
|
01023900,"Live buffalo, other than purebred breeding animals"
|
||||||
|
01029000,"Live bovine animals, other than cattle and buffalo"
|
||||||
|
01031000,Live purebred breeding swine
|
||||||
|
BIN
assets/templates/上传模版.xls
Normal file
BIN
assets/templates/上传模版.xls
Normal file
Binary file not shown.
18
assets/templates/查询模版.csv
Normal file
18
assets/templates/查询模版.csv
Normal file
@ -0,0 +1,18 @@
|
|||||||
|
text
|
||||||
|
horses
|
||||||
|
Live purebred breeding cattle
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
BIN
assets/templates/查询模版.xls
Normal file
BIN
assets/templates/查询模版.xls
Normal file
Binary file not shown.
@ -71,68 +71,68 @@ async def configure_tortoise_logging(enable_logging: bool = True, log_level: int
|
|||||||
if aiomysql_logger.hasHandlers():
|
if aiomysql_logger.hasHandlers():
|
||||||
aiomysql_logger.handlers.clear()
|
aiomysql_logger.handlers.clear()
|
||||||
|
|
||||||
if enable_logging:
|
# if enable_logging:
|
||||||
# 设置日志格式
|
# # 设置日志格式
|
||||||
fmt = logging.Formatter(
|
# fmt = logging.Formatter(
|
||||||
fmt="%(asctime)s - %(name)s:%(lineno)d - %(levelname)s - %(message)s",
|
# fmt="%(asctime)s - %(name)s:%(lineno)d - %(levelname)s - %(message)s",
|
||||||
datefmt="%Y-%m-%d %H:%M:%S",
|
# datefmt="%Y-%m-%d %H:%M:%S",
|
||||||
)
|
# )
|
||||||
|
#
|
||||||
# 创建控制台处理器(输出到控制台)
|
# # 创建控制台处理器(输出到控制台)
|
||||||
console_handler = logging.StreamHandler(sys.stdout)
|
# console_handler = logging.StreamHandler(sys.stdout)
|
||||||
console_handler.setLevel(log_level)
|
# console_handler.setLevel(log_level)
|
||||||
console_handler.setFormatter(fmt)
|
# console_handler.setFormatter(fmt)
|
||||||
|
#
|
||||||
# 创建文件处理器(输出到文件)
|
# # 创建文件处理器(输出到文件)
|
||||||
file_handler = RotatingFileHandler(
|
# file_handler = RotatingFileHandler(
|
||||||
filename=log_path_sql,
|
# filename=log_path_sql,
|
||||||
maxBytes=50 * 1024 * 1024, # 日志文件大小达到 50MB 时轮换
|
# maxBytes=50 * 1024 * 1024, # 日志文件大小达到 50MB 时轮换
|
||||||
backupCount=5, # 保留 5 个旧日志文件
|
# backupCount=5, # 保留 5 个旧日志文件
|
||||||
encoding="utf-8",
|
# encoding="utf-8",
|
||||||
)
|
# )
|
||||||
file_handler.setLevel(log_level)
|
# file_handler.setLevel(log_level)
|
||||||
file_handler.setFormatter(fmt)
|
# file_handler.setFormatter(fmt)
|
||||||
|
#
|
||||||
# 配置 tortoise 顶级日志记录器
|
# # 配置 tortoise 顶级日志记录器
|
||||||
tortoise_logger.setLevel(log_level)
|
# tortoise_logger.setLevel(log_level)
|
||||||
tortoise_logger.addHandler(console_handler) # 添加控制台处理器
|
# tortoise_logger.addHandler(console_handler) # 添加控制台处理器
|
||||||
tortoise_logger.addHandler(file_handler) # 添加文件处理器
|
# tortoise_logger.addHandler(file_handler) # 添加文件处理器
|
||||||
|
#
|
||||||
# 配置 aiomysql 日志记录器
|
# # 配置 aiomysql 日志记录器
|
||||||
aiomysql_logger.setLevel(log_level)
|
# aiomysql_logger.setLevel(log_level)
|
||||||
aiomysql_logger.addHandler(console_handler) # 添加控制台处理器
|
# aiomysql_logger.addHandler(console_handler) # 添加控制台处理器
|
||||||
aiomysql_logger.addHandler(file_handler) # 添加文件处理器
|
# aiomysql_logger.addHandler(file_handler) # 添加文件处理器
|
||||||
# 配置 SQL 查询日志记录器
|
# # 配置 SQL 查询日志记录器
|
||||||
sql_logger = logging.getLogger("tortoise.db_client")
|
# sql_logger = logging.getLogger("tortoise.db_client")
|
||||||
sql_logger.setLevel(log_level)
|
# sql_logger.setLevel(log_level)
|
||||||
|
#
|
||||||
class SQLResultLogger(logging.Handler):
|
# class SQLResultLogger(logging.Handler):
|
||||||
async def emit(self, record):
|
# async def emit(self, record):
|
||||||
# 只处理 SQL 查询相关的日志
|
# # 只处理 SQL 查询相关的日志
|
||||||
if "SELECT" in record.getMessage() or "INSERT" in record.getMessage() or "UPDATE" in record.getMessage() or "DELETE" in record.getMessage():
|
# if "SELECT" in record.getMessage() or "INSERT" in record.getMessage() or "UPDATE" in record.getMessage() or "DELETE" in record.getMessage():
|
||||||
# 输出 SQL 查询语句
|
# # 输出 SQL 查询语句
|
||||||
console_handler.emit(record)
|
# console_handler.emit(record)
|
||||||
file_handler.emit(record)
|
# file_handler.emit(record)
|
||||||
|
#
|
||||||
# 异步获取并记录查询结果
|
# # 异步获取并记录查询结果
|
||||||
await self.log_query_result(record)
|
# await self.log_query_result(record)
|
||||||
|
#
|
||||||
async def log_query_result(self, record):
|
# async def log_query_result(self, record):
|
||||||
"""
|
# """
|
||||||
执行查询并返回结果。
|
# 执行查询并返回结果。
|
||||||
"""
|
# """
|
||||||
try:
|
# try:
|
||||||
from tortoise import Tortoise
|
# from tortoise import Tortoise
|
||||||
connection = Tortoise.get_connection("default")
|
# connection = Tortoise.get_connection("default")
|
||||||
result = await connection.execute_query_dict(record.getMessage())
|
# result = await connection.execute_query_dict(record.getMessage())
|
||||||
return result
|
# return result
|
||||||
except Exception as e:
|
# except Exception as e:
|
||||||
return f"获取查询结果失败: {str(e)}"
|
# return f"获取查询结果失败: {str(e)}"
|
||||||
|
#
|
||||||
# 添加自定义 SQL 查询日志处理器
|
# # 添加自定义 SQL 查询日志处理器
|
||||||
sql_result_handler = SQLResultLogger()
|
# sql_result_handler = SQLResultLogger()
|
||||||
sql_result_handler.setLevel(log_level)
|
# sql_result_handler.setLevel(log_level)
|
||||||
sql_logger.addHandler(sql_result_handler)
|
# sql_logger.addHandler(sql_result_handler)
|
||||||
else:
|
# else:
|
||||||
# 如果禁用日志,设置日志级别为 WARNING 以抑制大部分输出
|
# # 如果禁用日志,设置日志级别为 WARNING 以抑制大部分输出
|
||||||
tortoise_logger.setLevel(logging.WARNING)
|
# tortoise_logger.setLevel(logging.WARNING)
|
||||||
|
|||||||
@ -313,6 +313,7 @@ class UploadSettings:
|
|||||||
'doc',
|
'doc',
|
||||||
'docx',
|
'docx',
|
||||||
'xls',
|
'xls',
|
||||||
|
'csv',
|
||||||
'xlsx',
|
'xlsx',
|
||||||
'ppt',
|
'ppt',
|
||||||
'pptx',
|
'pptx',
|
||||||
@ -330,6 +331,7 @@ class UploadSettings:
|
|||||||
'rmvb',
|
'rmvb',
|
||||||
# pdf
|
# pdf
|
||||||
'pdf',
|
'pdf',
|
||||||
|
'json'
|
||||||
]
|
]
|
||||||
"""
|
"""
|
||||||
默认允许上传的文件扩展名列表。
|
默认允许上传的文件扩展名列表。
|
||||||
|
|||||||
@ -132,7 +132,8 @@ class QueryController:
|
|||||||
keepAlive="permission__keep_alive",
|
keepAlive="permission__keep_alive",
|
||||||
hiddenTag="permission__hidden_tag",
|
hiddenTag="permission__hidden_tag",
|
||||||
showLink="permission__show_link",
|
showLink="permission__show_link",
|
||||||
showParent="permission__show_parent"
|
showParent="permission__show_parent",
|
||||||
|
isAdmin="permission__is_admin",
|
||||||
)
|
)
|
||||||
permissions.extend(permission)
|
permissions.extend(permission)
|
||||||
return permissions
|
return permissions
|
||||||
|
|||||||
@ -6,7 +6,7 @@
|
|||||||
# @Software : PyCharm
|
# @Software : PyCharm
|
||||||
# @Comment : 本程序
|
# @Comment : 本程序
|
||||||
|
|
||||||
from models.code import Code, QueryCode, QueryCodeLog
|
from models.code import Code, QueryCode, QueryCodeLog, CodeFeedback, CodeImport
|
||||||
from models.config import Config
|
from models.config import Config
|
||||||
from models.department import Department
|
from models.department import Department
|
||||||
from models.file import File
|
from models.file import File
|
||||||
@ -15,6 +15,7 @@ from models.log import LoginLog, OperationLog
|
|||||||
from models.permission import Permission
|
from models.permission import Permission
|
||||||
from models.role import Role, RolePermission
|
from models.role import Role, RolePermission
|
||||||
from models.user import User, UserRole
|
from models.user import User, UserRole
|
||||||
|
from models.hts import Version, HtsClass, HtsItem
|
||||||
|
|
||||||
__all__ = [
|
__all__ = [
|
||||||
'Department',
|
'Department',
|
||||||
@ -30,6 +31,11 @@ __all__ = [
|
|||||||
'Locale',
|
'Locale',
|
||||||
'Config',
|
'Config',
|
||||||
'Code',
|
'Code',
|
||||||
|
'CodeFeedback',
|
||||||
|
'CodeImport',
|
||||||
'QueryCode',
|
'QueryCode',
|
||||||
'QueryCodeLog'
|
'QueryCodeLog',
|
||||||
|
'HtsClass',
|
||||||
|
'HtsItem',
|
||||||
|
'Version',
|
||||||
]
|
]
|
||||||
|
|||||||
@ -24,6 +24,12 @@ class Code(BaseModel):
|
|||||||
description="描述",
|
description="描述",
|
||||||
source_field="description"
|
source_field="description"
|
||||||
)
|
)
|
||||||
|
user = fields.ForeignKeyField(
|
||||||
|
"models.User",
|
||||||
|
related_name="codes",
|
||||||
|
description="用户",
|
||||||
|
source_field="user_id"
|
||||||
|
)
|
||||||
|
|
||||||
class Meta:
|
class Meta:
|
||||||
table = "code"
|
table = "code"
|
||||||
@ -137,6 +143,11 @@ class QueryCode(BaseModel):
|
|||||||
"""
|
"""
|
||||||
查询编码模型
|
查询编码模型
|
||||||
"""
|
"""
|
||||||
|
id=fields.UUIDField(
|
||||||
|
pk=True,
|
||||||
|
description="主键",
|
||||||
|
source_field="id"
|
||||||
|
)
|
||||||
session = fields.ForeignKeyField(
|
session = fields.ForeignKeyField(
|
||||||
"models.QueryCodeLog",
|
"models.QueryCodeLog",
|
||||||
related_name="query_code",
|
related_name="query_code",
|
||||||
@ -160,3 +171,70 @@ class QueryCode(BaseModel):
|
|||||||
class Meta:
|
class Meta:
|
||||||
table = "query_code"
|
table = "query_code"
|
||||||
table_description = "查询编码表"
|
table_description = "查询编码表"
|
||||||
|
|
||||||
|
|
||||||
|
class CodeFeedback(BaseModel):
|
||||||
|
"""
|
||||||
|
编码反馈模型
|
||||||
|
"""
|
||||||
|
code = fields.ForeignKeyField(
|
||||||
|
"models.Code",
|
||||||
|
related_name="feedbacks",
|
||||||
|
description="编码",
|
||||||
|
null=True,
|
||||||
|
source_field="code_id"
|
||||||
|
)
|
||||||
|
user = fields.ForeignKeyField(
|
||||||
|
"models.User",
|
||||||
|
related_name="feedbacks",
|
||||||
|
description="用户",
|
||||||
|
source_field="user_id"
|
||||||
|
)
|
||||||
|
feedback_code = fields.CharField(
|
||||||
|
max_length=255,
|
||||||
|
description="反馈编码",
|
||||||
|
source_field="feedback_code"
|
||||||
|
)
|
||||||
|
feedback_description = fields.TextField(
|
||||||
|
description="反馈内容",
|
||||||
|
source_field="feedback_description"
|
||||||
|
)
|
||||||
|
status = fields.SmallIntField(
|
||||||
|
default=3,
|
||||||
|
description="状态,0删除,1审核通过,2审核不通过,3待审核)",
|
||||||
|
source_field="status"
|
||||||
|
)
|
||||||
|
|
||||||
|
class Meta:
|
||||||
|
table = "code_feedback"
|
||||||
|
table_description = "编码反馈表"
|
||||||
|
|
||||||
|
|
||||||
|
class CodeImport(BaseModel):
|
||||||
|
"""
|
||||||
|
编码导入模型
|
||||||
|
"""
|
||||||
|
code = fields.CharField(
|
||||||
|
max_length=255,
|
||||||
|
description="编码",
|
||||||
|
source_field="code"
|
||||||
|
)
|
||||||
|
description = fields.TextField(
|
||||||
|
description="描述",
|
||||||
|
source_field="description"
|
||||||
|
)
|
||||||
|
user = fields.ForeignKeyField(
|
||||||
|
"models.User",
|
||||||
|
related_name="imports",
|
||||||
|
description="用户",
|
||||||
|
source_field="user_id"
|
||||||
|
)
|
||||||
|
status = fields.SmallIntField(
|
||||||
|
default=1,
|
||||||
|
description="状态,0删除,1审核通过,2审核不通过,3待审核)",
|
||||||
|
source_field="status"
|
||||||
|
)
|
||||||
|
|
||||||
|
class Meta:
|
||||||
|
table = "code_import"
|
||||||
|
table_description = "编码导入表"
|
||||||
|
|||||||
165
models/hts.py
Normal file
165
models/hts.py
Normal file
@ -0,0 +1,165 @@
|
|||||||
|
# _*_ coding : UTF-8 _*_
|
||||||
|
# @Time : 2025/03/24 02:29:03
|
||||||
|
# @UpdateTime : 2025/03/24 02:29:03
|
||||||
|
# @Author : sonder
|
||||||
|
# @File : hts.py
|
||||||
|
# @Comment : 本程序用于海关编码税率模型
|
||||||
|
|
||||||
|
from tortoise import fields
|
||||||
|
|
||||||
|
from models.common import BaseModel
|
||||||
|
|
||||||
|
|
||||||
|
class Version(BaseModel):
|
||||||
|
"""
|
||||||
|
版本信息模型
|
||||||
|
"""
|
||||||
|
|
||||||
|
version = fields.CharField(max_length=255, description="版本号", source_field="version")
|
||||||
|
"""
|
||||||
|
版本号。
|
||||||
|
- 最大长度为 255 个字符
|
||||||
|
- 映射到数据库字段 version
|
||||||
|
"""
|
||||||
|
|
||||||
|
date = fields.CharField(max_length=255,description="版本日期", source_field="date")
|
||||||
|
"""
|
||||||
|
版本日期。
|
||||||
|
- 映射到数据库字段 date
|
||||||
|
"""
|
||||||
|
|
||||||
|
url = fields.CharField(max_length=255, description="下载地址", source_field="url")
|
||||||
|
"""
|
||||||
|
下载地址。
|
||||||
|
- 最大长度为 255 个字符
|
||||||
|
- 映射到数据库字段 url
|
||||||
|
"""
|
||||||
|
|
||||||
|
class Meta:
|
||||||
|
table = "version"
|
||||||
|
table_description = "版本信息"
|
||||||
|
|
||||||
|
|
||||||
|
class HtsClass(BaseModel):
|
||||||
|
"""
|
||||||
|
编码类信息模型
|
||||||
|
"""
|
||||||
|
|
||||||
|
class_name = fields.TextField(description="类名", source_field="class_name")
|
||||||
|
"""
|
||||||
|
类名。
|
||||||
|
- 映射到数据库字段 class_name
|
||||||
|
"""
|
||||||
|
|
||||||
|
class_description = fields.TextField(description="类描述", source_field="class_description")
|
||||||
|
"""
|
||||||
|
类描述。
|
||||||
|
- 映射到数据库字段 class_description
|
||||||
|
"""
|
||||||
|
|
||||||
|
chapter_name = fields.TextField(description="章节", source_field="chapter_name")
|
||||||
|
"""
|
||||||
|
章节。
|
||||||
|
- 映射到数据库字段 chapter_name
|
||||||
|
"""
|
||||||
|
|
||||||
|
chapter_description = fields.TextField(description="章节描述", source_field="chapter_description")
|
||||||
|
"""
|
||||||
|
章节描述。
|
||||||
|
- 映射到数据库字段 chapter_description
|
||||||
|
"""
|
||||||
|
|
||||||
|
class Meta:
|
||||||
|
table = "hts_class"
|
||||||
|
table_description = "编码类信息"
|
||||||
|
|
||||||
|
|
||||||
|
class HtsItem(BaseModel):
|
||||||
|
"""
|
||||||
|
编码项信息模型
|
||||||
|
"""
|
||||||
|
|
||||||
|
parent_id = fields.CharField(max_length=255, null=True, description="父编码", source_field="parent_id")
|
||||||
|
"""
|
||||||
|
父编码。
|
||||||
|
- 最大长度为 255 个字符
|
||||||
|
- 映射到数据库字段 parent_id
|
||||||
|
- 可为空
|
||||||
|
"""
|
||||||
|
|
||||||
|
htsno = fields.CharField(max_length=255, description="编码", source_field="htsno")
|
||||||
|
"""
|
||||||
|
编码。
|
||||||
|
- 最大长度为 255 个字符
|
||||||
|
- 映射到数据库字段 htsno
|
||||||
|
"""
|
||||||
|
|
||||||
|
indent = fields.IntField(description="层级",null=True, source_field="indent")
|
||||||
|
"""
|
||||||
|
缩进。
|
||||||
|
- 映射到数据库字段 indent
|
||||||
|
"""
|
||||||
|
|
||||||
|
description = fields.TextField(description="描述",null=True, source_field="description")
|
||||||
|
"""
|
||||||
|
描述。
|
||||||
|
- 映射到数据库字段 description
|
||||||
|
"""
|
||||||
|
|
||||||
|
units = fields.JSONField(description="单位列表", null=True, source_field="units")
|
||||||
|
"""
|
||||||
|
单位列表。
|
||||||
|
- 映射到数据库字段 units
|
||||||
|
"""
|
||||||
|
|
||||||
|
general = fields.TextField(description="通用税率",null=True, source_field="general")
|
||||||
|
"""
|
||||||
|
通用税率。
|
||||||
|
- 最大长度为 50 个字符
|
||||||
|
- 映射到数据库字段 general
|
||||||
|
"""
|
||||||
|
|
||||||
|
special = fields.TextField(description="特殊税率,适用于特定国家或地区",null=True, source_field="special")
|
||||||
|
"""
|
||||||
|
特殊税率,适用于特定国家或地区。
|
||||||
|
- 最大长度为 255 个字符
|
||||||
|
- 映射到数据库字段 special
|
||||||
|
"""
|
||||||
|
|
||||||
|
other = fields.TextField(description="其他税率",null=True, source_field="other")
|
||||||
|
"""
|
||||||
|
其他税率。
|
||||||
|
- 最大长度为 50 个字符
|
||||||
|
- 映射到数据库字段 other
|
||||||
|
"""
|
||||||
|
|
||||||
|
quota_quantity = fields.TextField(description="配额数量", null=True, source_field="quota_quantity")
|
||||||
|
"""
|
||||||
|
配额数量。
|
||||||
|
- 最大长度为 50 个字符
|
||||||
|
- 映射到数据库字段 quota_quantity
|
||||||
|
"""
|
||||||
|
|
||||||
|
additional_duties = fields.TextField(description="附加税",null=True, source_field="additional_duties")
|
||||||
|
"""
|
||||||
|
附加税。
|
||||||
|
- 最大长度为 50 个字符
|
||||||
|
- 映射到数据库字段 additional_duties
|
||||||
|
"""
|
||||||
|
|
||||||
|
footnotes = fields.JSONField(null=True, description="脚注列表", source_field="footnotes")
|
||||||
|
"""
|
||||||
|
脚注列表。
|
||||||
|
- 映射到数据库字段 footnotes
|
||||||
|
- 可为空
|
||||||
|
"""
|
||||||
|
|
||||||
|
class_ = fields.ForeignKeyField("models.HtsClass", related_name="class_items", on_delete=fields.CASCADE,
|
||||||
|
description="所属类", source_field="class_id")
|
||||||
|
|
||||||
|
version = fields.ForeignKeyField("models.Version", related_name="version_items", on_delete=fields.CASCADE,
|
||||||
|
description="所属版本", source_field="version_id")
|
||||||
|
|
||||||
|
class Meta:
|
||||||
|
table = "hts_item"
|
||||||
|
table_description = "编码项信息"
|
||||||
@ -280,6 +280,18 @@ class Permission(BaseModel):
|
|||||||
- 映射到数据库字段 show_parent。
|
- 映射到数据库字段 show_parent。
|
||||||
"""
|
"""
|
||||||
|
|
||||||
|
is_admin = fields.BooleanField(
|
||||||
|
default=False,
|
||||||
|
description="是否为管理专属页面",
|
||||||
|
source_field="is_admin" # 映射到数据库字段 is_admin
|
||||||
|
)
|
||||||
|
"""
|
||||||
|
是否为管理专属页面。
|
||||||
|
- 是否为管理专属页面,仅管理员可见。
|
||||||
|
- 默认为 False。
|
||||||
|
- 映射到数据库字段 is_admin。
|
||||||
|
"""
|
||||||
|
|
||||||
class Meta:
|
class Meta:
|
||||||
table = "permission" # 数据库表名
|
table = "permission" # 数据库表名
|
||||||
table_description = "权限表" # 表描述
|
table_description = "权限表" # 表描述
|
||||||
|
|||||||
114
schemas/code.py
114
schemas/code.py
@ -28,6 +28,11 @@ class CodeInfo(BaseModel):
|
|||||||
update_time: Optional[datetime] = Field(default=None, description="更新时间")
|
update_time: Optional[datetime] = Field(default=None, description="更新时间")
|
||||||
code: str = Field(..., description="编码")
|
code: str = Field(..., description="编码")
|
||||||
description: str = Field(..., description="描述")
|
description: str = Field(..., description="描述")
|
||||||
|
user_id: str = Field(default="", description="用户ID")
|
||||||
|
username: str = Field(default="", description="用户名称")
|
||||||
|
nickname: str = Field(default="", description="用户昵称")
|
||||||
|
department_id: str = Field(default="", description="用户部门ID")
|
||||||
|
department_name: str = Field(default="", description="用户部门名称")
|
||||||
|
|
||||||
|
|
||||||
class AddCodeParams(BaseModel):
|
class AddCodeParams(BaseModel):
|
||||||
@ -176,3 +181,112 @@ class GetCodeLogAllResponse(BaseResponse):
|
|||||||
查询所有编码日志响应
|
查询所有编码日志响应
|
||||||
"""
|
"""
|
||||||
data: GetCodeLogAllResult = Field(..., description="查询编码日志结果")
|
data: GetCodeLogAllResult = Field(..., description="查询编码日志结果")
|
||||||
|
|
||||||
|
|
||||||
|
class CodeFeedbackInfo(BaseModel):
|
||||||
|
"""
|
||||||
|
编码反馈信息
|
||||||
|
"""
|
||||||
|
model_config = ConfigDict(
|
||||||
|
alias_generator=to_camel
|
||||||
|
)
|
||||||
|
id: str = Field(default="", description="主键")
|
||||||
|
code_id: str = Field(default="", description="编码ID")
|
||||||
|
code: str = Field(default="", description="编码")
|
||||||
|
description: str = Field(default="", description="描述")
|
||||||
|
feedback_code: str = Field(default="", description="反馈编码")
|
||||||
|
feedback_description: str = Field(default="", description="反馈文本")
|
||||||
|
status: int = Field(default=3, description="反馈状态")
|
||||||
|
user_id: str = Field(default="", description="用户ID")
|
||||||
|
username: str = Field(default="", description="用户名称")
|
||||||
|
nickname: str = Field(default="", description="用户昵称")
|
||||||
|
department_id: str = Field(default="", description="用户部门ID")
|
||||||
|
department_name: str = Field(default="", description="用户部门名称")
|
||||||
|
create_time: str = Field(default="", description="创建时间")
|
||||||
|
update_time: str = Field(default="", description="更新时间")
|
||||||
|
|
||||||
|
|
||||||
|
class GetCodeFeedbackResponse(BaseResponse):
|
||||||
|
"""
|
||||||
|
获取编码结果
|
||||||
|
"""
|
||||||
|
data: CodeFeedbackInfo = Field(default={}, description="编码反馈结果")
|
||||||
|
|
||||||
|
|
||||||
|
class AddCodeFeedbackParams(BaseModel):
|
||||||
|
"""
|
||||||
|
添加编码反馈参数
|
||||||
|
"""
|
||||||
|
code_id: str = Field(..., description="编码ID")
|
||||||
|
feedback_code: str = Field(..., description="反馈编码")
|
||||||
|
feedback_description: str = Field(..., description="反馈文本")
|
||||||
|
|
||||||
|
|
||||||
|
class CodeFeedbackResult(ListQueryResult):
|
||||||
|
"""
|
||||||
|
编码反馈结果
|
||||||
|
"""
|
||||||
|
result: List[CodeFeedbackInfo] = Field(..., description="编码反馈列表")
|
||||||
|
|
||||||
|
|
||||||
|
class GetCodeFeedbackListResponse(BaseResponse):
|
||||||
|
"""
|
||||||
|
获取编码反馈响应
|
||||||
|
"""
|
||||||
|
data: CodeFeedbackResult = Field(..., description="编码反馈结果")
|
||||||
|
|
||||||
|
|
||||||
|
class UpdateCodeFeedbackStatusParams(BaseModel):
|
||||||
|
"""
|
||||||
|
更新编码反馈参数
|
||||||
|
"""
|
||||||
|
status: int = Field(..., description="状态")
|
||||||
|
ids: List[str] = Field(default=[], description="编码反馈ID")
|
||||||
|
|
||||||
|
|
||||||
|
class CodeImportInfo(BaseModel):
|
||||||
|
"""
|
||||||
|
编码导入信息
|
||||||
|
"""
|
||||||
|
model_config = ConfigDict(
|
||||||
|
alias_generator=to_camel
|
||||||
|
)
|
||||||
|
code: str = Field(default="", description="编码")
|
||||||
|
description: str = Field(default="", description="描述")
|
||||||
|
status: int = Field(default=3, description="反馈状态")
|
||||||
|
user_id: str = Field(default="", description="用户ID")
|
||||||
|
username: str = Field(default="", description="用户名称")
|
||||||
|
nickname: str = Field(default="", description="用户昵称")
|
||||||
|
department_id: str = Field(default="", description="用户部门ID")
|
||||||
|
department_name: str = Field(default="", description="用户部门名称")
|
||||||
|
create_time: str = Field(default="", description="创建时间")
|
||||||
|
update_time: str = Field(default="", description="更新时间")
|
||||||
|
|
||||||
|
|
||||||
|
class GetCodeImportResponse(BaseResponse):
|
||||||
|
"""
|
||||||
|
获取编码导入响应
|
||||||
|
"""
|
||||||
|
data: CodeImportInfo = Field(default={}, description="编码导入结果")
|
||||||
|
|
||||||
|
|
||||||
|
class CodeImportResult(ListQueryResult):
|
||||||
|
"""
|
||||||
|
编码导入结果
|
||||||
|
"""
|
||||||
|
result: List[CodeImportInfo] = Field(..., description="编码导入列表")
|
||||||
|
|
||||||
|
|
||||||
|
class GetCodeImportListResponse(BaseResponse):
|
||||||
|
"""
|
||||||
|
获取编码导入响应
|
||||||
|
"""
|
||||||
|
data: CodeImportResult = Field(..., description="编码导入结果")
|
||||||
|
|
||||||
|
|
||||||
|
class UpdateCodeImportStatusParams(BaseModel):
|
||||||
|
"""
|
||||||
|
更新编码反馈参数
|
||||||
|
"""
|
||||||
|
status: int = Field(..., description="状态")
|
||||||
|
ids: List[str] = Field(default=[], description="编码反馈ID")
|
||||||
|
|||||||
391
schemas/hts.py
Normal file
391
schemas/hts.py
Normal file
@ -0,0 +1,391 @@
|
|||||||
|
# _*_ coding : UTF-8 _*_
|
||||||
|
# @Time : 2025/03/24 02:29:03
|
||||||
|
# @UpdateTime : 2025/03/24 02:29:03
|
||||||
|
# @Author : sonder
|
||||||
|
# @File : version.py
|
||||||
|
# @Comment : 本程序用于生成版本信息参数和响应模型
|
||||||
|
|
||||||
|
from datetime import datetime
|
||||||
|
from typing import Optional, List
|
||||||
|
from pydantic import BaseModel, Field, ConfigDict
|
||||||
|
from pydantic.alias_generators import to_snake
|
||||||
|
from schemas.common import BaseResponse, ListQueryResult
|
||||||
|
|
||||||
|
|
||||||
|
class VersionInfo(BaseModel):
|
||||||
|
"""税率编码版本信息"""
|
||||||
|
model_config = ConfigDict(alias_generator=to_snake, populate_by_name=True)
|
||||||
|
|
||||||
|
id: Optional[str] = Field(
|
||||||
|
title="主键"
|
||||||
|
)
|
||||||
|
|
||||||
|
create_time: Optional[datetime] = Field(
|
||||||
|
title="创建时间"
|
||||||
|
)
|
||||||
|
|
||||||
|
update_time: Optional[datetime] = Field(
|
||||||
|
title="更新时间"
|
||||||
|
)
|
||||||
|
|
||||||
|
version: str = Field(
|
||||||
|
title="版本号"
|
||||||
|
)
|
||||||
|
|
||||||
|
date: str = Field(
|
||||||
|
title="版本日期"
|
||||||
|
)
|
||||||
|
|
||||||
|
url: str = Field(
|
||||||
|
title="下载地址"
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
class AddVersionParams(BaseModel):
|
||||||
|
"""新增税率编码版本参数"""
|
||||||
|
model_config = ConfigDict(alias_generator=to_snake, populate_by_name=True)
|
||||||
|
|
||||||
|
version: str = Field(
|
||||||
|
title="版本号"
|
||||||
|
)
|
||||||
|
|
||||||
|
date: str = Field(
|
||||||
|
title="版本日期"
|
||||||
|
)
|
||||||
|
|
||||||
|
url: str = Field(
|
||||||
|
title="下载地址"
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
class UpdateVersionParams(BaseModel):
|
||||||
|
"""更新税率编码版本参数"""
|
||||||
|
model_config = ConfigDict(alias_generator=to_snake, populate_by_name=True)
|
||||||
|
|
||||||
|
version: str = Field(
|
||||||
|
title="版本号"
|
||||||
|
)
|
||||||
|
|
||||||
|
date: str = Field(
|
||||||
|
title="版本日期"
|
||||||
|
)
|
||||||
|
|
||||||
|
url: str = Field(
|
||||||
|
title="下载地址"
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
class GetVersionInfoResponse(BaseResponse):
|
||||||
|
"""获取税率编码版本信息响应"""
|
||||||
|
data: VersionInfo = Field(None, title="版本信息信息")
|
||||||
|
|
||||||
|
|
||||||
|
class GetVersionInfoListResult(ListQueryResult):
|
||||||
|
"""获取税率编码版本信息列表响应结果"""
|
||||||
|
result: List[VersionInfo] = Field(None, title="版本信息信息列表")
|
||||||
|
|
||||||
|
|
||||||
|
class GetVersionListResponse(BaseResponse):
|
||||||
|
"""获取税率编码版本信息列表响应"""
|
||||||
|
data: GetVersionInfoListResult = Field(None, title="版本信息信息列表")
|
||||||
|
|
||||||
|
|
||||||
|
class HTSClassInfo(BaseModel):
|
||||||
|
"""编码类别信息"""
|
||||||
|
model_config = ConfigDict(alias_generator=to_snake, populate_by_name=True)
|
||||||
|
|
||||||
|
id: Optional[str] = Field(
|
||||||
|
title="主键"
|
||||||
|
)
|
||||||
|
|
||||||
|
create_time: Optional[datetime] = Field(
|
||||||
|
title="创建时间"
|
||||||
|
)
|
||||||
|
|
||||||
|
update_time: Optional[datetime] = Field(
|
||||||
|
title="更新时间"
|
||||||
|
)
|
||||||
|
|
||||||
|
class_name: str = Field(
|
||||||
|
title="类名"
|
||||||
|
)
|
||||||
|
|
||||||
|
class_description: str = Field(
|
||||||
|
title="类描述"
|
||||||
|
)
|
||||||
|
|
||||||
|
chapter_name: str = Field(
|
||||||
|
title="章节"
|
||||||
|
)
|
||||||
|
|
||||||
|
chapter_description: str = Field(
|
||||||
|
title="章节描述"
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
class AddHTSClassParams(BaseModel):
|
||||||
|
"""新增编码类别参数"""
|
||||||
|
model_config = ConfigDict(alias_generator=to_snake, populate_by_name=True)
|
||||||
|
|
||||||
|
class_name: str = Field(
|
||||||
|
title="类名"
|
||||||
|
)
|
||||||
|
|
||||||
|
class_description: str = Field(
|
||||||
|
title="类描述"
|
||||||
|
)
|
||||||
|
|
||||||
|
chapter_name: str = Field(
|
||||||
|
title="章节"
|
||||||
|
)
|
||||||
|
|
||||||
|
chapter_description: str = Field(
|
||||||
|
title="章节描述"
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
class UpdateHTSClassParams(BaseModel):
|
||||||
|
"""更新编码类别参数"""
|
||||||
|
model_config = ConfigDict(alias_generator=to_snake, populate_by_name=True)
|
||||||
|
|
||||||
|
class_name: str = Field(
|
||||||
|
title="类名"
|
||||||
|
)
|
||||||
|
|
||||||
|
class_description: str = Field(
|
||||||
|
title="类描述"
|
||||||
|
)
|
||||||
|
|
||||||
|
chapter_name: str = Field(
|
||||||
|
title="章节"
|
||||||
|
)
|
||||||
|
|
||||||
|
chapter_description: str = Field(
|
||||||
|
title="章节描述"
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
class GetHTSClassInfoResponse(BaseResponse):
|
||||||
|
"""获取编码类别信息响应"""
|
||||||
|
data: HTSClassInfo = Field(None, title="编码类信息信息")
|
||||||
|
|
||||||
|
|
||||||
|
class GetHTSClassInfoListResult(ListQueryResult):
|
||||||
|
"""获取编码类别信息列表响应结果"""
|
||||||
|
result: List[HTSClassInfo] = Field(None, title="编码类信息信息列表")
|
||||||
|
|
||||||
|
|
||||||
|
class GetHTSClassListResponse(BaseResponse):
|
||||||
|
"""获取编码类别信息列表响应"""
|
||||||
|
data: GetHTSClassInfoListResult = Field(None, title="编码类信息信息列表")
|
||||||
|
|
||||||
|
|
||||||
|
class HtsItemInfo(BaseModel):
|
||||||
|
"""编码项目信息"""
|
||||||
|
model_config = ConfigDict(alias_generator=to_snake, populate_by_name=True)
|
||||||
|
|
||||||
|
id: Optional[str] = Field(
|
||||||
|
title="主键"
|
||||||
|
)
|
||||||
|
|
||||||
|
create_time: Optional[datetime] = Field(
|
||||||
|
title="创建时间"
|
||||||
|
)
|
||||||
|
|
||||||
|
update_time: Optional[datetime] = Field(
|
||||||
|
title="更新时间"
|
||||||
|
)
|
||||||
|
|
||||||
|
parent_id: Optional[str] = Field(
|
||||||
|
title="父编码"
|
||||||
|
)
|
||||||
|
|
||||||
|
htsno: str = Field(
|
||||||
|
title="编码"
|
||||||
|
)
|
||||||
|
|
||||||
|
indent: int = Field(
|
||||||
|
title="层级"
|
||||||
|
)
|
||||||
|
|
||||||
|
description: str = Field(
|
||||||
|
title="描述"
|
||||||
|
)
|
||||||
|
|
||||||
|
units: dict = Field(
|
||||||
|
title="单位列表"
|
||||||
|
)
|
||||||
|
|
||||||
|
general: str = Field(
|
||||||
|
title="通用税率"
|
||||||
|
)
|
||||||
|
|
||||||
|
special: str = Field(
|
||||||
|
title="特殊税率,适用于特定国家或地区"
|
||||||
|
)
|
||||||
|
|
||||||
|
other: str = Field(
|
||||||
|
title="其他税率"
|
||||||
|
)
|
||||||
|
|
||||||
|
quota_quantity: str = Field(
|
||||||
|
title="配额数量"
|
||||||
|
)
|
||||||
|
|
||||||
|
additional_duties: str = Field(
|
||||||
|
title="附加税"
|
||||||
|
)
|
||||||
|
|
||||||
|
footnotes: dict = Field(
|
||||||
|
title="脚注列表"
|
||||||
|
)
|
||||||
|
|
||||||
|
class_id: str = Field(
|
||||||
|
title="所属类"
|
||||||
|
)
|
||||||
|
|
||||||
|
version_id: str = Field(
|
||||||
|
title="所属版本"
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
class AddHtsItemParams(BaseModel):
|
||||||
|
"""新增编码项目参数"""
|
||||||
|
model_config = ConfigDict(alias_generator=to_snake, populate_by_name=True)
|
||||||
|
|
||||||
|
parent_id: Optional[str] = Field(
|
||||||
|
title="父编码"
|
||||||
|
)
|
||||||
|
|
||||||
|
htsno: str = Field(
|
||||||
|
title="编码"
|
||||||
|
)
|
||||||
|
|
||||||
|
indent: int = Field(
|
||||||
|
title="层级"
|
||||||
|
)
|
||||||
|
|
||||||
|
description: str = Field(
|
||||||
|
title="描述"
|
||||||
|
)
|
||||||
|
|
||||||
|
units: dict = Field(
|
||||||
|
title="单位列表"
|
||||||
|
)
|
||||||
|
|
||||||
|
general: str = Field(
|
||||||
|
title="通用税率"
|
||||||
|
)
|
||||||
|
|
||||||
|
special: str = Field(
|
||||||
|
title="特殊税率,适用于特定国家或地区"
|
||||||
|
)
|
||||||
|
|
||||||
|
other: str = Field(
|
||||||
|
title="其他税率"
|
||||||
|
)
|
||||||
|
|
||||||
|
quota_quantity: str = Field(
|
||||||
|
title="配额数量"
|
||||||
|
)
|
||||||
|
|
||||||
|
additional_duties: str = Field(
|
||||||
|
title="附加税"
|
||||||
|
)
|
||||||
|
|
||||||
|
footnotes: dict = Field(
|
||||||
|
title="脚注列表"
|
||||||
|
)
|
||||||
|
|
||||||
|
class_id: str = Field(
|
||||||
|
title="所属类"
|
||||||
|
)
|
||||||
|
|
||||||
|
version_id: str = Field(
|
||||||
|
title="所属版本"
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
class UpdateHtsItemParams(BaseModel):
|
||||||
|
"""更新编码项目参数"""
|
||||||
|
model_config = ConfigDict(alias_generator=to_snake, populate_by_name=True)
|
||||||
|
|
||||||
|
parent_id: Optional[str] = Field(
|
||||||
|
title="父编码"
|
||||||
|
)
|
||||||
|
|
||||||
|
htsno: str = Field(
|
||||||
|
title="编码"
|
||||||
|
)
|
||||||
|
|
||||||
|
indent: int = Field(
|
||||||
|
title="层级"
|
||||||
|
)
|
||||||
|
|
||||||
|
description: str = Field(
|
||||||
|
title="描述"
|
||||||
|
)
|
||||||
|
|
||||||
|
units: dict = Field(
|
||||||
|
title="单位列表"
|
||||||
|
)
|
||||||
|
|
||||||
|
general: str = Field(
|
||||||
|
title="通用税率"
|
||||||
|
)
|
||||||
|
|
||||||
|
special: str = Field(
|
||||||
|
title="特殊税率,适用于特定国家或地区"
|
||||||
|
)
|
||||||
|
|
||||||
|
other: str = Field(
|
||||||
|
title="其他税率"
|
||||||
|
)
|
||||||
|
|
||||||
|
quota_quantity: str = Field(
|
||||||
|
title="配额数量"
|
||||||
|
)
|
||||||
|
|
||||||
|
additional_duties: str = Field(
|
||||||
|
title="附加税"
|
||||||
|
)
|
||||||
|
|
||||||
|
footnotes: dict = Field(
|
||||||
|
title="脚注列表"
|
||||||
|
)
|
||||||
|
|
||||||
|
class_id: str = Field(
|
||||||
|
title="所属类"
|
||||||
|
)
|
||||||
|
|
||||||
|
version_id: str = Field(
|
||||||
|
title="所属版本"
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
class ImportHtsItemParams(BaseModel):
|
||||||
|
model_config = ConfigDict(alias_generator=to_snake, populate_by_name=True)
|
||||||
|
file_id: str = Field(
|
||||||
|
title="文件ID"
|
||||||
|
)
|
||||||
|
version_id: str = Field(
|
||||||
|
title="版本ID"
|
||||||
|
)
|
||||||
|
class_id: str = Field(
|
||||||
|
title="类ID"
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
class GetHtsItemInfoResponse(BaseResponse):
|
||||||
|
"""获取编码项目信息响应"""
|
||||||
|
data: HtsItemInfo = Field(None, title="编码项信息信息")
|
||||||
|
|
||||||
|
|
||||||
|
class GetHtsItemInfoListResult(ListQueryResult):
|
||||||
|
"""获取编码项目信息列表响应结果"""
|
||||||
|
result: List[HtsItemInfo] = Field(None, title="编码项信息信息列表")
|
||||||
|
|
||||||
|
|
||||||
|
class GetHtsItemListResponse(BaseResponse):
|
||||||
|
"""获取编码项目信息列表响应"""
|
||||||
|
data: GetHtsItemInfoListResult = Field(None, title="编码项信息信息列表")
|
||||||
@ -43,6 +43,7 @@ class PermissionInfo(BaseModel):
|
|||||||
fixed_tag: bool = Field(default=False, description="固定标签页")
|
fixed_tag: bool = Field(default=False, description="固定标签页")
|
||||||
show_link: bool = Field(default=True, description="显示菜单")
|
show_link: bool = Field(default=True, description="显示菜单")
|
||||||
show_parent: bool = Field(default=True, description="显示父级菜单")
|
show_parent: bool = Field(default=True, description="显示父级菜单")
|
||||||
|
is_admin: bool = Field(default=False, description="是否为管理专属页面")
|
||||||
|
|
||||||
class Config:
|
class Config:
|
||||||
json_schema_extra = {
|
json_schema_extra = {
|
||||||
@ -72,7 +73,8 @@ class PermissionInfo(BaseModel):
|
|||||||
"hidden_tag": False,
|
"hidden_tag": False,
|
||||||
"fixed_tag": False,
|
"fixed_tag": False,
|
||||||
"show_link": True,
|
"show_link": True,
|
||||||
"show_parent": True
|
"show_parent": True,
|
||||||
|
"is_admin": False
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -109,6 +111,7 @@ class AddPermissionParams(BaseModel):
|
|||||||
show_parent: bool = Field(default=True, description="显示父级菜单")
|
show_parent: bool = Field(default=True, description="显示父级菜单")
|
||||||
parent_id: str = Field(default="", max_length=36, description="父级菜单ID")
|
parent_id: str = Field(default="", max_length=36, description="父级菜单ID")
|
||||||
menu_type: int = Field(default=0, description="菜单类型")
|
menu_type: int = Field(default=0, description="菜单类型")
|
||||||
|
is_admin: bool = Field(default=False, description="是否为管理专属页面")
|
||||||
|
|
||||||
class Config:
|
class Config:
|
||||||
json_schema_extra = {
|
json_schema_extra = {
|
||||||
@ -133,7 +136,8 @@ class AddPermissionParams(BaseModel):
|
|||||||
"show_link": True,
|
"show_link": True,
|
||||||
"show_parent": True,
|
"show_parent": True,
|
||||||
"parent_id": "",
|
"parent_id": "",
|
||||||
"menu_type": 0
|
"menu_type": 0,
|
||||||
|
"is_admin": False
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@ -387,3 +387,19 @@ class GetUserStatisticsResponse(BaseResponse):
|
|||||||
获取用户统计信息响应模型。
|
获取用户统计信息响应模型。
|
||||||
"""
|
"""
|
||||||
data: GetUserStatisticsResult = Field(default=None, description="响应数据")
|
data: GetUserStatisticsResult = Field(default=None, description="响应数据")
|
||||||
|
|
||||||
|
|
||||||
|
class UpdateBaseUserInfoParams(BaseModel):
|
||||||
|
"""修改基础信息参数"""
|
||||||
|
name: str
|
||||||
|
"""姓名"""
|
||||||
|
gender: int
|
||||||
|
"""性别"""
|
||||||
|
|
||||||
|
class Config:
|
||||||
|
json_schema_extra = {
|
||||||
|
"example": {
|
||||||
|
"name": "张三",
|
||||||
|
"gender": 1,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@ -5,6 +5,8 @@
|
|||||||
# @File : common.py
|
# @File : common.py
|
||||||
# @Software : PyCharm
|
# @Software : PyCharm
|
||||||
# @Comment : 本程序
|
# @Comment : 本程序
|
||||||
|
from typing import List, Any, Optional, Type
|
||||||
|
|
||||||
|
|
||||||
def bytes2human(n, format_str='%(value).1f%(symbol)s'):
|
def bytes2human(n, format_str='%(value).1f%(symbol)s'):
|
||||||
"""Used by various scripts. See:
|
"""Used by various scripts. See:
|
||||||
@ -26,11 +28,14 @@ def bytes2human(n, format_str='%(value).1f%(symbol)s'):
|
|||||||
return format_str % dict(symbol=symbols[0], value=n)
|
return format_str % dict(symbol=symbols[0], value=n)
|
||||||
|
|
||||||
|
|
||||||
async def filterKeyValues(dataList: list, key: str) -> list:
|
async def filterKeyValues(dataList: List[dict], key: str, default: Any = None, convert_type: Optional[Type] = None) -> List[Any]:
|
||||||
"""
|
"""
|
||||||
获取列表字段数据
|
获取列表字段数据,并可选择进行类型转换。
|
||||||
:param dataList: 数据列表
|
|
||||||
:param key: 关键字
|
:param dataList: 数据列表(列表中的元素是字典)
|
||||||
:return:
|
:param key: 要提取的字段
|
||||||
|
:param default: 如果字段不存在,返回的默认值
|
||||||
|
:param convert_type: 需要转换的类型(如 int、str、float 等),默认为 None 不转换
|
||||||
|
:return: 提取并转换后的值列表
|
||||||
"""
|
"""
|
||||||
return [item[key] for item in dataList]
|
return [convert_type(item.get(key, default)) if convert_type else item.get(key, default) for item in dataList]
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user