コード例 #1
0
    def test_get_node_detail_with_scope(self):
        node, _ = Node.retrieve_node('d_level_1')
        node.user_scope = ['not_existed']
        node.save()

        res = self.client.get(reverse('siteapi:node_detail', args=('d_level_1', )))
        self.assertEqual([], res.json()['user_scope'])
コード例 #2
0
ファイル: migrate.py プロジェクト: ttloveyy/arkid-core
    def core_post(users_file, node_uid):
        '''
        creat or update users
        '''
        cli = CLI()

        reader = csv.DictReader(io.StringIO(users_file.read().decode('utf-8')))
        res = []
        users = []
        for index, row in enumerate(reader):
            username = row.get('username')
            user = User.valid_objects.filter(username=username).first()

            try:
                if user:
                    cli.update_user(user, row)
                else:
                    user = cli.create_user(row)
            except ValidationError as exc:
                raise ValidationError({index + 1: exc.detail})

            users.append(user)    # 重复导入的后果可以接受,故不处理
            res.append(UserCSVSerializer(user).data)

        if node_uid:
            node, node_subject = TreeNode.retrieve_node(node_uid)
            if node_subject == 'dept':
                cli.add_users_to_dept(users, node)
            elif node_subject == 'group':
                cli.add_users_to_group(users, node)
            else:
                raise ValueError

        return res
コード例 #3
0
ファイル: tasks.py プロジェクト: zhangdeng-github/arkid-core
def update_nodes_cache(node_uids):
    '''
    更新节点缓存
    '''
    print("update_nodes_cache", node_uids)
    nodes = Node.retrieve_nodes(node_uids)
    for node in nodes:
        node.update_cache()
コード例 #4
0
    def all_manage_node_uids(self):
        '''
        所有可管理的节点(包含直接管理的节点,及其下属节点)
        '''

        res = set()
        for node_uid in self.manage_node_uids:
            res.update(Node.get_downstream_uids(node_uid))
        return res
コード例 #5
0
ファイル: group.py プロジェクト: zhiyuChen-aqumon/arkid-core
 def get_nodes(instance):
     '''
     管理节点范围
     '''
     if instance.scope_subject == 2:
         for node in Node.retrieve_nodes(instance.nodes):
             yield {
                 'node_uid': node.node_uid,
                 'node_subject': node.node_subject,
                 'name': node.name,
             }
コード例 #6
0
ファイル: shortcut.py プロジェクト: ttloveyy/arkid-core
 def retrieve_nodes(request, uids):
     '''
     获取节点
     '''
     res = []
     for uid in uids:
         node, _ = Node.retrieve_node(uid)
         if not node or not node.is_visible_to_manager(request.user):
             raise ValidationError({'node_uids': [f'{uid} not found']})
         res.append(node.detail_serializer.data)
     return res
コード例 #7
0
 def all_node_uids(self):
     '''
     所有直属节点以及隶属节点的uid
     '''
     key = f'oneid:user:{self.username}:upstream_node'
     cache_data = cache.get(key)
     if cache_data is None:
         res = set()
         for parent_node in Node.retrieve_nodes(self.node_uids):
             res.update(parent_node.upstream_uids)
         cache.set(key, res)
         return res
     return cache_data
コード例 #8
0
    def all_manage_node_uids(self):
        '''
        所有可管理的节点(包含直接管理的节点,及其下属节点)
        '''
        res = set()
        for node_uid in self.manage_node_uids:
            res.update(Node.get_downstream_uids(node_uid))

        # 过滤根节点管理权限
        discard_node = {'g_root', 'g_manager', 'g_extern', 'g_intra'}
        for node in discard_node:
            res.discard(node)
        return res
コード例 #9
0
ファイル: node.py プロジェクト: zhiyuChen-aqumon/arkid-core
    def get_object(self):
        '''
        find node
        '''
        node, _ = Node.retrieve_node(self.kwargs['uid'])
        if node is None:
            raise NotFound

        if node.__class__ == Dept:
            self.serializer_class = DeptTreeSerializer
        else:
            self.serializer_class = GroupTreeSerializer
        return node
コード例 #10
0
ファイル: migrate.py プロジェクト: ttloveyy/arkid-core
    def post(self, request, *args, **kwargs):    # pylint: disable=unused-argument
        '''
        导入
        '''
        users_file = request.FILES.get('users', None)
        if not users_file:
            raise ValidationError({'users': ["this field is required"]})

        node_uid = request.data.get('node_uid', '')

        if node_uid:
            node, _ = TreeNode.retrieve_node(node_uid)
            if not node:
                raise NotFound

        res = self.core_post(users_file, node_uid='')

        return Response(res)
コード例 #11
0
ファイル: node.py プロジェクト: zhiyuChen-aqumon/arkid-core
    def get_object(self):
        '''
        find node
        '''
        node, node_subject = Node.retrieve_node(self.kwargs['uid'])
        if node is None:
            raise NotFound

        try:
            self.check_object_permissions(self.request, node)
        except PermissionDenied:
            raise NotFound

        if node_subject == 'dept':
            self.serializer_class = DeptSerializer
        else:
            self.serializer_class = GroupSerializer
        return node