Create payment and wifi objects if they don't exist.
authorMadhusudan.C.S <madhusudancs@gmail.com>
Thu, 09 Dec 2010 19:11:15 +0530
changeset 343 debfd9506f40
parent 342 21dc37050a37
child 344 47db5bd2a791
Create payment and wifi objects if they don't exist.
project/scipycon/registration/views.py
--- a/project/scipycon/registration/views.py	Thu Dec 09 19:08:12 2010 +0530
+++ b/project/scipycon/registration/views.py	Thu Dec 09 19:11:15 2010 +0530
@@ -378,12 +378,13 @@
         row = []
         row.append(reg.registrant.get_full_name())
         row.append(reg.city)
-        row.append('Yes' if reg.registrant.payment_set.get().confirmed
-           else 'No')
+        payment, create = reg.registrant.payment_set.get_or_create()
+        row.append('Yes' if payment.confirmed else 'No')
         row.append('Yes' if reg.conference else 'No')
         row.append('Yes' if reg.tutorial else 'No')
         row.append('Yes' if reg.sprint else 'No')
-        row.append(reg.registrant.wifi_set.get().registration_id)
+        wifi, create = reg.registrant.wifi_set.get_or_create()
+        row.append(wifi.registration_id)
         row.append('Yes' if reg.registrant.payment_set.get().acco_confirmed
            else 'No')
         acco, created = reg.registrant.accommodation_set.get_or_create()