Add job to models

2 files changed, 21 insertions, 2 deletions

diff --git a/opendc/models/job.py b/opendc/models/job.py
new file mode 100644
index 00000000..fb133a72
--- /dev/null
+++ b/opendc/models/job.py
@@ -0,0 +1,19 @@
+from opendc.models.model import Model
+
+class Job(Model):
+
+    JSON_TO_PYTHON_DICT = {
+        'Job': {
+            'id': 'id',
+            'name': 'name'
+        }
+    }
+
+    TABLE_NAME = 'jobs'
+    COLUMNS = ['id', 'name']
+    COLUMNS_PRIMARY_KEY = ['id']
+
+    def google_id_has_at_least(self, google_id, authorization_level):
+        """Return True if the user has at least the given auth level over this Job."""
+
+        return authorization_level not in ['EDIT', 'OWN']
diff --git a/opendc/models/task.py b/opendc/models/task.py
index aa9838a5..da0f5785 100644
--- a/opendc/models/task.py
+++ b/opendc/models/task.py
@@ -7,14 +7,14 @@ class Task(Model):
             'id': 'id',
             'startTick': 'start_tick',
             'totalFlopCount': 'total_flop_count',
-            'traceId': 'trace_id',
+            'jobId': 'job_id',
             'taskDependencyId': 'task_dependency_id',
             'parallelizability': 'parallelizability'
         }
     }
 
     TABLE_NAME = 'tasks'
-    COLUMNS = ['id', 'start_tick', 'total_flop_count', 'trace_id', 'task_dependency_id', 'parallelizability']
+    COLUMNS = ['id', 'start_tick', 'total_flop_count', 'job_id', 'task_dependency_id', 'parallelizability']
     COLUMNS_PRIMARY_KEY = ['id']
 
     def google_id_has_at_least(self, google_id, authorization_level):