1 .修改了新建评估指标删除的bug
This commit is contained in:
144
src/api/dimension.py
Normal file
144
src/api/dimension.py
Normal file
@@ -0,0 +1,144 @@
|
||||
"""
|
||||
评测维度管理 API 路由
|
||||
"""
|
||||
import os
|
||||
import pymysql
|
||||
import yaml
|
||||
from flask import Blueprint, request, jsonify
|
||||
|
||||
# 获取项目根目录
|
||||
PROJECT_ROOT = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
||||
|
||||
# 创建蓝图
|
||||
dimension_bp = Blueprint('dimension', __name__, url_prefix='/api/dimension')
|
||||
|
||||
|
||||
def get_db_connection():
|
||||
"""获取数据库连接"""
|
||||
CONFIG_PATH = os.path.join(PROJECT_ROOT, 'config.yaml')
|
||||
with open(CONFIG_PATH, 'r', encoding='utf-8') as f:
|
||||
CONFIG = yaml.safe_load(f)
|
||||
db_config = CONFIG['database']
|
||||
return pymysql.connect(
|
||||
host=db_config['host'],
|
||||
port=db_config['port'],
|
||||
user=db_config['username'],
|
||||
password=db_config['password'],
|
||||
database=db_config['name'],
|
||||
charset=db_config.get('charset', 'utf8mb4'),
|
||||
cursorclass=pymysql.cursors.DictCursor
|
||||
)
|
||||
|
||||
|
||||
def generic_get_all(table_name, order_by='create_time DESC'):
|
||||
"""通用查询所有"""
|
||||
conn = get_db_connection()
|
||||
cursor = conn.cursor()
|
||||
cursor.execute(f"SELECT * FROM {table_name} ORDER BY {order_by}")
|
||||
result = cursor.fetchall()
|
||||
cursor.close()
|
||||
conn.close()
|
||||
return result
|
||||
|
||||
|
||||
def generic_create(table_name, data):
|
||||
"""通用创建"""
|
||||
conn = get_db_connection()
|
||||
cursor = conn.cursor()
|
||||
columns = ', '.join(data.keys())
|
||||
placeholders = ', '.join(['%s'] * len(data))
|
||||
sql = f"INSERT INTO {table_name} ({columns}) VALUES ({placeholders})"
|
||||
cursor.execute(sql, list(data.values()))
|
||||
conn.commit()
|
||||
new_id = cursor.lastrowid
|
||||
cursor.close()
|
||||
conn.close()
|
||||
return new_id
|
||||
|
||||
|
||||
def generic_update(table_name, id_val, data):
|
||||
"""通用更新"""
|
||||
conn = get_db_connection()
|
||||
cursor = conn.cursor()
|
||||
set_clause = ', '.join([f"{k} = %s" for k in data.keys()])
|
||||
sql = f"UPDATE {table_name} SET {set_clause} WHERE id = %s"
|
||||
values = list(data.values()) + [id_val]
|
||||
cursor.execute(sql, values)
|
||||
conn.commit()
|
||||
cursor.close()
|
||||
conn.close()
|
||||
|
||||
|
||||
def generic_delete(table_name, id_val):
|
||||
"""通用删除"""
|
||||
conn = get_db_connection()
|
||||
cursor = conn.cursor()
|
||||
cursor.execute(f"DELETE FROM {table_name} WHERE id = %s", (id_val,))
|
||||
conn.commit()
|
||||
cursor.close()
|
||||
conn.close()
|
||||
|
||||
|
||||
def generic_get_by_id(table_name, id_val):
|
||||
"""通用按ID查询"""
|
||||
conn = get_db_connection()
|
||||
cursor = conn.cursor()
|
||||
cursor.execute(f"SELECT * FROM {table_name} WHERE id = %s", (id_val,))
|
||||
result = cursor.fetchone()
|
||||
cursor.close()
|
||||
conn.close()
|
||||
return result
|
||||
|
||||
|
||||
# ============ 评测维度 CRUD ============
|
||||
|
||||
@dimension_bp.route('', methods=['GET'])
|
||||
def get_dimensions():
|
||||
"""获取所有评测维度"""
|
||||
return jsonify({'code': 0, 'data': generic_get_all('model_dimension')})
|
||||
|
||||
|
||||
@dimension_bp.route('/<int:id>', methods=['GET'])
|
||||
def get_dimension_by_id(id):
|
||||
"""获取单个评测维度"""
|
||||
dimension = generic_get_by_id('model_dimension', id)
|
||||
if dimension:
|
||||
return jsonify({'code': 0, 'data': dimension})
|
||||
return jsonify({'code': 1, 'message': '维度不存在'})
|
||||
|
||||
|
||||
@dimension_bp.route('', methods=['POST'])
|
||||
def create_dimension():
|
||||
"""创建评测维度"""
|
||||
data = request.json
|
||||
insert_data = {
|
||||
'name': data.get('name'),
|
||||
'type': data.get('type'),
|
||||
'description': data.get('description', '')
|
||||
}
|
||||
new_id = generic_create('model_dimension', insert_data)
|
||||
return jsonify({'code': 0, 'message': '创建成功', 'id': new_id})
|
||||
|
||||
|
||||
@dimension_bp.route('/<int:id>', methods=['PUT'])
|
||||
def update_dimension(id):
|
||||
"""更新评测维度"""
|
||||
data = request.json
|
||||
update_data = {}
|
||||
if 'name' in data:
|
||||
update_data['name'] = data['name']
|
||||
if 'type' in data:
|
||||
update_data['type'] = data['type']
|
||||
if 'description' in data:
|
||||
update_data['description'] = data['description']
|
||||
|
||||
if update_data:
|
||||
generic_update('model_dimension', id, update_data)
|
||||
return jsonify({'code': 0, 'message': '更新成功'})
|
||||
|
||||
|
||||
@dimension_bp.route('/<int:id>', methods=['DELETE'])
|
||||
def delete_dimension(id):
|
||||
"""删除评测维度"""
|
||||
generic_delete('model_dimension', id)
|
||||
return jsonify({'code': 0, 'message': '删除成功'})
|
||||
Reference in New Issue
Block a user