General-AI-Platform-Backend/website/handlers/enterprise_busi_model/handler.py

# -*- coding: utf-8 -*-

import logging
from sqlalchemy import text

from website import db_mysql, errors
from website.handler import APIHandler, authenticated
from website.util import shortuuid
from website.db import enterprise_busi_model as DB_BusiModel


class ListHandler(APIHandler):
    """
    
    """
    @authenticated
    def post(self):

        self.finish()


class AddHandler(APIHandler):
    """
    - 描述：添加企业部署的业务模型
    - 请求方式：post
    - 请求参数：
    > - entity_id, int, 企业id
    > - name, string, 业务模型名称
    > - comment, string, 简介
    > - basemodel_ids, string, 基础模型id, 多个使用逗号分割
    > - business_logic, string, 业务代码压缩包的md5
    > - business_conf_file, string, 业务配置压缩包的文件md5
    > - business_conf_param, string, 业务配置的参数，json字符串，eg: '{"a":1, "b":2}'
    > - link_node_ids, string, 关联节点id, 多个节点逗号分割
    - 返回值：无
    """
    @authenticated
    def post(self):
        entity_id = self.get_argument("entity_id")
        name = self.get_escaped_argument("name", "")
        comment = self.get_escaped_argument("comment", "")
        basemodel_ids = self.get_escaped_argument("basemodel_ids", "")
        business_logic = self.get_escaped_argument("business_logic", "")
        business_conf_file = self.get_escaped_argument("business_conf_file", "")
        business_conf_param = self.get_escaped_argument("business_conf_param", "")
        link_node_ids = self.get_escaped_argument("link_node_ids", "")
    
        if not entity_id or not name or not basemodel_ids:
            raise errors.HTTPAPIError(errors.ERROR_BAD_REQUEST, "参数错误")
    
        if not business_logic:
            raise errors.HTTPAPIError(errors.ERROR_BAD_REQUEST, "业务代码参数错误")
        
        if not business_conf_file and not business_conf_param:
            raise errors.HTTPAPIError(errors.ERROR_BAD_REQUEST, "业务配置参数错误")
        
        if not link_node_ids:
            raise errors.HTTPAPIError(errors.ERROR_BAD_REQUEST, "关联节点参数错误")

        model_data = {
            "entity_id": entity_id,
            "name": name,
            "comment": comment,
            "basemodel_ids": basemodel_ids,
            "business_logic": business_logic,
            "business_conf_file": business_conf_file,
            "business_conf_param": business_conf_param,
            "link_node_ids": link_node_ids,
        }

        db_busimodel = DB_BusiModel.EnterpriseBusiModelRepository()
        busimodel_id, busimodel_suid = db_busimodel.insert_busi_model(model_data)

        model_node_data = {
            "entity_id": entity_id,
            "busimodel_id": busimodel_id,
            "busimodel_suid": busimodel_suid,
            "node_ids": link_node_ids
        }
        db_model_node = DB_BusiModel.EnterpriseBusiModelNodeRepository()
        db_model_node.insert_busi_model_nodes(model_node_data)

        self.finish()


class InfoHandler(APIHandler):
    """
    - 描述：企业部署的业务模型详情
    - 请求方式：post
    - 请求参数：
    > - id, int, 业务模型id
    - 返回值：
    ```
        {
            "name": "xxx", 
            "comment": "xxx",
            "basemodel_list": "xxx,xx,xx",
            "business_logic": "xxx",
            "business_conf_file": "xxx",
            "business_conf_param": "xxx",
            "link_node_list": "xxx,xx,xx"
        }
    ```
    """
    @authenticated
    def post(self):
        busimodel_id = self.get_argument("id")
    
        if not busimodel_id:
            raise errors.HTTPAPIError(errors.ERROR_BAD_REQUEST, "参数错误")
    
        db_busimodel = DB_BusiModel.EnterpriseBusiModelRepository()
        busi_model_data = db_busimodel.get_busi_model_by_id(busimodel_id)

        if not busi_model_data:
            raise errors.HTTPAPIError(errors.ERROR_BAD_REQUEST, "业务模型不存在")
    
        

        model_data = {
            "name": busi_model_data.name, 
            "comment": busi_model_data.comment,
            "basemodel_ids": ",".join([str(x) for x in busi_model_data.basemodel_list]),
            "business_logic": busi_model_data.business_logic,
            "business_conf_file": busi_model_data.business_conf_file,
            "business_conf_param": busi_model_data.business_conf
        }


        self.finish({
            "name": model_data["name"], 
            "comment": model_data["comment"],
            "basemodel_list": model_data["basemodel_ids"],
            "business_logic": model_data["business_logic"],
            "business_conf_file": model_data["business_conf_file"],
            "business_conf_param": model_data["business_conf_param"],
            "link_node_list": model_data["link_node_ids"]
        })



class EditHandler(APIHandler):
    """
    
    """
    @authenticated
    def post(self):

        self.finish()