From: Scott Baker <smbaker@gmail.com>
Date: Sun, 13 Jul 2014 18:17:58 +0000 (-0700)
Subject: start working on expanded objects
X-Git-Url: http://git.onelab.eu/?a=commitdiff_plain;h=5f4770d9c1d4e573ae71d6d96948695d7ecbc7fa;p=plstackapi.git

start working on expanded objects
---

diff --git a/planetstack/core/xoslib/objects/sliceplus.py b/planetstack/core/xoslib/objects/sliceplus.py
new file mode 100644
index 0000000..a68602d
--- /dev/null
+++ b/planetstack/core/xoslib/objects/sliceplus.py
@@ -0,0 +1,22 @@
+from core.models.slice import Slice
+
+class SlicePlus(Slice):
+    def getSliceInfo(self, user=None):
+        used_sites = []
+        used_deployments = []
+        sliverCount = 0
+        for sliver in self.slivers.all():
+            site = sliver.node.site
+            deployment = sliver.node.deployment
+            used_sites[site.name] = used.sites.get(site.name, 0) + 1
+            used_deployments[deployment.name] = used_deployments.get(deployment.name, 0) + 1
+
+        roles = []
+        if (user!=None):
+            roles = [x.role for x in self.slice_privileges.filter(user=user)]
+
+        return {"sitesUsed": used_sites,
+                "deploymentsUsed": used_deployments,
+                "sliverCount": sliceCount,
+                "siteCount": len(used_sites.keys()),
+                "roles": roles}