mirror of
https://github.com/jumpserver/jumpserver.git
synced 2026-04-09 22:03:26 +00:00
164f48e131
* [Update] 统一url地址 * [Update] 修改api * [Update] 使用规范的签名 * [Update] 修改url * [Update] 修改swagger * [Update] 添加serializer class避免报错 * [Update] 修改token * [Update] 支持api key * [Update] 支持生成api key * [Update] 修改api重定向 * [Update] 修改翻译 * [Update] 添加说明文档 * [Update] 修复浏览器关闭后session不失效的问题 * [Update] 修改一些内容 * [Update] 修改 jms脚本 * [Update] 修改重定向 * [Update] 修改搜索trim * [Update] 修改搜索trim * [Update] 添加sys log * [Bugfix] 修改登陆错误 * [Update] 优化User操作private_token的接口 (#3091) * [Update] 优化User操作private_token的接口 * [Update] 优化User操作private_token的接口 2 * [Bugfix] 解决授权了一个节点，当移动节点后，被移动的节点下的资产会放到未分组节点下的问题 * [Update] 升级jquery * [Update] 默认使用page * [Update] 修改使用Orgmodel view set * [Update] 支持 nv的硬盘 https://github.com/jumpserver/jumpserver/issues/1804 * [UPdate] 解决命令执行宽度问题 * [Update] 优化节点 * [Update] 修改nodes过多时创建比较麻烦 * [Update] 修改导入 * [Update] 节点获取更新 * [Update] 修改nodes * [Update] nodes显示full value * [Update] 统一使用nodes select2 函数 * [Update] 修改磁盘大小小数 * [Update] 修改 Node service * [Update] 优化授权节点 * [Update] 修改 node permission * [Update] 修改asset permission * [Stash] * [Update] 修改node assets api * [Update] 修改tree service，支持资产数量 * [Update] 修改暂时完成 * [Update] 修改一些bug
60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
from django.shortcuts import get_object_or_404
|
|
from rest_framework.viewsets import ModelViewSet
|
|
from rest_framework_bulk import BulkModelViewSet
|
|
from common.mixins import IDInCacheFilterMixin
|
|
|
|
from ..utils import set_to_root_org
|
|
from ..models import Organization
|
|
|
|
__all__ = [
|
|
'RootOrgViewMixin', 'OrgMembershipModelViewSetMixin', 'OrgModelViewSet',
|
|
'OrgBulkModelViewSet',
|
|
]
|
|
|
|
|
|
class RootOrgViewMixin:
|
|
def dispatch(self, request, *args, **kwargs):
|
|
set_to_root_org()
|
|
return super().dispatch(request, *args, **kwargs)
|
|
|
|
|
|
class OrgModelViewSet(IDInCacheFilterMixin, ModelViewSet):
|
|
def get_queryset(self):
|
|
return super().get_queryset().all()
|
|
|
|
|
|
class OrgBulkModelViewSet(IDInCacheFilterMixin, BulkModelViewSet):
|
|
def get_queryset(self):
|
|
queryset = super().get_queryset().all()
|
|
if hasattr(self, 'action') and self.action == 'list' and \
|
|
hasattr(self, 'serializer_class') and \
|
|
hasattr(self.serializer_class, 'setup_eager_loading'):
|
|
queryset = self.serializer_class.setup_eager_loading(queryset)
|
|
return queryset
|
|
|
|
def allow_bulk_destroy(self, qs, filtered):
|
|
return False
|
|
|
|
|
|
class OrgMembershipModelViewSetMixin:
|
|
org = None
|
|
membership_class = None
|
|
lookup_field = 'user'
|
|
lookup_url_kwarg = 'user_id'
|
|
http_method_names = ['get', 'post', 'delete', 'head', 'options']
|
|
|
|
def dispatch(self, request, *args, **kwargs):
|
|
self.org = get_object_or_404(Organization, pk=kwargs.get('org_id'))
|
|
return super().dispatch(request, *args, **kwargs)
|
|
|
|
def get_serializer_context(self):
|
|
context = super().get_serializer_context()
|
|
context['org'] = self.org
|
|
return context
|
|
|
|
def get_queryset(self):
|
|
queryset = self.membership_class.objects.filter(organization=self.org)
|
|
return queryset
|