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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
##############################################################################
# Copyright (c) 2018 Parker Berberian, Sawyer Bergeron and others.
#
# All rights reserved. This program and the accompanying materials
# are made available under the terms of the Apache License, Version 2.0
# which accompanies this distribution, and is available at
# http://www.apache.org/licenses/LICENSE-2.0
##############################################################################
import os
from notifier.models import Notification
from django.core.mail import send_mail
from django.template.loader import render_to_string
class NotificationHandler(object):
@classmethod
def notify_new_booking(cls, booking):
template = "notifier/new_booking.html"
titles = ["You have a new Booking", "You have been added to a Booking"]
cls.booking_notify(booking, template, titles)
@classmethod
def notify_booking_end(cls, booking):
template = "notifier/end_booking.html"
titles = ["Your booking has ended", "A booking you collaborate on has ended"]
cls.booking_notify(booking, template, titles)
@classmethod
def booking_notify(cls, booking, template, titles):
"""
Creates a notification for a booking owner and collaborators
using the template.
titles is a list - the first is the title for the owner's notification,
the last is the title for the collaborators'
"""
owner_notif = Notification.objects.create(
title=titles[0],
content=render_to_string(template, context={
"booking": booking,
"owner": True
})
)
owner_notif.recipients.add(booking.owner)
if not booking.collaborators.all().exists():
return # no collaborators - were done
collab_notif = Notification.objects.create(
title=titles[-1],
content=render_to_string(template, context={
"booking": booking,
"owner": False
})
)
for c in booking.collaborators.all():
collab_notif.recipients.add(c)
@classmethod
def email_job_fulfilled(cls, job):
template_name = "notifier/email_fulfilled.txt"
all_tasks = job.get_tasklist()
users = list(job.booking.collaborators.all())
users.append(job.booking.owner)
for user in users:
user_tasklist = []
# gather up all the relevant messages from the lab
for task in all_tasks:
if (not hasattr(task, "user")) or task.user == user:
user_tasklist.append({
"title": task.type_str + " Message: ",
"content": task.message
})
# gather up all the other needed info
context = {
"user_name": user.userprofile.full_name,
"messages": user_tasklist,
"booking_url": os.environ.get("DASHBOARD_URL", "<Dashboard url>") + "/booking/detail/" + str(job.booking.id) + "/"
}
# render email template
message = render_to_string(template_name, context)
# finally, send the email
send_mail(
"Your Booking is Ready",
message,
os.environ.get("DEFAULT_FROM_EMAIL", "opnfv@pharos-dashboard"),
user.userprofile.email_addr,
fail_silently=False
)
@classmethod
def email_booking_over(cls, booking):
template_name = "notifier/email_ended.txt"
hostnames = [host.template.resource.name for host in booking.resource.hosts.all()]
users = list(booking.collaborators.all())
users.append(booking.owner)
for user in users:
context = {
"user_name": user.userprofile.full_name,
"booking": booking,
"hosts": hostnames,
"booking_url": os.environ.get("DASHBOARD_URL", "<Dashboard url>") + "/booking/detail/" + str(booking.id) + "/"
}
message = render_to_string(template_name, context)
send_mail(
"Your Booking has Expired",
message,
os.environ.get("DEFAULT_FROM_EMAIL", "opnfv@pharos-dashboard"),
user.userprofile.email_addr,
fail_silently=False
)
@classmethod
def task_updated(cls, task):
"""
called every time a lab updated info about a task.
currently only checks if the job is now done so I can send an email,
may add more functionality later
"""
if task.job.is_fulfilled():
cls.email_job_fulfilled(task.job)
|
