Initial commit

1 files changed, 44 insertions, 0 deletions

diff --git a/opendc/api/v1/simulations/simulationId/experiments/experimentId/statistics/task-durations/endpoint.py b/opendc/api/v1/simulations/simulationId/experiments/experimentId/statistics/task-durations/endpoint.py
new file mode 100644
index 00000000..1bb60ca2
--- /dev/null
+++ b/opendc/api/v1/simulations/simulationId/experiments/experimentId/statistics/task-durations/endpoint.py
@@ -0,0 +1,44 @@
+from opendc.models.experiment import Experiment
+from opendc.models.task_duration import TaskDuration
+from opendc.util import database, exceptions
+from opendc.util.rest import Response
+
+def GET(request):
+    """Get this Experiment's Task Durations."""
+
+    # Make sure required parameters are there
+
+    try:
+        request.check_required_parameters(
+            path = {
+                'simulationId': 'int',
+                'experimentId': 'int'
+            }
+        )
+
+    except exceptions.ParameterError as e:
+        return Response(400, e.message)
+
+    # Instantiate an Experiment from the database
+
+    experiment = Experiment.from_primary_key((request.params_path['experimentId'],))
+
+    # Make sure this Experiment exisits
+
+    if not experiment.exists():
+        return Response(404, '{} not found.'.format(experiment))
+
+    # Make sure this user is authorized to view this Experiment's Task Durations
+
+    if not experiment.google_id_has_at_least(request.google_id, 'VIEW'):
+        return Response(403, 'Forbidden from viewing Task Durations for {}.'.format(experiment))
+
+    # Get and return the Task Durations
+
+    task_durations = TaskDuration.from_experiment_id(request.params_path['experimentId'])
+
+    return Response(
+        200,
+        'Successfully retrieved Task Durations for {}.'.format(experiment),
+        [x.to_JSON() for x in task_durations]
+    )