summaryrefslogtreecommitdiff
path: root/web-server/opendc/models_old/machine_state.py
diff options
context:
space:
mode:
authorGeorgios Andreadis <info@gandreadis.com>2020-06-29 16:05:23 +0200
committerFabian Mastenbroek <mail.fabianm@gmail.com>2020-08-24 16:18:36 +0200
commit4f9a40abdc7836345113c047f27fcc96800cb3f5 (patch)
treee443d14e34a884b1a4d9c549f81d51202eddd5f7 /web-server/opendc/models_old/machine_state.py
parentcd5f7bf3a72913e1602cb4c575e61ac7d5519be0 (diff)
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.
Diffstat (limited to 'web-server/opendc/models_old/machine_state.py')
-rw-r--r--web-server/opendc/models_old/machine_state.py71
1 files changed, 71 insertions, 0 deletions
diff --git a/web-server/opendc/models_old/machine_state.py b/web-server/opendc/models_old/machine_state.py
new file mode 100644
index 00000000..0e9f7dad
--- /dev/null
+++ b/web-server/opendc/models_old/machine_state.py
@@ -0,0 +1,71 @@
+from opendc.models_old.model import Model
+from opendc.util import database
+
+
+class MachineState(Model):
+ JSON_TO_PYTHON_DICT = {
+ 'MachineState': {
+ 'machineId': 'machine_id',
+ 'temperatureC': 'temperature_c',
+ 'inUseMemoryMb': 'in_use_memory_mb',
+ 'loadFraction': 'load_fraction',
+ 'tick': 'tick'
+ }
+ }
+
+ COLLECTION_NAME = 'machine_states'
+ COLUMNS = ['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(machine_id=row[1], temperature_c=row[4], in_use_memory_mb=row[5], load_fraction=row[6], tick=row[3])
+
+ @classmethod
+ def from_experiment_id(cls, experiment_id):
+ """Query MachineStates by their Experiment id."""
+
+ machine_states = []
+
+ statement = 'SELECT * FROM machine_states WHERE experiment_id = %s'
+ results = database.fetch_all(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 = %s AND machine_states.tick = %s'
+ results = database.fetch_all(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 = %s'
+ result = database.fetch_one(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