mirror of
https://github.com/tencentmusic/cube-studio.git
synced 2024-12-21 06:19:31 +08:00
113 lines
3.5 KiB
Python
113 lines
3.5 KiB
Python
from flask_appbuilder import Model
|
|
import json
|
|
from myapp import app,db
|
|
from sqlalchemy import (
|
|
Text,
|
|
Enum
|
|
)
|
|
from myapp.models.helpers import AuditMixinNullable
|
|
from sqlalchemy.orm import backref, relationship
|
|
from myapp.models.base import MyappModelBase
|
|
|
|
from flask_appbuilder.models.decorators import renders
|
|
from flask import Markup
|
|
from sqlalchemy import String,Column,Integer,ForeignKey,UniqueConstraint
|
|
metadata = Model.metadata
|
|
conf = app.config
|
|
|
|
|
|
|
|
class Project(Model,AuditMixinNullable,MyappModelBase):
|
|
__tablename__ = 'project'
|
|
id = Column(Integer, primary_key=True)
|
|
name = Column(String(50), nullable=False)
|
|
describe = Column(String(500), nullable=False)
|
|
type = Column(String(50)) # org, job_template, model
|
|
expand = Column(Text(65536), default='{}')
|
|
|
|
export_children = ["user"]
|
|
|
|
__table_args__ = (
|
|
UniqueConstraint('name','type'),
|
|
)
|
|
|
|
def __repr__(self):
|
|
return self.name
|
|
|
|
@renders('expand')
|
|
def expand_html(self):
|
|
return Markup('<pre><code>' + self.expand + '</code></pre>')
|
|
|
|
def get_creators(self):
|
|
creators = db.session().query(Project_User).filter_by(project_id=self.id).all()
|
|
if creators:
|
|
return [creator.user.username for creator in creators]
|
|
else:
|
|
return []
|
|
|
|
@property
|
|
def node_selector(self):
|
|
try:
|
|
expand = json.loads(self.expand) if self.expand else {}
|
|
node_selector = expand.get('node_selector', '')
|
|
return node_selector
|
|
except Exception as e:
|
|
print(e)
|
|
return ''
|
|
|
|
@property
|
|
def volume_mount(self):
|
|
try:
|
|
expand = json.loads(self.expand) if self.expand else {}
|
|
volume_mount = expand.get('volume_mount', '')
|
|
if not volume_mount:
|
|
volume_mount='kubeflow-user-workspace(pvc):/mnt,kubeflow-archives(pvc):/archives'
|
|
return volume_mount
|
|
except Exception as e:
|
|
print(e)
|
|
return ''
|
|
|
|
@property
|
|
def cluster(self):
|
|
all_clusters = conf.get('CLUSTERS')
|
|
default=all_clusters.get(conf.get('ENVIRONMENT'),{})
|
|
try:
|
|
expand = json.loads(self.expand) if self.expand else {}
|
|
project_cluster = expand.get('cluster','')
|
|
if project_cluster and project_cluster in all_clusters:
|
|
return all_clusters[project_cluster]
|
|
return default
|
|
except Exception as e:
|
|
print(e)
|
|
return default
|
|
@property
|
|
def org(self):
|
|
expand = json.loads(self.expand) if self.expand else {}
|
|
return expand.get('org','public')
|
|
|
|
class Project_User(Model,AuditMixinNullable,MyappModelBase):
|
|
__tablename__ = "project_user"
|
|
id = Column(Integer, primary_key=True)
|
|
project_id = Column(Integer, ForeignKey("project.id"),nullable=False)
|
|
project = relationship(
|
|
"Project",
|
|
backref=backref("user", cascade="all, delete-orphan"),
|
|
foreign_keys=[project_id],
|
|
)
|
|
user_id = Column(Integer, ForeignKey("ab_user.id"),nullable=False)
|
|
user = relationship(
|
|
"MyUser",
|
|
backref=backref("user", cascade="all, delete-orphan"),
|
|
foreign_keys=[user_id],
|
|
)
|
|
role = Column(Enum('dev', 'ops','creator'),nullable=False,default='dev')
|
|
# role = Column(String(50), nullable=False, default='dev')
|
|
export_parent = "project"
|
|
|
|
def __repr__(self):
|
|
return self.user.username+"(%s)"%self.role
|
|
|
|
__table_args__ = (
|
|
UniqueConstraint('project_id','user_id'),
|
|
)
|