1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
from opendc.models.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'
}
}
TABLE_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.fetchall(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 Use rhas at least the given auth level over this TaskState."""
if authorization_level in ['EDIT', 'OWN']:
return False
return True
|
