update select_by_user()

[plstackapi.git] / planetstack / core / models / network.py

diff --git a/planetstack/core/models/network.py b/planetstack/core/models/network.py
index 23ef6ea..57d5530 100644 (file)
--- a/planetstack/core/models/network.py
+++ b/planetstack/core/models/network.py
@@ -51,15 +51,16 @@ class Network(PlCoreBase):
         super(Network, self).save(*args, **kwds)
 
     def can_update(self, user):
-        return self.slice.can_update(user)
+        return self.owner.can_update(user)
 
     @staticmethod
     def select_by_user(user):
         if user.is_admin:
             qs = Network.objects.all()
         else:
-            slice_ids = [s.id for s in Slice.select_by_user(user)]
-            qs = Network.objects.filter(id__in=slice_ids)
+            slices = Slice.select_by_user(user)
+            #slice_ids = [s.id for s in Slice.select_by_user(user)]
+            qs = Network.objects.filter(owner__in=slices)
         return qs
 
 class NetworkSlice(PlCoreBase):