Merge pull request #131 from CTPUG/bugs/issue-128-slot-deletion-handling

Bugs/issue 128 slot deletion handling

wafer/schedule/models.py

@@ -4,7 +4,7 @@
 from django.core.urlresolvers import reverse
 from django.core.exceptions import ValidationError
 from django.db import models
-from django.db.models.signals import post_save
+from django.db.models.signals import post_save, post_delete
 from django.utils.encoding import python_2_unicode_compatible
 
 from wafer.snippets.markdown_field import MarkdownTextField
@@ -170,9 +170,13 @@ def get_details(self):
         return self.get_desc()
 
     def get_start_time(self):
-        start_slot = list(self.slots.all())[0]
-        start = start_slot.get_start_time().strftime('%H:%M')
-        return u'%s, %s' % (start_slot.get_day(), start)
+        slots = list(self.slots.all())
+        if slots:
+            start = slots[0].get_start_time().strftime('%H:%M')
+            day = slots[0].get_day()
+            return u'%s, %s' % (day, start)
+        else:
+            return 'WARNING: No Time and Day Specified'
 
     def __str__(self):
         return u'%s in %s at %s' % (self.get_desc(), self.venue,
@@ -207,3 +211,8 @@ def invalidate_check_schedule(*args, **kw):
 post_save.connect(invalidate_check_schedule, sender=Venue)
 post_save.connect(invalidate_check_schedule, sender=Slot)
 post_save.connect(invalidate_check_schedule, sender=ScheduleItem)
+
+post_delete.connect(invalidate_check_schedule, sender=Day)
+post_delete.connect(invalidate_check_schedule, sender=Venue)
+post_delete.connect(invalidate_check_schedule, sender=Slot)
+post_delete.connect(invalidate_check_schedule, sender=ScheduleItem)