Experiment

1 files changed, 42 insertions, 0 deletions

diff --git a/opendc/models_old/task_state.py b/opendc/models_old/task_state.py
new file mode 100644
index 00000000..cc3fdd89
--- /dev/null
+++ b/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