KGPython/controller/OperationController.py

import json
import traceback
from app import app
from robyn import jsonify, Response
from service.OperationService import OperationService
from urllib.parse import unquote

# 实例化业务逻辑对象
operation_service = OperationService()

def create_response(status_code, data_dict):
    """
    统一响应格式封装，强制使用 UTF-8 防止中文乱码。
    """
    return Response(
        status_code=status_code,
        description=jsonify(data_dict),
        headers={"Content-Type": "application/json; charset=utf-8"}
    )

def parse_request_body(req):
    """
    通用请求体解析器：解决解析 bytes、字符串或多重序列化的情况。
    """
    body = req.body
    try:
        if isinstance(body, (bytes, bytearray)):
            body = json.loads(body.decode('utf-8'))
        if isinstance(body, str):
            body = json.loads(body)
        return body if isinstance(body, dict) else {}
    except Exception as e:
        print(f"Request Body Parse Error: {e}")
        return {}

# --- 1. 获取全量动态标签 ---
@app.get("/api/kg/labels")
def get_labels(req):
    try:
        labels = operation_service.get_all_labels()
        return create_response(200, {"code": 200, "data": labels, "msg": "success"})
    except Exception as e:
        traceback.print_exc()
        return create_response(500, {"code": 500, "msg": f"获取标签失败: {str(e)}"})

# --- 2. 输入联想建议 ---
@app.get("/api/kg/node/suggest")
def suggest_node(req):
    try:
        query_data = getattr(req, "queries", getattr(req, "query_params", {}))
        raw_keyword = query_data.get("keyword", "")
        keyword = raw_keyword[0] if isinstance(raw_keyword, list) and raw_keyword else str(raw_keyword)
        clean_keyword = unquote(keyword).strip()

        suggestions = operation_service.suggest_nodes(clean_keyword)
        return create_response(200, {"code": 200, "data": suggestions, "msg": "success"})
    except Exception as e:
        traceback.print_exc()
        return create_response(500, {"code": 500, "msg": str(e)})

# --- 3. 获取分页节点列表 ---
@app.get("/api/kg/nodes")
def get_nodes(req):
    try:
        query_data = getattr(req, "queries", getattr(req, "query_params", {}))
        page = int(query_data.get("page", "1"))
        page_size = int(query_data.get("pageSize", "20"))

        res_data = operation_service.get_nodes_subset(page, page_size)
        return create_response(200, {"code": 200, "data": res_data, "msg": "success"})
    except Exception as e:
        traceback.print_exc()
        return create_response(500, {"code": 500, "msg": f"获取节点列表失败: {str(e)}"})

# --- 4. 获取分页关系列表 ---
@app.get("/api/kg/relationships")
def get_relationships(req):
    try:
        query_data = getattr(req, "queries", getattr(req, "query_params", {}))
        page = int(query_data.get("page", "1"))
        page_size = int(query_data.get("pageSize", "20"))

        res_data = operation_service.get_relationships_subset(page, page_size)
        return create_response(200, {"code": 200, "data": res_data, "msg": "success"})
    except Exception as e:
        traceback.print_exc()
        return create_response(500, {"code": 500, "msg": f"获取关系列表失败: {str(e)}"})

# --- 5. 新增节点 ---
@app.post("/api/kg/node/add")
def add_node(req):
    try:
        body = parse_request_body(req)
        # 增加 strip() 防止空格导致的匹配失败
        label = str(body.get("label", "Drug")).strip()
        name = str(body.get("name", "")).strip()

        if not name:
            return create_response(200, {"code": 400, "msg": "添加失败：节点名称不能为空"})

        result = operation_service.add_node(label, name)
        return create_response(200, {"code": 200 if result["success"] else 400, "msg": result["msg"]})
    except Exception as e:
        traceback.print_exc()
        return create_response(200, {"code": 500, "msg": f"服务器内部错误: {str(e)}"})

# --- 6. 修改节点 ---
@app.post("/api/kg/node/update")
def update_node(req):
    try:
        body = parse_request_body(req)
        node_id = body.get("id")
        name = str(body.get("name", "")).strip()
        label = str(body.get("label", "")).strip()

        if not node_id or not name:
            return create_response(200, {"code": 400, "msg": "修改失败：参数缺失(ID或Name)"})

        result = operation_service.update_node(node_id, name, label)
        return create_response(200, {"code": 200 if result["success"] else 400, "msg": result["msg"]})
    except Exception as e:
        traceback.print_exc()
        return create_response(200, {"code": 500, "msg": str(e)})

# --- 7. 新增关系 ---
@app.post("/api/kg/rel/add")
def add_relationship(req):
    try:
        body = parse_request_body(req)
        # 强制格式化所有字符串，彻底杜绝数据类型导致的 MATCH 失败
        source = str(body.get("source", "")).strip()
        target = str(body.get("target", "")).strip()
        rel_type = str(body.get("type", "")).strip()
        rel_label = str(body.get("label", "")).strip()

        if not all([source, target, rel_type]):
            return create_response(200, {"code": 400, "msg": "添加失败：起始点、结束点和类型不能为空"})

        result = operation_service.add_relationship(source, target, rel_type, rel_label)
        return create_response(200, {"code": 200 if result["success"] else 400, "msg": result["msg"]})
    except Exception as e:
        traceback.print_exc()
        return create_response(200, {"code": 500, "msg": f"新增关系异常: {str(e)}"})

# --- 8. 修改关系 ---
@app.post("/api/kg/rel/update")
def update_rel(req):
    try:
        body = parse_request_body(req)
        rel_id = body.get("id")
        source = str(body.get("source", "")).strip()
        target = str(body.get("target", "")).strip()
        rel_type = str(body.get("type", "")).strip()
        rel_label = str(body.get("label", "")).strip()

        if not all([rel_id, source, target, rel_type]):
            return create_response(200, {"code": 400, "msg": "修改失败：必要参数(ID/Source/Target/Type)缺失"})

        result = operation_service.update_relationship(rel_id, source, target, rel_type, rel_label)
        return create_response(200, {"code": 200 if result["success"] else 400, "msg": result["msg"]})
    except Exception as e:
        traceback.print_exc()
        return create_response(200, {"code": 500, "msg": str(e)})

# --- 9. 删除节点 ---
@app.post("/api/kg/node/delete")
def delete_node(req):
    try:
        body = parse_request_body(req)
        node_id = body.get("id")
        if not node_id:
            return create_response(200, {"code": 400, "msg": "删除失败：节点ID不能为空"})

        operation_service.delete_node(node_id)
        return create_response(200, {"code": 200, "msg": "节点及其关联关系已成功删除"})
    except Exception as e:
        traceback.print_exc()
        return create_response(200, {"code": 500, "msg": str(e)})

# --- 10. 删除关系 ---
@app.post("/api/kg/rel/delete")
def delete_rel(req):
    try:
        body = parse_request_body(req)
        rel_id = body.get("id")
        if not rel_id:
            return create_response(200, {"code": 400, "msg": "删除失败：关系ID不能为空"})

        operation_service.delete_relationship(rel_id)
        return create_response(200, {"code": 200, "msg": "关系已成功从数据库移除"})
    except Exception as e:
        traceback.print_exc()
        return create_response(200, {"code": 500, "msg": str(e)})