cube-studio/myapp/views/view_team.py

from flask import render_template,redirect
from flask_appbuilder.models.sqla.interface import SQLAInterface
from flask_appbuilder import ModelView, ModelRestApi
from flask_appbuilder import ModelView,AppBuilder,expose,BaseView,has_access
from flask_babel import gettext as __
from flask_babel import lazy_gettext as _
# 将model添加成视图，并控制在前端的显示
from myapp.models.model_team import Project,Project_User
from flask_appbuilder.actions import action
from wtforms import BooleanField, IntegerField, SelectField, StringField,FloatField,DateField,DateTimeField,SelectMultipleField,FormField,FieldList
from flask_appbuilder.models.sqla.filters import FilterEqualFunction, FilterStartsWith,FilterEqual,FilterNotEqual
from wtforms.validators import EqualTo,Length
from flask_babel import lazy_gettext,gettext
from flask_appbuilder.security.decorators import has_access
from myapp.utils import core
from myapp import app, appbuilder,db
from wtforms.ext.sqlalchemy.fields import QuerySelectField
from flask_appbuilder.fieldwidgets import Select2Widget
from myapp.exceptions import MyappException
from myapp import conf, db, get_feature_flags, security_manager,event_logger
from myapp.forms import MyBS3TextFieldWidget
from flask import (
    abort,
    flash,
    g,
    Markup,
    redirect,
    render_template,
    request,
    Response,
    url_for,
)
from .base import (
    api,
    BaseMyappView,
    check_ownership,
    data_payload_response,
    DeleteMixin,
    generate_download_headers,
    get_error_msg,
    get_user_roles,
    handle_api_exception,
    json_error_response,
    json_success,
    MyappFilter,
    MyappModelView,
)
from .baseApi import (
    MyappModelRestApi
)
import pysnooper,datetime,time,json
from flask_appbuilder import CompactCRUDMixin, expose

# table show界面下的
class Project_User_ModelView_Base():
    label_title='组成员'
    datamodel = SQLAInterface(Project_User)
    add_columns = ['project','user','role']
    edit_columns = add_columns
    list_columns = ['user','role']

    add_form_extra_fields = {
        "project": QuerySelectField(
            "项目组",
            query_factory=lambda: db.session.query(Project),
            allow_blank=True,
            widget=Select2Widget(extra_classes="readonly"),
            description='只有creator可以添加修改组成员，可以添加多个creator'
        )
    }
    edit_form_extra_fields = add_form_extra_fields

    # @pysnooper.snoop()
    def pre_add(self, item):
        user_roles = [role.name.lower() for role in list(get_user_roles())]
        if "admin" in user_roles:
            return
        if not g.user.username in item.project.get_creators():
            raise MyappException('just creator can add/edit user')


    def pre_update(self, item):
        user_roles = [role.name.lower() for role in list(get_user_roles())]
        if "admin" in user_roles:
            return
        if not g.user.username in item.project.get_creators():
            raise MyappException('just creator can add/edit user')

class Project_User_ModelView(Project_User_ModelView_Base,CompactCRUDMixin, MyappModelView):
    datamodel = SQLAInterface(Project_User)

appbuilder.add_view_no_menu(Project_User_ModelView)

# 添加api
class Project_User_ModelView_Api(Project_User_ModelView_Base,MyappModelRestApi):
    datamodel = SQLAInterface(Project_User)
    route_base = '/project_user_modelview/api'

appbuilder.add_api(Project_User_ModelView_Api)



# 获取某类project分组
class Project_Filter(MyappFilter):
    # @pysnooper.snoop()
    def apply(self, query, value):
        # user_roles = [role.name.lower() for role in list(get_user_roles())]
        # if "admin" in user_roles:
        #     return query.filter(Project.type == value).order_by(Project.id.desc())
        return query.filter(self.model.type==value).order_by(self.model.id.desc())



# 获取自己参加的某类project分组
class Project_Join_Filter(MyappFilter):
    # @pysnooper.snoop()
    def apply(self, query, value):
        if g.user.is_admin():
            return query.filter(self.model.type == value).order_by(self.model.id.desc())
        join_projects_id = security_manager.get_join_projects_id(db.session)
        return query.filter(self.model.id.in_(join_projects_id)).filter(self.model.type==value).order_by(self.model.id.desc())



# 获取查询自己所在的项目组的project
def filter_join_org_project():
    query = db.session.query(Project)
    # user_roles = [role.name.lower() for role in list(get_user_roles())]
    # if "admin" in user_roles:
    if g.user.is_admin():
        return query.filter(Project.type=='org').order_by(Project.id.desc())

    # 查询自己拥有的项目
    my_user_id = g.user.get_id() if g.user else 0
    owner_ids_query = db.session.query(Project_User.project_id).filter(Project_User.user_id == my_user_id)

    return query.filter(Project.id.in_(owner_ids_query)).filter(Project.type=='org').order_by(Project.id.desc())


class Project_ModelView_Base():
    label_title='项目组'
    datamodel = SQLAInterface(Project)
    base_permissions = ['can_add', 'can_edit', 'can_delete', 'can_list', 'can_show']  # 默认为这些
    base_order = ('id', 'desc')
    list_columns = ['name','user','type']
    related_views = [Project_User_ModelView,]
    add_columns = ['type','name','describe','expand']
    edit_columns = add_columns
    edit_form_extra_fields={
        'type': StringField(
            _(datamodel.obj.lab('type')),
            description="项目分组",
            widget=MyBS3TextFieldWidget(value='org', readonly=1),
            default='org',
        )
    }
    add_form_extra_fields=edit_form_extra_fields

    # @pysnooper.snoop()
    def pre_add_get(self):
        self.edit_form_extra_fields['type'] = StringField(
            _(self.datamodel.obj.lab('type')),
            description="项目分组",
            widget=MyBS3TextFieldWidget(value=self.project_type,readonly=1),
            default=self.project_type,
        )
        self.add_form_extra_fields = self.edit_form_extra_fields


    def pre_add(self, item):
        if item.expand:
            core.validate_json(item.expand)
            item.expand = json.dumps(json.loads(item.expand),indent=4,ensure_ascii=False)

    def pre_update(self, item):
        if item.expand:
            core.validate_json(item.expand)
            item.expand = json.dumps(json.loads(item.expand),indent=4,ensure_ascii=False)
        user_roles = [role.name.lower() for role in list(get_user_roles())]
        if "admin" in user_roles:
            return
        if not g.user.username in item.get_creators():
            raise MyappException('just creator can add/edit')

        # 检测是否具有编辑权限，只有creator和admin可以编辑

    # 打开编辑前，校验权限
    def pre_update_get(self, item):
        self.pre_add_get()
        user_roles = [role.name.lower() for role in list(get_user_roles())]
        if "admin" in user_roles:
            return
        if not g.user.username in item.get_creators():
            flash('just creator can add/edit user','warning')
            raise MyappException('just creator can add/edit user')





    # 添加创始人
    def post_add(self, item):
        creator = Project_User(role='creator',user=g.user,project=item)
        db.session.add(creator)
        db.session.commit()


    # @pysnooper.snoop()
    def post_list(self,items):
        return core.sort_expand_index(items,db.session)



class Project_ModelView_job_template(Project_ModelView_Base,MyappModelView):
    project_type = 'job-template'
    base_filters = [["id", Project_Filter, project_type]]  # 设置权限过滤器
    datamodel = SQLAInterface(Project)
    label_title = '模板分类'

appbuilder.add_view(Project_ModelView_job_template,"模板分类",icon = 'fa-tasks',category = '项目组',category_icon = 'fa-users')


# 添加api
class Project_ModelView_job_template_Api(Project_ModelView_Base,MyappModelRestApi):
    route_base = '/project_modelview/job_template/api'
    datamodel = SQLAInterface(Project)
    project_type = 'job-template'
    base_filters = [["id", Project_Filter, project_type]]  # 设置权限过滤器

appbuilder.add_api(Project_ModelView_job_template_Api)


class Project_ModelView_org(Project_ModelView_Base,MyappModelView):
    project_type='org'
    base_filters = [["id", Project_Filter, project_type]]  # 设置权限过滤器
    datamodel = SQLAInterface(Project)
    label_title = '项目分组'

appbuilder.add_view(Project_ModelView_org,"项目分组",icon = 'fa-sitemap',category = '项目组',category_icon = 'fa-users')

# 添加api
class Project_ModelView_org_Api(Project_ModelView_Base,MyappModelRestApi):
    route_base = '/project_modelview/org/api'
    datamodel = SQLAInterface(Project)
    project_type = 'org'
    base_filters = [["id", Project_Join_Filter, project_type]]  # 设置权限过滤器

appbuilder.add_api(Project_ModelView_org_Api)



class Project_ModelView_train_model(Project_ModelView_Base,MyappModelView):
    project_type = 'model'
    base_filters = [["id", Project_Filter, project_type]]  # 设置权限过滤器
    datamodel = SQLAInterface(Project)
    label_title = '模型分组'

appbuilder.add_view(Project_ModelView_train_model,"模型分组",icon = 'fa-address-book-o',category = '项目组',category_icon = 'fa-users')


# 添加api
class Project_ModelView_train_model_Api(Project_ModelView_Base,MyappModelRestApi):
    route_base = '/project_modelview/model/api'
    datamodel = SQLAInterface(Project)
    project_type = 'model'
    base_filters = [["id", Project_Filter, project_type]]  # 设置权限过滤器

appbuilder.add_api(Project_ModelView_train_model_Api)


# 添加视图和菜单


# 添加api
class Project_ModelView_Api(Project_ModelView_Base,MyappModelRestApi):
    datamodel = SQLAInterface(Project)
    route_base = '/project_modelview/api'

appbuilder.add_api(Project_ModelView_Api)