From 7e3cf908a1b10c0901e3c846b30b1fcb6c00ac7e Mon Sep 17 00:00:00 2001
From: ibuler <ibuler@qq.com>
Date: Fri, 27 Jul 2018 11:09:37 +0800
Subject: [PATCH] =?UTF-8?q?[Update]=20=E4=BF=AE=E6=94=B9get=5Fall=5Fassets?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/assets/models/node.py | 40 ++++++++++++++++++++++----------------
 1 file changed, 23 insertions(+), 17 deletions(-)

diff --git a/apps/assets/models/node.py b/apps/assets/models/node.py
index 8507906c8..34becbdb6 100644
--- a/apps/assets/models/node.py
+++ b/apps/assets/models/node.py
@@ -110,16 +110,22 @@ class Node(OrgModelMixin):
 
     def get_all_assets(self):
         from .asset import Asset
-        if self.is_root():
-            assets = Asset.objects.all()
+        pattern = r'^{0}$|^{0}:'.format(self.key)
+        args = []
+        kwargs = {}
+        if self.is_default_node():
+            args.append(Q(nodes__key__regex=pattern) | Q(nodes__key=None))
         else:
-            pattern = r'^{0}$|^{0}:'.format(self.key)
-            assets = Asset.objects.filter(nodes__key__regex=pattern)
+            kwargs['nodes__key__regex'] = pattern
+        assets = Asset.objects.filter(*args, **kwargs)
         return assets
 
     def get_all_valid_assets(self):
         return self.get_all_assets().valid()
 
+    def is_default_node(self):
+        return self.is_root() and self.key == '0'
+
     def is_root(self):
         if self.key.isdigit():
             return True
@@ -129,7 +135,7 @@ class Node(OrgModelMixin):
     @property
     def parent(self):
         if self.is_root():
-            return self.__class__.root()
+            return self
         parent_key = ":".join(self.key.split(":")[:-1])
         try:
             parent = self.__class__.objects.get(key=parent_key)
@@ -139,17 +145,17 @@ class Node(OrgModelMixin):
 
     @parent.setter
     def parent(self, parent):
-        if self.is_node:
-            children = self.get_all_children()
-            old_key = self.key
-            with transaction.atomic():
-                self.key = parent.get_next_child_key()
-                for child in children:
-                    child.key = child.key.replace(old_key, self.key, 1)
-                    child.save()
-                self.save()
-        else:
-            self.key = parent.key+':fake'
+        if not self.is_node:
+            self.key = parent.key + ':fake'
+            return
+        children = self.get_all_children()
+        old_key = self.key
+        with transaction.atomic():
+            self.key = parent.get_next_child_key()
+            for child in children:
+                child.key = child.key.replace(old_key, self.key, 1)
+                child.save()
+            self.save()
 
     def get_ancestor(self, with_self=False):
         if self.is_root():
@@ -175,7 +181,7 @@ class Node(OrgModelMixin):
             set_current_org(Organization.root())
             org_nodes_roots = cls.objects.filter(key__regex=r'^[0-9]+$')
             org_nodes_roots_keys = org_nodes_roots.values_list('key', flat=True)
-            max_value = max([int(k) for k in org_nodes_roots_keys]) if org_nodes_roots_keys else 0
+            max_value = max([int(k) for k in org_nodes_roots_keys]) if org_nodes_roots_keys else -1
             set_current_org(_current_org)
             root = cls.objects.create(key=str(max_value+1), value=_current_org.name)
             return root