From 4f9a40abdc7836345113c047f27fcc96800cb3f5 Mon Sep 17 00:00:00 2001 From: Georgios Andreadis Date: Mon, 29 Jun 2020 16:05:23 +0200 Subject: Prepare web-server repository for monorepo This change prepares the web-server Git repository for the monorepo residing at https://github.com/atlarge-research.com/opendc. To accomodate for this, we move all files into a web-server subdirectory. --- web-server/opendc/models_old/task_state.py | 42 ++++++++++++++++++++++++++++++ 1 file changed, 42 insertions(+) create mode 100644 web-server/opendc/models_old/task_state.py (limited to 'web-server/opendc/models_old/task_state.py') diff --git a/web-server/opendc/models_old/task_state.py b/web-server/opendc/models_old/task_state.py new file mode 100644 index 00000000..cc3fdd89 --- /dev/null +++ b/web-server/opendc/models_old/task_state.py @@ -0,0 +1,42 @@ +from opendc.models_old.model import Model +from opendc.util import database + + +class TaskState(Model): + JSON_TO_PYTHON_DICT = { + 'TaskState': { + 'id': 'id', + 'taskId': 'task_id', + 'experimentId': 'experiment_id', + 'tick': 'tick', + 'flopsLeft': 'flops_left', + 'coresUsed': 'cores_used' + } + } + + COLLECTION_NAME = 'task_states' + + COLUMNS = ['id', 'task_id', 'experiment_id', 'tick', 'flops_left', 'cores_used'] + COLUMNS_PRIMARY_KEY = ['id'] + + @classmethod + def from_experiment_id_and_tick(cls, experiment_id, tick): + """Query Task States by their Experiment id and tick.""" + + task_states = [] + + statement = 'SELECT * FROM task_states WHERE experiment_id = %s AND tick = %s' + results = database.fetch_all(statement, (experiment_id, tick)) + + for row in results: + task_states.append(cls(id=row[0], task_id=row[1], experiment_id=row[2], tick=row[3], flops_left=row[4])) + + return task_states + + 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 TaskState.""" + + if authorization_level in ['EDIT', 'OWN']: + return False + + return True -- cgit v1.2.3