提交
This commit is contained in:
ZhuJW
0
app/blueprints/label_data/__inin__.py
Normal file
0
app/blueprints/label_data/__inin__.py
Normal file
294
app/blueprints/label_data/rotutes.py
Normal file
294
app/blueprints/label_data/rotutes.py
Normal file
@@ -0,0 +1,294 @@
|
||||
from datetime import datetime
|
||||
from flask import Blueprint, current_app, request, jsonify
|
||||
from flask_jwt_extended import jwt_required
|
||||
import requests
|
||||
from sqlalchemy import func
|
||||
from app import db
|
||||
from app.models import Fst
|
||||
from app.utils.fst_tree import build_tree
|
||||
|
||||
|
||||
label_bp = Blueprint('label', __name__)
|
||||
|
||||
|
||||
def format_fst_item(fst_item):
|
||||
"""将Fst对象转换为指定格式的字典"""
|
||||
# 从reserved_json中获取type，如果没有则默认为空字符串
|
||||
item_type = ""
|
||||
if fst_item.reserved_json and "type" in fst_item.reserved_json:
|
||||
item_type = fst_item.reserved_json["type"]
|
||||
|
||||
return {
|
||||
"date": fst_item.update_time.strftime('%Y-%m-%d'), # 格式化日期
|
||||
"id": fst_item.id,
|
||||
"name": fst_item.name,
|
||||
"parentId": None if fst_item.parent_id==591 else fst_item.parent_id, # 注意这里是驼峰命名
|
||||
"name_cn": fst_item.name_cn,
|
||||
"annotation":fst_item.annotation,
|
||||
"level":fst_item.level
|
||||
}
|
||||
|
||||
@label_bp.route('/fst-tree', methods=['GET'])
|
||||
def get_fst_tree():
|
||||
"""获取所有FST数据并格式化为树形结构列表"""
|
||||
try:
|
||||
# 查询所有FST记录
|
||||
all_fst = Fst.query.all()
|
||||
|
||||
# 转换为指定格式
|
||||
formatted_data = [format_fst_item(item) for item in all_fst]
|
||||
|
||||
# 返回JSON响应
|
||||
return jsonify({
|
||||
"success": True,
|
||||
"data": formatted_data
|
||||
})
|
||||
except Exception as e:
|
||||
return jsonify({
|
||||
"success": False,
|
||||
"error": str(e)
|
||||
}), 500
|
||||
|
||||
|
||||
@label_bp.route('/create-levelone', methods=['POST'])
|
||||
@jwt_required()
|
||||
def create_levelone():
|
||||
# 获取请求数据
|
||||
data = request.get_json()
|
||||
|
||||
# 验证必要参数
|
||||
required_fields = ['name', 'level', 'annotation']
|
||||
for field in required_fields:
|
||||
if field not in data:
|
||||
return jsonify({'error': f'Missing required field: {field}'}), 400
|
||||
|
||||
# 验证层级是否为1
|
||||
if data['level'] != 1:
|
||||
return jsonify({'error': 'Level must be 1 for first-level tags'}), 400
|
||||
|
||||
|
||||
try:
|
||||
# 查询当前最大的id值
|
||||
max_id_result = db.session.query(func.max(Fst.id)).first()
|
||||
max_id = max_id_result[0] if max_id_result[0] is not None else 0
|
||||
|
||||
# 新id为最大id + 1
|
||||
new_id = max_id + 1
|
||||
|
||||
# 创建新的一级标签
|
||||
new_tag = Fst(
|
||||
id=new_id,
|
||||
name=data['name'],
|
||||
level=data['level'],
|
||||
annotation=data['annotation'],
|
||||
parent_id=591, # 固定设置parent_id为591
|
||||
update_time=datetime.now() # 设置更新时间
|
||||
# 其他字段如name_cn、reserved_json、bag_sum如果有默认值可以不用指定
|
||||
)
|
||||
|
||||
# 添加到数据库会话并提交
|
||||
db.session.add(new_tag)
|
||||
db.session.commit()
|
||||
|
||||
# print(f"新标签的ID: {new_tag.id}")
|
||||
# 返回创建成功的标签信息
|
||||
return jsonify({
|
||||
'message': 'First-level tag created successfully',
|
||||
'tag': {
|
||||
'id': new_tag.id,
|
||||
'name': new_tag.name,
|
||||
'level': new_tag.level,
|
||||
'annotation': new_tag.annotation,
|
||||
'parent_id': new_tag.parent_id,
|
||||
'update_time': new_tag.update_time.isoformat()
|
||||
}
|
||||
}), 201
|
||||
|
||||
except Exception as e:
|
||||
# 发生错误时回滚
|
||||
db.session.rollback()
|
||||
return jsonify({'error': str(e)}), 500
|
||||
|
||||
|
||||
@label_bp.route('/update-fst-annotation', methods=['POST'])
|
||||
@jwt_required()
|
||||
def update_fst_annotation():
|
||||
# 获取请求数据
|
||||
data = request.get_json()
|
||||
|
||||
# 验证必要参数
|
||||
required_fields = ['name', 'level', 'annotation']
|
||||
for field in required_fields:
|
||||
if field not in data:
|
||||
return jsonify({'error': f'Missing required field: {field}'}), 400
|
||||
|
||||
try:
|
||||
# 根据name和level查询对应的记录
|
||||
fst_record = Fst.query.filter_by(
|
||||
name=data['name'],
|
||||
level=data['level']
|
||||
).first()
|
||||
|
||||
# 检查记录是否存在
|
||||
if not fst_record:
|
||||
return jsonify({
|
||||
'error': f'No record found with name: {data["name"]} and level: {data["level"]}'
|
||||
}), 404
|
||||
|
||||
# 更新annotation字段和更新时间
|
||||
fst_record.annotation = data['annotation']
|
||||
fst_record.update_time = datetime.now() # 更新时间戳
|
||||
|
||||
# 提交修改
|
||||
db.session.commit()
|
||||
|
||||
# 返回更新成功的信息
|
||||
return jsonify({
|
||||
'message': 'Annotation updated successfully',
|
||||
'updated_record': {
|
||||
'id': fst_record.id,
|
||||
'name': fst_record.name,
|
||||
'level': fst_record.level,
|
||||
'annotation': fst_record.annotation,
|
||||
'update_time': fst_record.update_time.isoformat()
|
||||
}
|
||||
}), 200
|
||||
|
||||
except Exception as e:
|
||||
# 发生错误时回滚
|
||||
db.session.rollback()
|
||||
return jsonify({'error': str(e)}), 500
|
||||
|
||||
|
||||
@label_bp.route('/add-fst', methods=['POST'])
|
||||
@jwt_required()
|
||||
def add_fst():
|
||||
# 获取请求数据
|
||||
data = request.get_json()
|
||||
|
||||
# 验证必要参数
|
||||
required_fields = ['name', 'level', 'parentName', 'annotation']
|
||||
for field in required_fields:
|
||||
if field not in data:
|
||||
return jsonify({'error': f'Missing required field: {field}'}), 400
|
||||
|
||||
try:
|
||||
# 根据parentName查询父标签的记录，获取parent_id
|
||||
parent_record = Fst.query.filter_by(name=data['parentName']).first()
|
||||
|
||||
# 检查父标签是否存在
|
||||
if not parent_record:
|
||||
return jsonify({
|
||||
'error': f'Parent tag not found with name: {data["parentName"]}'
|
||||
}), 404
|
||||
|
||||
# 查询当前最大的id值
|
||||
max_id_result = db.session.query(func.max(Fst.id)).first()
|
||||
max_id = max_id_result[0] if max_id_result[0] is not None else 0
|
||||
|
||||
# 新id为最大id + 1
|
||||
new_id = max_id + 1
|
||||
|
||||
# 创建新的fst记录
|
||||
new_fst = Fst(
|
||||
id=new_id,
|
||||
name=data['name'],
|
||||
level=data['level'],
|
||||
parent_id=parent_record.id, # 使用查询到的父标签ID
|
||||
annotation=data['annotation'],
|
||||
update_time=datetime.now()
|
||||
)
|
||||
|
||||
# 添加到数据库并提交
|
||||
db.session.add(new_fst)
|
||||
db.session.commit()
|
||||
|
||||
# 返回成功信息
|
||||
return jsonify({
|
||||
'message': 'Fst record created successfully',
|
||||
'record': {
|
||||
'id': new_fst.id,
|
||||
'name': new_fst.name,
|
||||
'level': new_fst.level,
|
||||
'parent_id': new_fst.parent_id,
|
||||
'parentName': data['parentName'],
|
||||
'annotation': new_fst.annotation,
|
||||
'update_time': new_fst.update_time.isoformat()
|
||||
}
|
||||
}), 201
|
||||
|
||||
except Exception as e:
|
||||
db.session.rollback()
|
||||
return jsonify({'error': str(e)}), 500
|
||||
|
||||
|
||||
|
||||
@label_bp.route("/sync-fst", methods=["POST"])
|
||||
def sync_fst():
|
||||
# 1. 解析请求参数
|
||||
params = request.get_json() or {}
|
||||
print(f"【请求参数】原始参数: {params}")
|
||||
|
||||
name = params.get("name", "")
|
||||
parent_name = params.get("parent_name", "")
|
||||
|
||||
# 参数校验
|
||||
if not name or not parent_name:
|
||||
return (
|
||||
jsonify(
|
||||
{
|
||||
"success": False,
|
||||
"message": "缺少必要参数（name 或 parent_name）",
|
||||
}
|
||||
),
|
||||
400,
|
||||
)
|
||||
|
||||
# 2. 转发请求配置
|
||||
API_URL = "http://10.0.240.4:5232/api/fst/update"
|
||||
TIMEOUT = 10 # 超时时间（秒）
|
||||
data = {"name": name, "parent_name": parent_name}
|
||||
|
||||
try:
|
||||
# 3. 发送转发请求
|
||||
res = requests.post(url=API_URL, json=data, timeout=TIMEOUT)
|
||||
res.raise_for_status() # 自动抛出 4xx/5xx 状态码的异常
|
||||
|
||||
# 4. 处理对方 API 的响应（假设返回 JSON 格式）
|
||||
try:
|
||||
remote_response = res.json()
|
||||
except ValueError:
|
||||
# 对方返回非 JSON 格式响应
|
||||
return (
|
||||
jsonify(
|
||||
{
|
||||
"success": False,
|
||||
"message": "同步失败：目标接口返回无效格式",
|
||||
"details": res.text,
|
||||
}
|
||||
),
|
||||
500,
|
||||
)
|
||||
|
||||
# 5. 返回成功响应
|
||||
return jsonify(
|
||||
{
|
||||
"success": True,
|
||||
"message": "同步成功",
|
||||
"data": remote_response, # 携带对方 API 的返回数据
|
||||
}
|
||||
)
|
||||
|
||||
except requests.exceptions.RequestException as e:
|
||||
# 捕获所有 requests 相关异常（超时、连接失败、4xx/5xx 等）
|
||||
error_msg = f"同步失败：{str(e)}"
|
||||
print(f"【转发请求错误】{error_msg}")
|
||||
return (
|
||||
jsonify(
|
||||
{
|
||||
"success": False,
|
||||
"message": error_msg,
|
||||
}
|
||||
),
|
||||
500,
|
||||
)
|
||||
Reference in New Issue
Block a user