1 files changed, 133 insertions, 0 deletions
diff --git a/src/notifier/manager.py b/src/notifier/manager.py
new file mode 100644
index 0000000..240cf85
--- /dev/null
+++ b/src/notifier/manager.py
@@ -0,0 +1,133 @@
+##############################################################################
+# 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 (" + str(booking.id) + ")", "You have been added to a booking (" + str(booking.id) + ")"]
+        cls.booking_notify(booking, template, titles)
+
+    @classmethod
+    def notify_booking_end(cls, booking):
+        template = "notifier/end_booking.html"
+        titles = ["Your booking (" + str(booking.id) + ") has ended", "A booking (" + str(booking.id) + ") that 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.userprofile)
+        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.userprofile)
+
+    @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)