From 86a50a4f6df9ece982743a3b7ca510846d248909 Mon Sep 17 00:00:00 2001 From: leonoverweel Date: Tue, 24 Jan 2017 12:05:15 +0100 Subject: Initial commit --- opendc/models/experiment.py | 51 +++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 51 insertions(+) create mode 100644 opendc/models/experiment.py (limited to 'opendc/models/experiment.py') diff --git a/opendc/models/experiment.py b/opendc/models/experiment.py new file mode 100644 index 00000000..35056d0f --- /dev/null +++ b/opendc/models/experiment.py @@ -0,0 +1,51 @@ +from opendc.models.model import Model +from opendc.models.simulation import Simulation +from opendc.util import database, exceptions + +class Experiment(Model): + + JSON_TO_PYTHON_DICT = { + 'Experiment': { + 'id': 'id', + 'simulationId': 'simulation_id', + 'pathId': 'path_id', + 'traceId': 'trace_id', + 'schedulerName': 'scheduler_name', + 'name': 'name' + } + } + + TABLE_NAME = 'experiments' + COLUMNS = ['id', 'simulation_id', 'path_id', 'trace_id', 'scheduler_name', 'name'] + COLUMNS_PRIMARY_KEY = ['id'] + + def get_last_simulated_tick(self): + """Get this Experiment's last simulated tick.""" + + statement = ''' + SELECT tick + FROM task_states + WHERE experiment_id = ? + ORDER BY tick DESC + LIMIT 1 + ''' + result = database.fetchone(statement, (self.id,)) + + if result is not None: + return result[0] + + return -1 + + 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 Experiment.""" + + # Get the Simulation + + try: + simulation = Simulation.from_primary_key((self.simulation_id,)) + except exceptions.RowNotFoundError: + return False + + # Check the Simulation's Authorization + + return simulation.google_id_has_at_least(google_id, authorization_level) -- cgit v1.2.3