diff --git a/apps/assets/api/node.py b/apps/assets/api/node.py index 39390447c..cc940757f 100644 --- a/apps/assets/api/node.py +++ b/apps/assets/api/node.py @@ -108,18 +108,18 @@ class NodeChildrenApi(mixins.ListModelMixin, generics.CreateAPIView): return node def get_queryset(self): - queryset = set() + queryset = [] query_all = self.request.query_params.get("all") query_assets = self.request.query_params.get('assets') node = self.get_object() if node == Node.root(): - queryset.add(node) + queryset.append(node) if query_all: children = node.get_all_children() else: children = node.get_children() - queryset.update(set(children)) + queryset.extend(list(children)) if query_assets: assets = node.get_assets() for asset in assets: @@ -128,7 +128,7 @@ class NodeChildrenApi(mixins.ListModelMixin, generics.CreateAPIView): node_fake.parent = node node_fake.value = asset.hostname node_fake.is_node = False - queryset.add(node_fake) + queryset.append(node_fake) queryset = sorted(queryset, key=lambda x: x.is_node, reverse=True) return queryset diff --git a/apps/assets/serializers/asset.py b/apps/assets/serializers/asset.py index 8556be3bf..cf844b5c7 100644 --- a/apps/assets/serializers/asset.py +++ b/apps/assets/serializers/asset.py @@ -18,8 +18,7 @@ class NodeTMPSerializer(serializers.ModelSerializer): class Meta: model = Node - fields = ['id', 'key', 'value', 'parent', 'assets_amount', - 'is_asset'] + fields = ['id', 'key', 'value', 'parent', 'assets_amount', 'is_node'] list_serializer_class = BulkListSerializer @staticmethod