diff options
Diffstat (limited to 'opendc/models')
| -rw-r--r-- | opendc/models/job.py | 19 | ||||
| -rw-r--r-- | opendc/models/task.py | 4 |
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): |