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('
' + self.expand + '
')
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'),
)