def handle_cycle_created(obj, manually):
today = datetime.date.today()
if manually:
create_notification = "manual_cycle_created"
else:
create_notification = "cycle_created"
pusher.update_or_create_notifications(obj, today, create_notification)
if not obj.is_current:
return
notification_names = [
get_notif_name_by_wf(obj.workflow), create_notification,
"cycle_task_overdue", "cycle_task_due_today"
]
tasks = list(obj.cycle_task_group_object_tasks)
query = pusher.get_notification_query(
*tasks, **{"notification_names": notification_names})
exists_notifications = collections.defaultdict(set)
for notification in query:
exists_notifications[notification.notification_type_id].add(
notification.object_id)
for notification_type in pusher.get_notification_types(
*notification_names):
object_ids = exists_notifications[notification_type.id]
notify_tasks = [
t for t in tasks if not (t.id in object_ids or t.is_done)
]
if create_notification == notification_type.name:
pusher.create_notifications_for_objects(notification_type, today,
*notify_tasks)
else:
for task in notify_tasks:
pusher.create_notifications_for_objects(
notification_type, task.end_date, task)
def handle_cycle_created(obj, manually):
today = datetime.date.today()
if manually:
create_notification = "manual_cycle_created"
else:
create_notification = "cycle_created"
pusher.update_or_create_notifications(obj, today, create_notification)
if not obj.is_current:
return
notification_names = [get_notif_name_by_wf(obj.workflow),
create_notification,
"cycle_task_overdue",
"cycle_task_due_today"]
tasks = list(obj.cycle_task_group_object_tasks)
query = pusher.get_notification_query(
*tasks,
**{"notification_names": notification_names})
exists_notifications = collections.defaultdict(set)
for notification in query:
exists_notifications[notification.notification_type_id].add(
notification.object_id)
for notification_type in pusher.get_notification_types(*notification_names):
object_ids = exists_notifications[notification_type.id]
notify_tasks = [t for t in tasks if not (t.id in object_ids or t.is_done)]
if create_notification == notification_type.name:
pusher.create_notifications_for_objects(notification_type,
today,
*notify_tasks)
else:
for task in notify_tasks:
pusher.create_notifications_for_objects(notification_type,
task.end_date,
task)
def handle_cycle_task_created(ctask):
"""Create notifications on CycleTask POST operation.
Args:
ctask: CycleTask instance
"""
notification_types = [get_notif_name_by_wf(ctask.workflow),
"cycle_task_due_today",
"cycle_task_overdue"]
notification_type_query = pusher.get_notification_types(*notification_types)
notification_types_dict = {t.name: t for t in notification_type_query}
for n_type in notification_types:
pusher.create_notifications_for_objects(notification_types_dict[n_type],
ctask.end_date, *[ctask])
def handle_cycle_task_created(ctask):
"""Create notifications on CycleTask POST operation.
Args:
ctask: CycleTask instance
"""
notification_types = [get_notif_name_by_wf(ctask.workflow),
"cycle_task_due_today",
"cycle_task_overdue"]
notification_type_query = pusher.get_notification_types(*notification_types)
notification_types_dict = {t.name: t for t in notification_type_query}
for n_type in notification_types:
pusher.create_notifications_for_objects(notification_types_dict[n_type],
ctask.end_date, *[ctask])
def not_done_tasks_notify(tasks):
"""Notification handling for tasks that moved to active state.
It will
- create notification for tasks if it's needed
- delete all notifications for related cycles
"""
if not tasks:
return
cycles = set()
notification_send_at_tasks_dict = collections.defaultdict(list)
notification_types = set()
for task in tasks:
cycles.add(task.cycle)
if task.status == models.CycleTaskGroupObjectTask.DECLINED:
notification_send_at_tasks_dict[
("cycle_task_declined", datetime.date.today())
].append(task)
notification_types.add("cycle_task_declined")
for n_type in [get_notif_name_by_wf(task.cycle.workflow),
"cycle_task_due_today",
"cycle_task_overdue"]:
notification_send_at_tasks_dict[
(n_type, task.end_date)
].append(task)
notification_types.add(n_type)
# delete all notifications for cycles about all tasks compleated and
# all tasks notifications that required to be updated
pusher.get_notification_query(
*(list(cycles) + tasks),
**{"notification_names": [
"all_cycle_tasks_completed"
] + list(notification_types)}
).delete(
synchronize_session="fetch"
)
notification_type_query = pusher.get_notification_types(*notification_types)
notification_types_dict = {t.name: t for t in notification_type_query}
for pair, pair_tasks in notification_send_at_tasks_dict.iteritems():
notification_type_name, send_at = pair
notification_type = notification_types_dict[notification_type_name]
pusher.create_notifications_for_objects(notification_type,
send_at,
*pair_tasks)
