Implement portfolio endpoints

2 files changed, 10 insertions, 6 deletions

diff --git a/web-server/opendc/models/model.py b/web-server/opendc/models/model.py
index cab283c9..bcb833ae 100644
--- a/web-server/opendc/models/model.py
+++ b/web-server/opendc/models/model.py
@@ -33,8 +33,12 @@ class Model:
             raise ClientError(Response(404, 'Not found.'))
 
     def set_property(self, key, value):
-        """Sets the given property on the enclosed object."""
-        self.obj[key] = value
+        """Sets the given property on the enclosed object, with support for simple nested access."""
+        if '.' in key:
+            keys = key.split('.')
+            self.obj[keys[0]][keys[1]] = value
+        else:
+            self.obj[key] = value
 
     def insert(self):
         """Inserts the enclosed object and generates a UUID for it."""
diff --git a/web-server/opendc/models/experiment.py b/web-server/opendc/models/portfolio.py
index 46373b99..32961b63 100644
--- a/web-server/opendc/models/experiment.py
+++ b/web-server/opendc/models/portfolio.py
@@ -4,10 +4,10 @@ from opendc.util.exceptions import ClientError
 from opendc.util.rest import Response
 
 
-class Experiment(Model):
-    """Model representing a Experiment."""
+class Portfolio(Model):
+    """Model representing a Portfolio."""
 
-    collection_name = 'experiments'
+    collection_name = 'portfolios'
 
     def check_user_access(self, google_id, edit_access):
         """Raises an error if the user with given [google_id] has insufficient access.
@@ -21,4 +21,4 @@ class Experiment(Model):
         authorizations = list(
             filter(lambda x: str(x['projectId']) == str(self.obj['projectId']), user.obj['authorizations']))
         if len(authorizations) == 0 or (edit_access and authorizations[0]['authorizationLevel'] == 'VIEW'):
-            raise ClientError(Response(403, 'Forbidden from retrieving/editing experiment.'))
+            raise ClientError(Response(403, 'Forbidden from retrieving/editing portfolio.'))