Add ServiceCollection

plum/service_collection.py

@@ -0,0 +1,37 @@
+from .service import Service
+
+def sort_service_dicts(services):
+    # Sort in dependency order
+    def cmp(x, y):
+        x_deps_y = y['name'] in x.get('links', [])
+        y_deps_x = x['name'] in y.get('links', [])
+        if x_deps_y and not y_deps_x:
+            return 1
+        elif y_deps_x and not x_deps_y:
+            return -1
+        return 0
+    return sorted(services, cmp=cmp)
+
+class ServiceCollection(list):
+    @classmethod
+    def from_dicts(cls, client, service_dicts):
+        """
+        Construct a ServiceCollection from a list of dicts representing services.
+        """
+        collection = ServiceCollection()
+        for service_dict in sort_service_dicts(service_dicts):
+            # Reference links by object
+            links = []
+            if 'links' in service_dict:
+                for name in service_dict.get('links', []):
+                    links.append(collection.get(name))
+                del service_dict['links']
+            collection.append(Service(client=client, links=links, **service_dict))
+        return collection
+
+    def get(self, name):
+        for service in self:
+            if service.name == name:
+                return service
+
+