Initial commit

1 files changed, 79 insertions, 0 deletions

diff --git a/opendc/models/machine_state.py b/opendc/models/machine_state.py
new file mode 100644
index 00000000..44bc6746
--- /dev/null
+++ b/opendc/models/machine_state.py
@@ -0,0 +1,79 @@
+from opendc.models.model import Model
+from opendc.util import database
+
+class MachineState(Model):
+
+    JSON_TO_PYTHON_DICT = {
+        'MachineState': {
+            'taskId': 'task_id',
+            'machineId': 'machine_id',
+            'temperatureC': 'temperature_c',
+            'inUseMemoryMb': 'in_use_memory_mb',
+            'loadFraction': 'load_fraction',
+            'tick': 'tick'
+        }
+    }
+
+    TABLE_NAME = 'machine_states'
+    COLUMNS = ['id', 'task_id', 'machine_id', 'experiment_id', 'tick', 'temperature_c', 'in_use_memory_mb', 'load_fraction']
+
+    COLUMNS_PRIMARY_KEY= ['id']
+
+    @classmethod
+    def _from_database_row(cls, row):
+        """Instantiate a MachineState from a database row (including tick from the TaskState)."""
+
+        return cls(
+            task_id = row[1],
+            machine_id = row[2],
+            temperature_c = row[5],
+            in_use_memory_mb = row[6],
+            load_fraction = row[7],
+            tick = row[4]
+        )
+
+    @classmethod
+    def from_experiment_id(cls, experiment_id):
+        """Query MachineStates by their Experiment id."""
+
+        machine_states = []
+        
+        statement = 'SELECT * FROM machine_states WHERE experiment_id = ?'
+        results = database.fetchall(statement, (experiment_id,))
+ 
+        for row in results:    
+            machine_states.append(cls._from_database_row(row))
+
+        return machine_states
+
+    @classmethod
+    def from_experiment_id_and_tick(cls, experiment_id, tick):
+        """Query MachineStates by their Experiment id and tick."""
+
+        machine_states = []
+ 
+        statement = 'SELECT * FROM machine_states WHERE experiment_id = ? AND machine_states.tick = ?'
+        results = database.fetchall(statement, (experiment_id, tick))
+ 
+        for row in results:    
+            machine_states.append(cls._from_database_row(row))
+
+        return machine_states
+
+    def read(self):
+        """Read this MachineState by also getting its tick."""
+
+        super(MachineState, self).read()
+
+        statement = 'SELECT tick FROM task_states WHERE id = ?'
+        result = database.fetchone(statement, (self.task_state_id,))
+
+        self.tick = result[0]
+
+    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 MachineState."""
+
+        if authorization_level in ['EDIT', 'OWN']:
+            return False
+
+        return True