1 files changed, 23 insertions, 10 deletions
diff --git a/src/notifier/models.py b/src/notifier/models.py
index 9ebc6fc..ed0edeb 100644
--- a/src/notifier/models.py
+++ b/src/notifier/models.py
@@ -1,5 +1,5 @@
 ##############################################################################
-# Copyright (c) 2016 Max Breitenfeldt and others.
+# Copyright (c) 2018 Sawyer Bergeron, Parker Berberian, and others.
 #
 # All rights reserved. This program and the accompanying materials
 # are made available under the terms of the Apache License, Version 2.0
@@ -8,14 +8,24 @@
 ##############################################################################
 
 from django.db import models
-from jira import JIRA, JIRAError
-from dashboard.models import Resource
 from booking.models import Booking
-from django.contrib.auth.models import User
 from account.models import UserProfile
-from django.contrib import messages
-from django.db.models.signals import pre_save
 from fernet_fields import EncryptedTextField
+from account.models import Lab
+
+
+class MetaBooking(models.Model):
+    id = models.AutoField(primary_key=True)
+    booking = models.OneToOneField(Booking, on_delete=models.CASCADE, related_name="metabooking")
+    ending_notified = models.BooleanField(default=False)
+    ended_notified = models.BooleanField(default=False)
+    created_notified = models.BooleanField(default=False)
+
+
+class LabMessage(models.Model):
+    lab = models.ForeignKey(Lab, on_delete=models.CASCADE)
+    msg = models.TextField()  # django template should be put here
+
 
 class Notifier(models.Model):
     id = models.AutoField(primary_key=True)
@@ -24,15 +34,18 @@ class Notifier(models.Model):
     user = models.ForeignKey(UserProfile, on_delete=models.CASCADE, null=True, blank=True)
     sender = models.CharField(max_length=240, default='unknown')
     message_type = models.CharField(max_length=240, default='email', choices=(
-        ('email','Email'), 
+        ('email', 'Email'),
         ('webnotification', 'Web Notification')))
     msg_sent = ''
 
-    import notifier.dispatchers
-
     def __str__(self):
         return self.title
 
+    """
+    Implement for next PR: send Notifier by media agreed to by user
+    """
+    def send(self):
+        pass
+
     def getEmail(self):
         return self.user.email_addr
-