From 97bf033d639600f2a69573c45ffd891535fe3f5d Mon Sep 17 00:00:00 2001
From: Madhusudan.C.S
Date: Thu, 9 Dec 2010 19:11:15 +0530
Subject: Create payment and wifi objects if they don't exist.

---
 project/scipycon/registration/views.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/project/scipycon/registration/views.py b/project/scipycon/registration/views.py
index 75925d9..505f14a 100644
--- a/project/scipycon/registration/views.py
+++ b/project/scipycon/registration/views.py
@@ -378,12 +378,13 @@ def regstats_download(request, scope):
         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()
-- 
cgit