|
|
# -*- coding: utf-8 -*-
|
|
|
import json
|
|
|
import copy
|
|
|
import logging
|
|
|
from sqlalchemy.ext.declarative import declarative_base
|
|
|
from sqlalchemy import Column, Integer, String, DateTime, func
|
|
|
from typing import Any, Dict, List, Optional, Tuple, Union
|
|
|
|
|
|
from website.db_mysql import get_session
|
|
|
from website.util import shortuuid
|
|
|
from website.db.enterprise_entity.enterprise_entity import EnterpriseEntityRepository
|
|
|
from website.db.enterprise_node import enterprise_node as DB_Node
|
|
|
from website.db.alg_model import alg_model as DB_alg_model
|
|
|
|
|
|
Base = declarative_base()
|
|
|
|
|
|
"""
|
|
|
CREATE TABLE `enterprise_busi_model` (
|
|
|
`id` int NOT NULL AUTO_INCREMENT,
|
|
|
`suid` varchar(10) NOT NULL DEFAULT '' COMMENT 'short uuid',
|
|
|
`entity_id` int NOT NULL COMMENT '企业id',
|
|
|
`entity_suid` varchar(10) NOT NULL COMMENT '企业uuid',
|
|
|
`name` varchar(255) NOT NULL,
|
|
|
`comment` varchar(255) DEFAULT '',
|
|
|
`basemodel_ids` varchar(255) NOT NULL COMMENT '关联模型，json list, [{"id":123,"suid":"xxx"},...]',
|
|
|
`business_logic` varchar(32) DEFAULT NULL COMMENT '业务代码压缩包的md5',
|
|
|
`business_conf_file` varchar(32) DEFAULT NULL COMMENT '业务配置参数压缩包的文件md5',
|
|
|
`business_conf_param` varchar(255) DEFAULT NULL COMMENT '业务配置的参数，json字符串，eg: ''{"a":1, "b":2}''',
|
|
|
`delete` tinyint(1) DEFAULT '0',
|
|
|
`create_time` datetime DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
|
|
|
`update_time` datetime DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
|
|
|
PRIMARY KEY (`id`)
|
|
|
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci COMMENT='企业业务模型表';
|
|
|
|
|
|
CREATE TABLE `enterprise_busi_model_node` (
|
|
|
`id` int NOT NULL AUTO_INCREMENT,
|
|
|
`suid` varchar(10) NOT NULL,
|
|
|
`entity_suid` varchar(10) DEFAULT NULL COMMENT '企业suid',
|
|
|
`busi_model_id` int DEFAULT NULL,
|
|
|
`busi_model_suid` varchar(10) DEFAULT NULL,
|
|
|
`node_id` int DEFAULT NULL,
|
|
|
`node_suid` varchar(10) DEFAULT NULL,
|
|
|
`create_time` datetime DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
|
|
|
PRIMARY KEY (`id`)
|
|
|
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci;
|
|
|
"""
|
|
|
|
|
|
|
|
|
class EnterpriseBusiModel(Base):
|
|
|
__tablename__ = 'enterprise_busi_model'
|
|
|
|
|
|
id = Column(Integer, primary_key=True, autoincrement=True)
|
|
|
suid = Column(String(10), nullable=False, default='', comment='short uuid')
|
|
|
entity_id = Column(Integer, nullable=False, comment='企业id')
|
|
|
entity_suid = Column(String(10), nullable=False, comment='企业uuid')
|
|
|
name = Column(String(255), nullable=False)
|
|
|
comment = Column(String(255), nullable=True, default='')
|
|
|
base_models = Column(String(255), nullable=False, comment='关联的基础模型，json list, [{"id":123,"suid":"xxx"},...]')
|
|
|
business_logic = Column(String(32), nullable=True, comment='业务代码压缩包的md5')
|
|
|
business_conf_file = Column(String(32), nullable=True, comment='业务配置参数压缩包的文件md5')
|
|
|
business_conf_param = Column(String(255), nullable=True, comment='业务配置的参数，json字符串，eg: ''{"a":1, "b":2}''')
|
|
|
delete = Column(Integer, default=0)
|
|
|
create_time = Column(DateTime, nullable=False, default=func.now(), onupdate=func.now())
|
|
|
update_time = Column(DateTime, nullable=False, default=func.now(), onupdate=func.now())
|
|
|
|
|
|
def __repr__(self):
|
|
|
return f"EnterpriseBusiModel(id={self.id}, suid='{self.suid}', name='{self.name}')"
|
|
|
|
|
|
class EnterpriseBusiModelNode(Base):
|
|
|
__tablename__ = 'enterprise_busi_model_node'
|
|
|
|
|
|
id = Column(Integer, primary_key=True)
|
|
|
suid = Column(String(10), nullable=False, default='')
|
|
|
entity_suid = Column(String(10))
|
|
|
busi_model_id = Column(Integer)
|
|
|
busi_model_suid = Column(String(10))
|
|
|
node_id = Column(Integer)
|
|
|
node_suid = Column(String(10))
|
|
|
create_time = Column(DateTime, default=func.current_timestamp())
|
|
|
def __repr__(self):
|
|
|
return f'<EnterpriseBusiModelNode(id={self.id}, suid={self.suid})>'
|
|
|
|
|
|
|
|
|
class EnterpriseBusiModelRepository(object):
|
|
|
|
|
|
def get_by_id(self, id: int) -> Optional[EnterpriseBusiModel]:
|
|
|
with get_session() as session:
|
|
|
model = session.query(EnterpriseBusiModel).filter(EnterpriseBusiModel.id == id).first()
|
|
|
return model
|
|
|
|
|
|
|
|
|
def insert_busi_model(self, data: Dict):
|
|
|
entity_suid = EnterpriseEntityRepository().get_entity_suid(data['entity_id'])
|
|
|
data['suid'] = shortuuid.ShortUUID().random(10)
|
|
|
data['entity_suid'] = entity_suid
|
|
|
|
|
|
base_model_ids = [int(model_id) for model_id in data['basemodel_ids'].split(',')]
|
|
|
base_model_db = DB_alg_model.ModelRepositry()
|
|
|
base_model = []
|
|
|
for base_model_id in base_model_ids:
|
|
|
base_model_suid = base_model_db.get_suid(base_model_id)
|
|
|
base_model.append({
|
|
|
'id': base_model_id,
|
|
|
'suid': base_model_suid
|
|
|
})
|
|
|
data['base_models'] = json.dumps(base_model)
|
|
|
new_data = copy.copy(data)
|
|
|
|
|
|
with get_session() as session:
|
|
|
model = EnterpriseBusiModel(**new_data)
|
|
|
session.add(model)
|
|
|
session.commit()
|
|
|
|
|
|
return model.id, model.suid
|
|
|
|
|
|
def edit_busi_model(self, data: Dict):
|
|
|
base_model_ids = [int(model_id) for model_id in data['basemodel_ids'].split(',')]
|
|
|
base_model_db = DB_alg_model.ModelRepositry()
|
|
|
base_model = []
|
|
|
for base_model_id in base_model_ids:
|
|
|
base_model_suid = base_model_db.get_suid(base_model_id)
|
|
|
base_model.append({
|
|
|
'id': base_model_id,
|
|
|
'suid': base_model_suid
|
|
|
})
|
|
|
|
|
|
data['base_models'] = json.dumps(base_model)
|
|
|
with get_session() as session:
|
|
|
try:
|
|
|
session.query(EnterpriseBusiModel).filter(EnterpriseBusiModel.id == data['id']).update(data)
|
|
|
except Exception as e:
|
|
|
logging.error("Failed to edit device")
|
|
|
session.commit()
|
|
|
|
|
|
return
|
|
|
|
|
|
|
|
|
def get_busi_model_by_id(self, id: int) -> Optional[EnterpriseBusiModel]:
|
|
|
with get_session() as session:
|
|
|
model = session.query(EnterpriseBusiModel).filter(EnterpriseBusiModel.id == id).first()
|
|
|
return model
|
|
|
|
|
|
|
|
|
def list_enterprise_busi_model(self, entity_id: int, page_no: int, page_size: int) -> Dict[Any, Any]:
|
|
|
"""
|
|
|
获取企业部署的业务模型列表
|
|
|
|
|
|
Args:
|
|
|
session (Session): SQLAlchemy 数据库会话
|
|
|
entity_id (int): 企业 ID
|
|
|
page_no (int): 页码
|
|
|
page_size (int): 每页数量
|
|
|
|
|
|
Returns:
|
|
|
dict: 包含总数和数据列表的字典
|
|
|
"""
|
|
|
with get_session() as session:
|
|
|
total_count = session.query(func.count(EnterpriseBusiModel.id)).filter(EnterpriseBusiModel.entity_id == entity_id).scalar()
|
|
|
|
|
|
models = (
|
|
|
session.query(
|
|
|
EnterpriseBusiModel.id.label("model_id"),
|
|
|
EnterpriseBusiModel.name.label("model_name"),
|
|
|
EnterpriseBusiModel.create_time
|
|
|
)
|
|
|
.filter(EnterpriseBusiModel.entity_id == entity_id)
|
|
|
.offset((page_no - 1) * page_size)
|
|
|
.limit(page_size)
|
|
|
.all()
|
|
|
)
|
|
|
|
|
|
return {
|
|
|
"count": total_count,
|
|
|
"data": [
|
|
|
{
|
|
|
"model_id": model.model_id,
|
|
|
"model_name": model.model_name,
|
|
|
"create_time": model.create_time.strftime("%Y-%m-%d %H:%M:%S"),
|
|
|
}
|
|
|
for model in models
|
|
|
]
|
|
|
}
|
|
|
|
|
|
class EnterpriseBusiModelNodeRepository(object):
|
|
|
# def get_by_id(self, id: int) -> Optional[EnterpriseBusiModelNode]:
|
|
|
# return self.db.query(EnterpriseBusiModelNode).filter(EnterpriseBusiModelNode.id == id).first()
|
|
|
|
|
|
def insert_busi_model_nodes(self, data: Dict):
|
|
|
data['suid'] = shortuuid.ShortUUID().random(10)
|
|
|
link_node_ids = [int(node_id) for node_id in data['node_ids'].split(',')]
|
|
|
with get_session() as session:
|
|
|
for node_id in link_node_ids:
|
|
|
node_db = DB_Node.EnterpriseNodeRepository()
|
|
|
node = node_db.get_node_by_id(node_id)
|
|
|
node_suid = node["suid"]
|
|
|
model_node = EnterpriseBusiModelNode(
|
|
|
suid=shortuuid.ShortUUID().random(10),
|
|
|
entity_suid=data['entity_suid'],
|
|
|
busi_model_id=data['busi_model_id'],
|
|
|
busi_model_suid=data['busi_model_suid'],
|
|
|
node_id=node_id,
|
|
|
node_suid=node_suid,
|
|
|
)
|
|
|
session.add(model_node)
|
|
|
session.commit()
|
|
|
return
|
|
|
|
|
|
|
|
|
def get_nodes_by_busi_model(self, busi_model_id: int) -> List[EnterpriseBusiModelNode]:
|
|
|
with get_session() as session:
|
|
|
nodes = session.query(EnterpriseBusiModelNode).filter(EnterpriseBusiModelNode.busi_model_id == busi_model_id).all()
|
|
|
return nodes
|
|
|
|
|
|
|
|
|
def delete_by_busi_model_id(self, busi_model_id: int) -> None:
|
|
|
with get_session() as session:
|
|
|
session.query(EnterpriseBusiModelNode).filter(EnterpriseBusiModelNode.busi_model_id == busi_model_id).delete()
|
|
|
session.commit() |