-
Notifications
You must be signed in to change notification settings - Fork 0
/
study_manager.py
102 lines (79 loc) · 3.23 KB
/
study_manager.py
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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
import json
import importlib
import os
from event_manager import EventManager
from sim_clock import Clock
from sim_user_manager import UserManager
class StudyManager:
def __init__(self):
self.user_manager = UserManager()
self.event_manager = EventManager(self)
self.clock = Clock(self.event_manager)
self.study = {}
self.interventions = {}
self.users = {}
self.init_study()
self.init_users()
def start(self):
self.clock.start()
def init_study(self):
f = open('studyconfig.json')
study_config = json.load(f)
self.study = study_config['study']
self.import_interventions()
def init_users(self):
self.user_manager.load_users()
self.users = self.user_manager.get_users()
for u in self.users.values():
self.init_user(u)
def init_user(self, user):
user_params = {}
iv_names = set()
# add all of the default interventions (ones that apply to all users)
for iv in self.study['all_conditions']['interventions']:
iv_name = iv['name'] # grab the intevention name
all_cond_iv_params = iv['user_params'] # grab the intervention-specific parameters
iv_names.add(iv_name) # add to the set, avoid dupes later
user_params[iv_name] = {} # create a location in user_params for iv_params
# construct each param with default value and add it to user_params
for param in all_cond_iv_params:
pname = param['name']
pvalue = param['default_value']
user_params[iv_name][pname] = pvalue
# go through all of this user's conditions and reger to the
# study's conditions to figure out what parameters to add/set/override
for ucond in user.conditions: # this user's conditions
for study_cond in self.study['conditions']: # the study's conditions
scond = study_cond['name']
# figure out which condition(s) this user is in
if ucond == scond:
for iv in study_cond['interventions']: # these will all apply to the user
iv_name = iv["name"]
iv_names.add(iv_name) # add to the set, will avoid dupes
cond_iv_params = iv['user_params'] # grab the intervention-specific parameters
if not iv_name in user_params: # if needed, create a loction in user_params for iv_params
user_params[iv_name] = {}
for param in cond_iv_params:
pname = param["name"]
pvalue = param["default_value"]
user_params[iv_name][pname] = pvalue
user.params = user_params
# initialize this user's interventions, save the intervention objects with the user
user.interventions = []
for iv in iv_names:
iv_obj = self.interventions[iv].init_intervention(self.event_manager, user, self.clock.get_time())
user.interventions.append(iv_obj)
def import_interventions(self):
cwd = os.getcwd()
cwd = cwd + '/interventions'
iv_names = []
for name in os.listdir(cwd):
if name.endswith(".py") and not name.startswith("__"):
name = name[:-3]
iv_names.append(name)
for name in iv_names:
module_name = 'interventions.' + name
self.interventions[name] = importlib.import_module(module_name)
if __name__ == "__main__":
sm = StudyManager()
sm.start()