diff --git a/apps/assets/api/node.py b/apps/assets/api/node.py index 939444ccc..2ea2d112c 100644 --- a/apps/assets/api/node.py +++ b/apps/assets/api/node.py @@ -45,16 +45,15 @@ class NodeViewSet(BulkModelViewSet): def get_serializer_class(self): show_current_asset = self.request.query_params.get('show_current_asset') - print(show_current_asset) if show_current_asset: return serializers.NodeCurrentSerializer else: return serializers.NodeSerializer - def perform_create(self, serializer): - child_key = Node.root().get_next_child_key() - serializer.validated_data["key"] = child_key - serializer.save() + # def perform_create(self, serializer): + # child_key = Node.root().get_next_child_key() + # serializer.validated_data["key"] = child_key + # serializer.save() class NodeWithAssetsApi(generics.ListAPIView): @@ -91,16 +90,27 @@ class NodeChildrenApi(mixins.ListModelMixin, generics.CreateAPIView): serializer_class = serializers.NodeSerializer instance = None + def counter(self): + values = [ + child.value[child.value.rfind(' '):] + for child in self.get_object().get_children() + if child.value.startswith("新节点 ") + ] + values = [int(value) for value in values if value.strip().isdigit()] + count = max(values)+1 if values else 1 + return count + def post(self, request, *args, **kwargs): if not request.data.get("value"): - request.data["value"] = _("New node {}").format( - Node.root().get_next_child_key().split(":")[-1] - ) + request.data["value"] = _("New node {}").format(self.counter()) return super().post(request, *args, **kwargs) def create(self, request, *args, **kwargs): instance = self.get_object() value = request.data.get("value") + values = [child.value for child in instance.get_children()] + if value in values: + raise AssertionError('The same level node name cannot be the same') node = instance.create_child(value=value) return Response( {"id": node.id, "key": node.key, "value": node.value}, @@ -199,6 +209,9 @@ class NodeRemoveAssetsApi(generics.UpdateAPIView): instance = self.get_object() if instance != Node.root(): instance.assets.remove(*tuple(assets)) + else: + assets = [asset for asset in assets if asset.nodes.count() > 1] + instance.assets.remove(*tuple(assets)) class NodeReplaceAssetsApi(generics.UpdateAPIView): diff --git a/apps/assets/models/node.py b/apps/assets/models/node.py index 89e4a6b70..14ac775fc 100644 --- a/apps/assets/models/node.py +++ b/apps/assets/models/node.py @@ -12,7 +12,10 @@ __all__ = ['Node'] class Node(models.Model): id = models.UUIDField(default=uuid.uuid4, primary_key=True) key = models.CharField(unique=True, max_length=64, verbose_name=_("Key")) # '1:1:1:1' - value = models.CharField(max_length=128, unique=True, verbose_name=_("Value")) + # value = models.CharField( + # max_length=128, unique=True, verbose_name=_("Value") + # ) + value = models.CharField(max_length=128, verbose_name=_("Value")) child_mark = models.IntegerField(default=0) date_create = models.DateTimeField(auto_now_add=True) @@ -48,7 +51,7 @@ class Node(models.Model): return child def get_children(self): - return self.__class__.objects.filter(key__regex=r'{}:[0-9]+$'.format(self.key)) + return self.__class__.objects.filter(key__regex=r'^{}:[0-9]+$'.format(self.key)) def get_all_children(self): return self.__class__.objects.filter(key__startswith='{}:'.format(self.key)) diff --git a/apps/assets/serializers/node.py b/apps/assets/serializers/node.py index 225d5655a..9f04fd465 100644 --- a/apps/assets/serializers/node.py +++ b/apps/assets/serializers/node.py @@ -51,6 +51,15 @@ class NodeSerializer(serializers.ModelSerializer): fields = ['id', 'key', 'value', 'parent', 'assets_amount', 'is_node'] list_serializer_class = BulkListSerializer + def update(self, instance, validated_data): + value = validated_data.get('value') + if not instance.is_root(): + children = instance.parent.get_children().exclude(key=instance.key) + values = [child.value for child in children] + if value in values: + raise AssertionError('The same level node name cannot be the same') + return super().update(instance, validated_data) + @staticmethod def get_parent(obj): return obj.parent.id diff --git a/apps/assets/templates/assets/_asset_list_modal.html b/apps/assets/templates/assets/_asset_list_modal.html index c5cd48857..db7d165e0 100644 --- a/apps/assets/templates/assets/_asset_list_modal.html +++ b/apps/assets/templates/assets/_asset_list_modal.html @@ -98,7 +98,10 @@ function initTree2() { $.get("{% url 'api-assets:node-list' %}", function(data, status){ $.each(data, function (index, value) { value["pId"] = value["parent"]; - value["open"] = true; + {#value["open"] = true;#} + if (value["key"] === "0") { + value["open"] = true; + } value["name"] = value["value"] + ' (' + value['assets_amount'] + ')'; value['value'] = value['value']; }); diff --git a/apps/assets/templates/assets/asset_detail.html b/apps/assets/templates/assets/asset_detail.html index 0b1b2865a..b07a7c348 100644 --- a/apps/assets/templates/assets/asset_detail.html +++ b/apps/assets/templates/assets/asset_detail.html @@ -190,7 +190,7 @@