pythonmao
diff --git a/‎bigbang/api/controller/base.py‎
Lines changed: 35 additions & 0 deletions b/‎bigbang/api/controller/base.py‎
Lines changed: 35 additions & 0 deletions
diff --git a/‎bigbang/api/controller/link.py‎
Lines changed: 49 additions & 0 deletions b/‎bigbang/api/controller/link.py‎
Lines changed: 49 additions & 0 deletions
diff --git a/‎bigbang/api/controller/root.py‎ b/‎bigbang/api/controller/root.py‎
@@ -0,0 +1,35 @@
+class APIBase(object):
+    def __init__(self, **kwargs):
+        for field in self.fields:
+            if field in kwargs:
+                value = kwargs[field]
+                setattr(self, field, value)
+
+    def __setattr__(self, field, value):
+        if field in self.fields:
+            validator = self.fields[field]['validate']
+            kwargs = self.fields[field].get('validate_args', {})
+            value = validator(value, **kwargs)
+        super(APIBase, self).__setattr__(field, value)
+
+    def as_dict(self):
+        """Render this object as a dict of its fields."""
+        return {f: getattr(self, f)
+                for f in self.fields
+                if hasattr(self, f)}
+
+    def __json__(self):
+        return self.as_dict()
+
+    def unset_fields_except(self, except_list=None):
+        """Unset fields so they don't appear in the message body.
+
+        :param except_list: A list of fields that won't be touched.
+
+        """
+        if except_list is None:
+            except_list = []
+
+        for k in self.as_dict():
+            if k not in except_list:
+                setattr(self, k, None)
@@ -0,0 +1,49 @@
+import pecan
+
+from bigbang.api.controllers import base
+from bigbang.api.controllers import types
+
+
+def build_url(resource, resource_args, bookmark=False, base_url=None):
+    if base_url is None:
+        base_url = pecan.request.host_url
+
+    template = '%(url)s/%(res)s' if bookmark else '%(url)s/v1/%(res)s'
+    # FIXME(lucasagomes): I'm getting a 404 when doing a GET on
+    # a nested resource that the URL ends with a  '/'.
+    # https://groups.google.com/forum/#!topic/pecan-dev/QfSeviLg5qs
+    template += '%(args)s' if resource_args.startswith('?') else '/%(args)s'
+    return template % {'url': base_url, 'res': resource, 'args': resource_args}
+
+
+class Link(base.APIBase):
+    """A link representation."""
+
+    fields = {
+        'href': {
+            'validate': types.Text.validate
+        },
+        'rel': {
+            'validate': types.Text.validate
+        },
+        'type': {
+            'validate': types.Text.validate
+        },
+    }
+
+    @staticmethod
+    def make_link(rel_name, url, resource, resource_args,
+                  bookmark=False, type=None):
+        href = build_url(resource, resource_args,
+                         bookmark=bookmark, base_url=url)
+        if type is None:
+            return Link(href=href, rel=rel_name)
+        else:
+            return Link(href=href, rel=rel_name, type=type)
+
+    @classmethod
+    def sample(cls):
+        sample = cls(href="http://localhost:8080/containers/"
+                          "eaaca217-e7d8-47b4-bb41-3f99f20eed89",
+                     rel="bookmark")
+        return sample