+ # properties
+
+ @property
+ def selected_day(self):
+ if not self.selected_date in self.days.keys():
+ self.days[self.selected_date] = self.add_day(date=self.selected_date)
+ return self.days[self.selected_date]
+
+ # table manipulations
+
+ def add_day(self, date, dict_source=None):
+ day = Day.from_dict(self, dict_source, date) if dict_source else Day(self, date)
+ self.days[date] = day
+ return day
+
+ def add_task(self, id_=None, dict_source=None):
+ id_ = id_ if id_ else str(uuid4())
+ t = Task.from_dict(self, dict_source, id_) if dict_source else Task(self, id_)
+ self.tasks[id_] = t
+ return t
+
+ def fork_task(self, id_):
+ origin = self.tasks[id_]
+ now = today_date(with_time=True)
+ fork_id = str(uuid4())
+ fork = Task(self, fork_id, {}, {}, {}, {}, origin.comment, id_)
+ self.tasks[fork_id] = fork
+ return fork
+
+ def update_task(self, id_, title, default_effort, tags, comment, dep_ids, depender_ids):
+ task = self.tasks[id_] if id_ in self.tasks.keys() else self.add_task(id_)
+ task.title.set(title)
+ task.default_effort.set(default_effort)
+ task.tags.set(tags)
+ task.dep_ids.set(dep_ids)
+ task.comment = comment
+ for depender in task.dependers:
+ if not depender.id_ in depender_ids:
+ depender_dep_ids = depender.dep_ids.now
+ depender_dep_ids.remove(task.id_)
+ depender.dep_ids.set(depender_dep_ids)
+ for depender_id in depender_ids:
+ depender= self.tasks[depender_id]
+ depender_dep_ids = depender.dep_ids.now
+ depender_dep_ids.add(task.id_)
+ depender.dep_ids.set(depender_dep_ids)
+ return task
+
+ def add_todo(self, todo_dict=None, id_=None, task=None, efforts=None, parenthood=''):
+ make_children = parenthood != 'childless'
+ adopt_if_possible = parenthood == 'adoption'
+ id_ = id_ if id_ else str(uuid4())
+ if todo_dict:
+ todo = Todo.from_dict(self, todo_dict, id_)
+ elif task and efforts:
+ todo = Todo(self, id_, task, efforts=efforts)
+ deps = []
+ if make_children and todo.latest_date in self.days.keys():
+ for dep_task in task.deps:
+ # if Todo expects dependencies, adopt any compatibles found in DB.selected_date
+ # before creating new depended Todos
+ dep_todo = None
+ if adopt_if_possible:
+ adoptable_todos = [t for t in self.days[todo.latest_date].linked_todos_as_list
+ if t.task.id_ == dep_task.id_]
+ if len(adoptable_todos) > 0:
+ dep_todo = adoptable_todos[0]
+ if not dep_todo:
+ dep_todo = self.add_todo(task=dep_task, efforts=efforts, parenthood=parenthood)
+ deps += [dep_todo]
+ todo.deps = deps
+ self.todos[id_] = todo
+ for date in todo.efforts.keys():
+ if date in self.days.keys(): # possibly not all Days have been initialized yet
+ self.days[date].linked_todos_as_list += [todo]
+ return todo
+
+ def _update_todo_shared(self, id_, done, comment, importance):
+ todo = self.todos[id_]
+ todo.done = done
+ todo.comment = comment
+ todo.importance = importance
+ return todo
+
+ def update_todo_for_day(self, id_, date, effort, done, comment, importance):
+ todo = self._update_todo_shared(id_, done, comment, importance)
+ todo.efforts[date] = effort
+
+ def update_todo(self, id_, efforts, done, comment, tags, importance, deps):
+ todo = self._update_todo_shared(id_, done, comment, importance)
+ if len(efforts) == 0 and not todo.deps:
+ raise PlomException('todo must have at least one effort!')
+ todo.deps = deps
+ todo.efforts = efforts
+ for date in todo.efforts.keys():
+ if not date in self.days.keys():
+ self.add_day(date=date)
+ if not self in self.days[date].linked_todos_as_list:
+ self.days[date].linked_todos_as_list += [todo]
+ todo.day_tags = tags
+
+ def delete_todo(self, id_):
+ todo = self.todos[id_]
+ dates_to_delete = []
+ for date in todo.efforts.keys():
+ dates_to_delete += [date]
+ for date in dates_to_delete:
+ self.delete_effort(todo, date, force=True)
+ for depender in todo.dependers:
+ depender._dep_ids.remove(todo.id_)
+ del self.todos[id_]
+
+ def delete_task(self, id_):
+ del self.tasks[id_]
+
+ def delete_effort(self, todo, date, force=False):
+ if (not force) and len(todo.efforts) == 1:
+ raise PlomException('todo must retain at least one effort!')
+ self.days[date].linked_todos_as_list.remove(todo)
+ del todo.efforts[date]
+
+ # views
+
+ def show_message(self, message):
+ return j2env.get_template('message.html').render(message=message)
+
+ def show_calendar_export(self, start_date_str, end_date_str):
+ days_to_show = self.init_calendar_items(start_date_str, end_date_str)
+ return j2env.get_template('calendar_export.html').render(days=days_to_show)
+
+ def show_calendar(self, start_date_str, end_date_str):
+ # self.tag_filter_and = ['calendar']
+ # self.tag_filter_not = ['deleted']
+
+ # todays_date_obj = datetime.strptime(today_date(), DATE_FORMAT)
+ # yesterdays_date_obj = todays_date_obj - timedelta(1)
+ # def get_day_limit_obj(index, day_limit_string):
+ # date_obj = datetime.strptime(sorted(self.days.keys())[index], DATE_FORMAT)
+ # if day_limit_string and len(day_limit_string) > 0:
+ # if day_limit_string in {'today', 'yesterday'}:
+ # date_obj = todays_date_obj if day_limit_string == 'today' else yesterdays_date_obj
+ # else:
+ # date_obj = datetime.strptime(day_limit_string, DATE_FORMAT)
+ # return date_obj
+ # start_date_obj = get_day_limit_obj(0, start_date_str)
+ # end_date_obj = get_day_limit_obj(-1, end_date_str)
+
+ # days_to_show = {}
+ # for n in range(int((end_date_obj - start_date_obj).days) + 1):
+ # date_obj = start_date_obj + timedelta(n)
+ # date_str = date_obj.strftime(DATE_FORMAT)
+ # if date_str not in self.days.keys():
+ # days_to_show[date_str] = self.add_day(date_str)
+ # else:
+ # days_to_show[date_str] = self.days[date_str]
+ # days_to_show[date_str].month_title = date_obj.strftime('%B') if date_obj.day == 1 else None
+ # days_to_show[date_str].weekday = datetime.strptime(date_str, DATE_FORMAT).strftime('%A')[:2]
+
+ days_to_show = self.init_calendar_items(start_date_str, end_date_str)
+ return j2env.get_template('calendar.html').render(
+ selected_date=self.selected_date,
+ days=days_to_show,
+ start_date=start_date_str,
+ end_date=end_date_str)
+
+ def show_day_todos(self, undone_sort_order=None, done_sort_order=None, is_tree_shaped=False, todo_parenthood=None):
+ legal_undone_sort_keys = {'title', 'sort_done', 'default_effort', 'importance'}
+ legal_done_sort_keys = {'title', 'effort_at_selected_date', 'family_effort'}
+
+ current_date = datetime.strptime(self.selected_date, DATE_FORMAT)
+ prev_date = current_date - timedelta(days=1)
+ prev_date_str = prev_date.strftime(DATE_FORMAT)
+ next_date = current_date + timedelta(days=1)
+ next_date_str = next_date.strftime(DATE_FORMAT)
+
+ adoptable_past_todos = []
+ for todo in [t for t in self.todos.values()
+ if t.visible
+ and (not t.done)
+ and t.earliest_date < self.selected_date]:
+ adoptable_past_todos += [todo]
+ undone_todos = self.selected_day.sorted_todos(False, is_tree_shaped, undone_sort_order,
+ legal_undone_sort_keys)
+ done_todos = self.selected_day.sorted_todos(True, is_tree_shaped, done_sort_order,
+ legal_done_sort_keys)
+
+ return j2env.get_template('day_todos.html').render(
+ day=self.selected_day,
+ tags=self.t_tags,
+ filter_and=self.tag_filter_and,
+ filter_not=self.tag_filter_not,
+ prev_date=prev_date_str,
+ adoptable_past_todos=adoptable_past_todos,
+ next_date=next_date_str,
+ all_tasks=[t for t in self.tasks.values()],
+ undone_todos=undone_todos,
+ done_todos=done_todos,
+ is_tree_shaped=is_tree_shaped,
+ undone_sort=undone_sort_order,
+ done_sort=done_sort_order,
+ parenthood=todo_parenthood)
+
+ def show_todo(self, id_, parenthood):
+ todo = self.todos[id_]
+ filtered_tasks = [t for t in self.tasks.values()
+ if t != todo.task]
+ filtered_todos = [t for t in self.todos.values()
+ if t != todo
+ and t not in todo.deps]
+
+ legal_dates = list(todo.efforts.keys())
+ date_filtered_todos = []
+ for date in legal_dates:
+ for filtered_todo in filtered_todos:
+ if filtered_todo in date_filtered_todos:
+ continue
+ if date in filtered_todo.efforts.keys():
+ date_filtered_todos += [filtered_todo]
+
+ dep_slots = []
+ for dep in todo.task.deps:
+ dep_slots += [{'task': dep,
+ 'todos': [t for t in todo.deps if t.task == dep]}]
+
+ suggested_todos = {}
+ for dep in todo.task.deps:
+ suggested_todos[dep.id_] = [t for t in date_filtered_todos if t.task.id_ == dep.id_]
+ additional_deps = [t for t in todo.deps if not t.task in todo.task.deps]
+
+ return j2env.get_template('todo.html').render(
+ tags=self.t_tags,
+ todo=todo,
+ filtered_todos=date_filtered_todos,
+ filtered_tasks=filtered_tasks,
+ dep_slots=dep_slots,
+ suggested_todos=suggested_todos,
+ additional_deps=additional_deps,
+ parentood=parenthood,
+ dep_todos=todo.deps)
+
+ def show_task(self, id_):
+ if id_:
+ if not id_ in self.tasks.keys():
+ raise PlomException('no Task for ID')
+ task = self.tasks[id_]