2 files changed, 42 insertions, 0 deletions
diff --git a/web-server/opendc/api/v2/experiments/experimentId/task-states/__init__.py b/web-server/opendc/api/v2/experiments/experimentId/task-states/__init__.py
new file mode 100644
index 00000000..e69de29b
--- /dev/null
+++ b/web-server/opendc/api/v2/experiments/experimentId/task-states/__init__.py
diff --git a/web-server/opendc/api/v2/experiments/experimentId/task-states/endpoint.py b/web-server/opendc/api/v2/experiments/experimentId/task-states/endpoint.py
new file mode 100644
index 00000000..c0ae47fc
--- /dev/null
+++ b/web-server/opendc/api/v2/experiments/experimentId/task-states/endpoint.py
@@ -0,0 +1,42 @@
+from opendc.models_old.experiment import Experiment
+from opendc.models_old.task_state import TaskState
+from opendc.util import exceptions
+from opendc.util.rest import Response
+
+
+def GET(request):
+    """Get this Experiment's Task States."""
+
+    # Make sure required parameters are there
+
+    try:
+        request.check_required_parameters(path={'experimentId': 'int'})
+
+    except exceptions.ParameterError as e:
+        return Response(400, str(e))
+
+    # Instantiate an Experiment from the database
+
+    experiment = Experiment.from_primary_key((request.params_path['experimentId'], ))
+
+    # Make sure this Experiment exists
+
+    if not experiment.exists():
+        return Response(404, '{} not found.'.format(experiment))
+
+    # Make sure this user is authorized to view Task States for this Experiment
+
+    if not experiment.google_id_has_at_least(request.google_id, 'VIEW'):
+        return Response(403, 'Forbidden from viewing Task States for {}.'.format(experiment))
+
+    # Get and return the Task States
+
+    if 'tick' in request.params_query:
+        task_states = TaskState.from_experiment_id_and_tick(request.params_path['experimentId'],
+                                                            request.params_query['tick'])
+
+    else:
+        task_states = TaskState.query('experiment_id', request.params_path['experimentId'])
+
+    return Response(200, 'Successfully retrieved Task States for {}.'.format(experiment),
+                    [x.to_JSON() for x in task_states])