class Scheduler(object):
def __init__(self, hours,
taskdb=TASK_DB, wantdb=WANT_DB, breakdb=BREAK_DB):
self.hours = hours
self.generator = TimeSlotGenerator(hours)
self.task_adapter = TaskJsonAdapter(file_name = taskdb)
self.want_adapter = WantJsonAdapter(file_name = wantdb)
self.break_adapter = BreakJsonAdapter(file_name = breakdb)
self.scorer = Scorer(db= None, tasks= self.task_adapter.items)
self.i_index = 0
self.g_index = 0
self.n_index = 0
self.i_items = []
self.g_items = []
self.n_items = []
# fill the items in
self._fill_items()
self.schedule = self.make_schedule()
def _fill_items(self):
for task in self.task_adapter.items:
placement = self.scorer.get_placement(task)
if placement == 'I':
self.i_items.append(task)
elif placement == 'G':
self.g_items.append(task)
elif placement == 'R':
self.n_items.append(task)
elif placement == 'N':
self.n_items.append(task)
else:
pass
def _reset_index(self):
self.i_index = 0
self.g_index = 0
self.n_index = 0
return
def make_schedule(self, repeat_items=True):
schedule = []
time_slots = self.generator.time_slots
chosen_item_list = self.i_items
chosen_index = self.i_index
number = 1
for slot in time_slots:
#breaks get 15 min slots
if slot == 15:
print("A break")
number_items = len(self.break_adapter.items)
random_int = random.randint(0, number_items-1)
schedule.append({"number" : number, "timeslot" : slot, "item" : self.break_adapter.get_ith_json(random_int)})
#a want or task
else:
random_int = random.randint(1,2)
# a want
if random_int == 1:
print("A want")
number_items = len(self.want_adapter.items)
random_int = random.randint(0, number_items-1)
schedule.append({"number" : number, "timeslot" : slot, "item" : self.want_adapter.get_ith_json(random_int)})
# a task
else:
print("A task")
if self.i_index < len(self.i_items):
print("Item in I bucket")
chosen_item_list = self.i_items
chosen_index = self.i_index
self.i_index = self.i_index + 1
else:
if self.g_index < len(self.g_items):
print("Item in G bucket")
chosen_item_list = self.g_items
chosen_index = self.g_index
self.g_index = self.g_index + 1
else:
if self.n_index < len(self.n_items):
print("Item in N bucket")
chosen_item_list = self.n_items
chosen_index = self.n_index
self.n_index = self.n_index + 1
else:
# we will need to reset or quit
if repeat_items:
print("Resetting buckets")
self._reset_index()
#starts off with the first thing of the most recent list
chosen_index = 0
else:
#return schedule
pass
task = chosen_item_list[chosen_index]
placement = self.scorer.get_placement(task)
task_struct = {
"class" : "Task",
"type" : str(type(task)),
"description" : task.name,
"due_date" : str(task.due_date),
"score" : task.get_score(),
"placement" : placement
}
schedule.append({"number" : number, "timeslot" : slot, "item" : task_struct})
number = number + 1
return schedule
def print_schedule(self):
in_minutes = self.hours*60
total = 0
for item in self.schedule:
total = total + item["timeslot"]
time_remaining = in_minutes - total
print("|--------------------------------->")
print("|Activity length : %s minutes\n|Activity : %s\n|Time Spent : %s minutes\n|Time Remaining : %s minutes" % (
item["timeslot"], item["item"], total, time_remaining))
print("|--------------------------------->")
