summaryrefslogtreecommitdiff
path: root/web-server/opendc/api/v2/portfolios
diff options
context:
space:
mode:
Diffstat (limited to 'web-server/opendc/api/v2/portfolios')
-rw-r--r--web-server/opendc/api/v2/portfolios/portfolioId/scenarios/endpoint.py7
1 files changed, 4 insertions, 3 deletions
diff --git a/web-server/opendc/api/v2/portfolios/portfolioId/scenarios/endpoint.py b/web-server/opendc/api/v2/portfolios/portfolioId/scenarios/endpoint.py
index ab32aae2..1c5e0ab6 100644
--- a/web-server/opendc/api/v2/portfolios/portfolioId/scenarios/endpoint.py
+++ b/web-server/opendc/api/v2/portfolios/portfolioId/scenarios/endpoint.py
@@ -1,4 +1,5 @@
from opendc.models.portfolio import Portfolio
+from opendc.models.scenario import Scenario
from opendc.util.rest import Response
@@ -29,14 +30,14 @@ def POST(request):
portfolio.check_exists()
portfolio.check_user_access(request.google_id, True)
- scenario = Portfolio(request.params_body['scenario'])
+ scenario = Scenario(request.params_body['scenario'])
scenario.set_property('portfolioId', request.params_path['portfolioId'])
scenario.set_property('simulationState', 'QUEUED')
scenario.insert()
- portfolio.obj['portfolioIds'].append(portfolio.get_id())
+ portfolio.obj['scenarioIds'].append(scenario.get_id())
portfolio.update()
- return Response(200, 'Successfully added Portfolio.', portfolio.obj)
+ return Response(200, 'Successfully added Scenario.', scenario.obj)