mirror of
https://github.com/tencentmusic/cube-studio.git
synced 2025-01-12 13:44:29 +08:00
84 lines
2.6 KiB
Python
84 lines
2.6 KiB
Python
from flask_appbuilder import Model
|
|
from sqlalchemy import Column, Integer, String, ForeignKey,Float
|
|
from sqlalchemy.orm import relationship
|
|
import datetime,time,json
|
|
from sqlalchemy import (
|
|
Boolean,
|
|
Column,
|
|
create_engine,
|
|
DateTime,
|
|
ForeignKey,
|
|
Integer,
|
|
MetaData,
|
|
String,
|
|
Table,
|
|
Text,
|
|
Enum,
|
|
)
|
|
from myapp.utils.py.py_k8s import K8s
|
|
from myapp.models.helpers import AuditMixinNullable, ImportMixin
|
|
from flask import escape, g, Markup, request
|
|
from .model_team import Project
|
|
from .model_job import Pipeline
|
|
from myapp import app,db
|
|
from myapp.models.base import MyappModelBase
|
|
from myapp.models.helpers import ImportMixin
|
|
# 添加自定义model
|
|
from sqlalchemy import Column, Integer, String, ForeignKey ,Date,DateTime
|
|
from flask_appbuilder.models.decorators import renders
|
|
from flask import Markup
|
|
import datetime
|
|
metadata = Model.metadata
|
|
conf = app.config
|
|
|
|
|
|
# 定义训练 model
|
|
class Training_Model(Model,AuditMixinNullable,MyappModelBase):
|
|
__tablename__ = 'model'
|
|
id = Column(Integer, primary_key=True)
|
|
name = Column(String(100), nullable=False)
|
|
version = Column(String(100))
|
|
describe = Column(String(1000))
|
|
path = Column(String(200))
|
|
download_url = Column(String(200))
|
|
project_id = Column(Integer, ForeignKey('project.id')) # 定义外键
|
|
project = relationship(
|
|
Project, foreign_keys=[project_id]
|
|
)
|
|
pipeline_id = Column(Integer,default=0) # 定义外键
|
|
run_id = Column(String(100),nullable=False) # 可能同一个pipeline产生多个模型
|
|
run_time = Column(String(100))
|
|
framework = Column(String(100))
|
|
metrics = Column(Text,default='{}')
|
|
md5 = Column(String(200),default='')
|
|
api_type = Column(String(100))
|
|
|
|
def __repr__(self):
|
|
return self.name
|
|
|
|
@property
|
|
def pipeline_url(self):
|
|
if self.pipeline_id:
|
|
pipeline = db.session.query(Pipeline).filter_by(id=self.pipeline_id).first()
|
|
if pipeline:
|
|
return Markup(f'<a target=_blank href="/frontend/showOutLink?url=%2Fstatic%2Fappbuilder%2Fvison%2Findex.html%3Fpipeline_id%3D{self.pipeline_id}">{pipeline.describe}</a>')
|
|
|
|
return Markup(f'未知')
|
|
|
|
@property
|
|
def project_url(self):
|
|
if self.project:
|
|
return Markup(f'{self.project.name}({self.project.describe})')
|
|
elif self.pipeline and self.pipeline.project:
|
|
return Markup(f'{self.pipeline.project.name}({self.pipeline.project.describe})')
|
|
else:
|
|
return Markup(f'未知')
|
|
|
|
@property
|
|
def deploy(self):
|
|
ops=f'''
|
|
<a href="/training_model_modelview/deploy/{self.id}">发布</a>
|
|
'''
|
|
return Markup(ops)
|
|
|