Merge pull request #7 from atlarge-research/feature/portfolio-scenario-endpoints

Portfolio and scenario endpoints

1 files changed, 36 insertions, 0 deletions

diff --git a/web-server/opendc/api/v2/projects/projectId/portfolios/endpoint.py b/web-server/opendc/api/v2/projects/projectId/portfolios/endpoint.py
new file mode 100644
index 00000000..c51dee14
--- /dev/null
+++ b/web-server/opendc/api/v2/projects/projectId/portfolios/endpoint.py
@@ -0,0 +1,36 @@
+from opendc.models.portfolio import Portfolio
+from opendc.models.project import Project
+from opendc.util.rest import Response
+
+
+def POST(request):
+    """Add a new Portfolio for this Project."""
+
+    request.check_required_parameters(path={'projectId': 'string'},
+                                      body={
+                                          'portfolio': {
+                                              'name': 'string',
+                                              'targets': {
+                                                  'enabledMetrics': 'list',
+                                                  'repeatsPerScenario': 'int',
+                                              },
+                                          }
+                                      })
+
+    project = Project.from_id(request.params_path['projectId'])
+
+    project.check_exists()
+    project.check_user_access(request.google_id, True)
+
+    portfolio = Portfolio(request.params_body['portfolio'])
+
+    portfolio.set_property('projectId', request.params_path['projectId'])
+    portfolio.set_property('scenarioIds', [])
+    portfolio.set_property('baseScenarioId', '-1')
+
+    portfolio.insert()
+
+    project.obj['portfolioIds'].append(portfolio.get_id())
+    project.update()
+
+    return Response(200, 'Successfully added Portfolio.', portfolio.obj)