Add DB handlers and rename to v2

5 files changed, 71 insertions, 0 deletions

diff --git a/opendc/api/v2/jobs/__init__.py b/opendc/api/v2/jobs/__init__.py
new file mode 100644
index 00000000..e69de29b
--- /dev/null
+++ b/opendc/api/v2/jobs/__init__.py
diff --git a/opendc/api/v2/jobs/jobId/__init__.py b/opendc/api/v2/jobs/jobId/__init__.py
new file mode 100644
index 00000000..e69de29b
--- /dev/null
+++ b/opendc/api/v2/jobs/jobId/__init__.py
diff --git a/opendc/api/v2/jobs/jobId/endpoint.py b/opendc/api/v2/jobs/jobId/endpoint.py
new file mode 100644
index 00000000..da4dcd9d
--- /dev/null
+++ b/opendc/api/v2/jobs/jobId/endpoint.py
@@ -0,0 +1,34 @@
+from opendc.models.job import Job
+from opendc.util import exceptions
+from opendc.util.rest import Response
+
+
+def GET(request):
+    """Get this Job."""
+
+    # Make sure required parameters are there
+
+    try:
+        request.check_required_parameters(
+            path={
+                'jobId': 'int'
+            }
+        )
+
+    except exceptions.ParameterError as e:
+        return Response(400, e.message)
+
+    # Instantiate a Job and make sure it exists
+
+    job = Job.from_primary_key((request.params_path['jobId'],))
+
+    if not job.exists():
+        return Response(404, '{} not found.'.format(job))
+
+    # Return this Job
+
+    return Response(
+        200,
+        'Successfully retrieved {}.'.format(job),
+        job.to_JSON()
+    )
diff --git a/opendc/api/v2/jobs/jobId/tasks/__init__.py b/opendc/api/v2/jobs/jobId/tasks/__init__.py
new file mode 100644
index 00000000..e69de29b
--- /dev/null
+++ b/opendc/api/v2/jobs/jobId/tasks/__init__.py
diff --git a/opendc/api/v2/jobs/jobId/tasks/endpoint.py b/opendc/api/v2/jobs/jobId/tasks/endpoint.py
new file mode 100644
index 00000000..04ac5b8c
--- /dev/null
+++ b/opendc/api/v2/jobs/jobId/tasks/endpoint.py
@@ -0,0 +1,37 @@
+from opendc.models.job import Job
+from opendc.models.task import Task
+from opendc.util import exceptions
+from opendc.util.rest import Response
+
+
+def GET(request):
+    """Get this Job's Tasks."""
+
+    # Make sure required parameters are there
+
+    try:
+        request.check_required_parameters(
+            path={
+                'jobId': 'int'
+            }
+        )
+
+    except exceptions.ParameterError as e:
+        return Response(400, e.message)
+
+    # Instantiate a Job and make sure it exists
+
+    job = Job.from_primary_key((request.params_path['jobId'],))
+
+    if not job.exists():
+        return Response(404, '{} not found.'.format(job))
+
+    # Get and return the Tasks
+
+    tasks = Task.query('job_id', request.params_path['jobId'])
+
+    return Response(
+        200,
+        'Successfully retrieved Tasks for {}.'.format(job),
+        [x.to_JSON() for x in tasks]
+    )