diff --git a/vendor/registrasion/registrasion/contrib/badger.py b/vendor/registrasion/registrasion/contrib/badger.py index 3049ccd2246efb3fd7ce5711bd05dc573c00930f..b00f01eaaf4698a7c0bef9d1e066b08aa412d354 100644 --- a/vendor/registrasion/registrasion/contrib/badger.py +++ b/vendor/registrasion/registrasion/contrib/badger.py @@ -25,6 +25,7 @@ import pdb from django.core.management.base import BaseCommand from django.contrib.auth.models import User, Group +from django.db.utils import OperationalError from pinaxcon.registrasion.models import AttendeeProfile from registrasion.controllers.cart import CartController from registrasion.controllers.invoice import InvoiceController @@ -124,8 +125,13 @@ def set_colour(soup, slice_id, colour): style = elem.get('style') elem.set('style', style.replace('fill:#316a9a', 'fill:#%s' % colour)) -Volunteers = Group.objects.filter(name='Conference volunteers').first().user_set.all() -Organisers = Group.objects.filter(name='Conference organisers').first().user_set.all() +## It's possible that this script will be run before the database has been populated +try: + Volunteers = Group.objects.filter(name='Conference volunteers').first().user_set.all() + Organisers = Group.objects.filter(name='Conference organisers').first().user_set.all() +except (OperationalError, AttributeError): + Volunteers = [] + Organisers = [] def is_volunteer(attendee): '''