def renew_membership(request, club_name):
"""new club memberships and renewals"""
user = request.user
up = user.get_profile()
club = Club.objects.get(safe_name=club_name)
try:
m = Membership.objects.get(user_prof=up, club=club, paid=True)
m.price = club.renew_cost
except Membership.DoesNotExist:
#create new membership
m = Membership()
m.club = club
m.user_prof = up
m.price = club.new_member_cost
m.start = datetime.date.today()
m.valid_thru = m.start + datetime.timedelta(days=365)
#create an order
order = Order()
order.user_prof = up
order.save()
m.order = order
m.save()
redirect_target = request.build_absolute_uri(
reverse('club_detail', kwargs={'club_name': club_name}))
# What you want the button to do.
paypal_dict = {
"business": club.paypal_email,
#'business': '*****@*****.**',
"amount": order.calc_total_price(),
"item_name": 'Membership for %s ' % club.name,
"invoice": order.pk,
"notify_url": request.build_absolute_uri(reverse('paypal-ipn')),
"return_url": redirect_target,
"cancel_return": redirect_target,
}
paypal_form = PayPalPaymentsForm(initial=paypal_dict)
context = {
'paypal_form': paypal_form.render(),
'price': paypal_dict['amount'],
'club': club,
'order': order,
'items': [i.as_leaf_model().cart_name() for i in order.items.all()]
}
return render_to_response('registration/start_pay.html',
context,
context_instance=RequestContext(request))
def test_auto_increment_number(self):
m= Membership()
m.user_prof = self.user2.get_profile()
m.club = self.c
m.start = datetime.date.today() - datetime.timedelta(days=300)
m.valid_thru = datetime.date.today()
m.full_clean()
m.save()
self.assertEqual(m.num,2)
def setUp(self):
self.c = Club()
self.c.name = "test club"
self.c.full_clean()
self.c.save()
self.user1 = User()
self.user1.first_name = "Justin"
self.user1.last_name = "Gray"
self.user1.username = "******"
self.user1.save()
self.user2 = User()
self.user2.first_name = "Sirius"
self.user2.last_name = "Gray"
self.user2.username = "******"
self.user2.save()
m = Membership()
m.user_prof = self.user1.get_profile()
m.club = self.c
m.num = 1
m.start = datetime.date.today() - datetime.timedelta(days=300)
m.valid_thru = datetime.date.today() + datetime.timedelta(days=1)
m.paid = True
m.full_clean()
m.save()
def new_membership(request, clubname):
"""create a new membership for the given user and club"""
username = request.POST['username']
u = User.objects.get(username=username)
up = u.get_profile()
c = Club.objects.get(safe_name=clubname)
admin = is_club_admin(request.user, c)
if not admin:
return HttpResponseForbidden()
try:
m = Membership.objects.get(club=c, user_prof=up)
except Membership.DoesNotExist:
m = Membership()
m.club = c
m.user_prof = up
m.start = datetime.date.today()
m.valid_thru = m.start + datetime.timedelta(days=365.2425)
m.save()
mr = MembershipResource()
mr_bundle = mr.build_bundle(obj=m, request=request)
return HttpResponse(mr.serialize(None, mr.full_dehydrate(mr_bundle), 'application/json'), mimetype='application/json')
def setUp(self):
self.c = Club()
self.c.name = "test club"
self.c.full_clean()
self.c.save()
self.user1 = User()
self.user1.first_name = "Justin"
self.user1.last_name = "Gray"
self.user1.username = "******"
self.user1.save()
self.user2 = User()
self.user2.first_name = "Sirius"
self.user2.last_name = "Gray"
self.user2.username = "******"
self.user2.save()
m= Membership()
m.user_prof = self.user1.get_profile()
m.club = self.c
m.num = 1
m.start = datetime.date.today() - datetime.timedelta(days=300)
m.valid_thru = datetime.date.today()+datetime.timedelta(days=1)
m.paid = True
m.full_clean()
m.save()
def test_total_price(self):
self.o = Order()
self.o.user_prof = self.user.get_profile()
self.o.save()
item1 = Registration()
item1.number = 11
item1.race_class = self.race_class
item1.pax_class = None
item1.event = self.e
item1.price = "40.00"
item1.order = self.o
item1.club = self.c
item1.save()
item2 = Membership()
item2.user_prof = self.user.get_profile()
item2.club = self.c
item2.num = 1
item2.start = datetime.date.today() - datetime.timedelta(days=300)
item2.valid_thru = datetime.date.today()+datetime.timedelta(days=1)
item2.price = "60.00"
item2.order = self.o
item2.save()
self.o.calc_total_price()
self.assertEqual(self.o.total_price, "100.00")
club = Club()
club.name ="test-test"
club.save()
c = Coupon()
c.discount_amount = "10.00"
c.permanent = True
c.code = "aaa"
c.expires = datetime.date.today() + datetime.timedelta(days=1)
c.club = club
c.save()
self.o.coupon = c
self.o.calc_total_price()
self.assertEqual(self.o.total_price, '90.00')
def test_auto_increment_number(self):
m = Membership()
m.user_prof = self.user2.get_profile()
m.club = self.c
m.start = datetime.date.today() - datetime.timedelta(days=300)
m.valid_thru = datetime.date.today()
m.full_clean()
m.save()
self.assertEqual(m.num, 2)
def test_is_active_member(self):
self.assertTrue(self.c.is_active_member(self.user1))
self.assertFalse(self.c.is_active_member(self.user2))
m = Membership()
m.user_prof = self.user2.get_profile()
m.club = self.c
m.num = 2
m.start = datetime.date.today() - datetime.timedelta(days=300)
m.valid_thru = datetime.date.today()
m.full_clean()
m.save()
#make sure expired memberships don't report as active
self.assertFalse(self.c.is_active_member(self.user2))
def test_duplicate_id(self):
m = Membership()
m.user_prof = self.user2.get_profile()
m.club = self.c
m.num = 1
m.start = datetime.date.today() - datetime.timedelta(days=300)
m.valid_thru = datetime.date.today() + datetime.timedelta(days=1)
try:
m.full_clean()
m.save()
except ValidationError as err:
self.assertEqual(
str(err),
"{'__all__': [u'A member with that number already exists']}")
else:
self.fail("ValidationError expected")
def test_duplicate_id(self):
m= Membership()
m.user_prof = self.user2.get_profile()
m.club = self.c
m.num = 1
m.start = datetime.date.today() - datetime.timedelta(days=300)
m.valid_thru = datetime.date.today()+datetime.timedelta(days=1)
try:
m.full_clean()
m.save()
except ValidationError as err:
self.assertEqual(str(err),"{'__all__': [u'A member with that number already exists']}")
else:
self.fail("ValidationError expected")
def test_is_active_member(self):
self.assertTrue(self.c.is_active_member(self.user1))
self.assertFalse(self.c.is_active_member(self.user2))
m= Membership()
m.user_prof = self.user2.get_profile()
m.club = self.c
m.num = 2
m.start = datetime.date.today() - datetime.timedelta(days=300)
m.valid_thru = datetime.date.today()
m.full_clean()
m.save()
#make sure expired memberships don't report as active
self.assertFalse(self.c.is_active_member(self.user2))
def test_total_price(self):
self.o = Order()
self.o.user_prof = self.user.get_profile()
self.o.save()
item1 = Registration()
item1.number = 11
item1.race_class = self.race_class
item1.pax_class = None
item1.event = self.e
item1.price = "40.00"
item1.order = self.o
item1.club = self.c
item1.save()
item2 = Membership()
item2.user_prof = self.user.get_profile()
item2.club = self.c
item2.num = 1
item2.start = datetime.date.today() - datetime.timedelta(days=300)
item2.valid_thru = datetime.date.today() + datetime.timedelta(days=1)
item2.price = "60.00"
item2.order = self.o
item2.save()
self.o.calc_total_price()
self.assertEqual(self.o.total_price, "100.00")
club = Club()
club.name = "test-test"
club.save()
c = Coupon()
c.discount_amount = "10.00"
c.permanent = True
c.code = "aaa"
c.expires = datetime.date.today() + datetime.timedelta(days=1)
c.club = club
c.save()
self.o.coupon = c
self.o.calc_total_price()
self.assertEqual(self.o.total_price, '90.00')
def handle(self, *args, **options):
clear_db()
s = Site.objects.all()[0]
s.domain = "dev.racerslane.com"
s.name = "Racerslane.com"
s.save()
print "Loading users"
reader = csv.DictReader(open('old_data/driver.csv', 'rU'))
for line in reader:
"""user_name","email","verified","activation_code",
"f_name","l_name","address","city","state",
"zip_code","_password"""
for k, v in line.iteritems():
if v == "NULL":
line[k] = None
u = User()
u.username = line['user_name']
u.email = line['email']
u.password = "******"%line['_password']
u.first_name = line['f_name']
u.last_name = line['l_name']
u.is_active = True
u.save()
up = u.get_profile()
up.address = line['address']
up.city = line['city']
up.state = line['state']
up.zip_code = line['zip_code']
up.save()
db.reset_queries()
#dev to make it so I can login to any account
justin = User.objects.get(username="******")
password = justin.password
#User.objects.all().update(password=password)
print "Loading Cars"
car_map = {}
reader = csv.DictReader(open('old_data/car.csv', 'rb'))
for line in reader:
"""id","nickname","make","model","color","avatar_file_loc",
"avatar_thumb_loc","year","owner_user_name"""
for k, v in line.iteritems():
if v == "NULL":
line[k] = None
if line['owner_user_name']:
try:
c = Car()
c.provisional = False
c.name = line['nickname']
c.make = line['make']
c.model = line['model']
if line['color']:
c.color = line['color']
c.year = line['year']
c.user_profile = User.objects.get(username=line['owner_user_name']).get_profile()
s_car_id = (line['owner_user_name'], line['nickname'])
if exists('old_data/avatars/%s_%s_avatar'%s_car_id):
c.avatar.save('%s_%s_avatar'%s_car_id, File(open('old_data/avatars/%s_%s_avatar'%s_car_id)))
c.thumb.save('%s_%s_thumb'%s_car_id, File(open('old_data/avatars/%s_%s_thumb'%s_car_id)))
c.save()
db.reset_queries()
car_map[line['id']] = c.pk
except:
continue
print "Loading Clubs"
#read in clubs
club_map = {}
for line in csv.DictReader(open('old_data/club.csv', 'rU')):
""""name","web_link","process_payments","points_calc_type",
"membership_cost","renewal_cost","membership_terms","paypal_email",
"index_point_method","address","city","state","zip_code","""
for k, v in line.iteritems():
if v == "NULL":
line[k] = None
c = Club()
c.city = line['city']
c.address = line['address']
c.zip_code = line['zip_code']
c.state = line['state']
c.name = line['name']
c.process_payments = line['process_payments']
c.paypal_email = line['paypal_email']
c.membership_terms = line['membership_terms']
c.new_member_cost = line['membership_cost']
c.renew_cost = line['renewal_cost']
c.save()
club_map[line['name']] = c
db.reset_queries()
print "loading coupons"
for line in csv.DictReader(open('old_data/coupon.csv')):
"""coupon_code","club_name","discount_amount","uses_left","expires",
"permanent","driver_user_name","registration_id"""
for k, v in line.iteritems():
if v == "NULL":
line[k] = None
c = Coupon()
c.code = line['coupon_code']
c.permanent = bool(line['permanent'])
c.club = club_map[line['club_name']]
c.uses_left = line['uses_left']
c.discount_amount = line['discount_amount']
if line['expires']:
c.expires = datetime.datetime.strptime(line['expires'], '%Y-%m-%d %H:%M:%S')
if line['driver_user_name']:
c.user_prof = User.objects.get(username=line['driver_user_name']).get_profile()
c.save()
db.reset_queries()
print "loading memberships"
reader = csv.DictReader(open('old_data/membership.csv', 'rb'))
for line in reader:
""""id","club_name","number","start_date","valid_thru_date",
"price","paid","token","payer_id","transaction_id","anon_f_name",
"anon_l_name","driver_user_name"""
for k, v in line.iteritems():
if v == "NULL":
line[k] = None
m = Membership()
m.num = line['number']
m.start = datetime.datetime.strptime(line['start_date'], '%Y-%m-%d %H:%M:%S')
m.valid_thru = datetime.datetime.strptime(line['valid_thru_date'], '%Y-%m-%d %H:%M:%S')
m.paid = bool(line['paid'])
if line['price']!=None:
m.price = float(line['price'])
try:
m.user_prof = User.objects.get(username=line['driver_user_name']).get_profile()
except User.DoesNotExist:
continue
m.club = club_map[line['club_name']]
m._anon_f_name = line['anon_f_name']
m._anon_l_name = line['anon_l_name']
m.save()
db.reset_queries()
print "loading locations"
location_map = {}
for line in csv.DictReader(open("old_data/location.csv")):
"""id","name","address","lat","lng","club_name"""
for k, v in line.iteritems():
if v == "NULL":
line[k] = None
club = Club.objects.get(name=line['club_name'])
l = Location()
l.name = line['name']
l.address = line['address']
#l.lat = line['lat']
#l.lon = line['lng']
l.club = club
l.save()
location_map[line['id']] = l
db.reset_queries()
print "loading seasons"
season_map = {}
for line in csv.DictReader(open('old_data/season.csv')):
s = Season()
""""id","year","active","club_name","drop_lowest_events"""
for k, v in line.iteritems():
if v == "NULL":
line[k] = None
club = Club.objects.get(name=line['club_name'])
s.club = club
s.year = int(line['year'])
s.drop_lowest_events = int(line['drop_lowest_events'])
s.save()
season_map[line['id']] = s
db.reset_queries()
print "loading events"
event_map = {}
for line in csv.DictReader(open('old_data/event.csv', 'rU')):
"""id","name","note","date","registration_close","member_cost",
"non_member_cost","pay_at_event_cost","location_id","club_name",
"season_id","multiplier","count_points"""
for k, v in line.iteritems():
if v == "NULL":
line[k] = None
e = Event()
e.name = line['name']
e.note = line['note']
e.date = datetime.datetime.strptime(line['date'], '%Y-%m-%d %H:%M:%S')
e.reg_close = line['registration_close']+"-05:00"
e.member_price = float(line['member_cost'])
e.non_member_price = float(line['non_member_cost'])
e.non_pre_pay_penalty = float(line['pay_at_event_cost'])
e.season = season_map[line['season_id']]
e.club = e.season.club
e.count_points = int(line['count_points'])
e.multiplier = int(line['multiplier'])
if line['location_id']:
e.location = location_map[line['location_id']]
e.save()
event_map[line['id']] = e
db.reset_queries()
print "loading race_classes"
race_class_map = {}
for line in csv.DictReader(open('old_data/raceclass.csv')):
"""id","pax","name","club_name"""
for k, v in line.iteritems():
if v == "NULL":
line[k] = None
club = Club.objects.get(name=line['club_name'])
r = RaceClass()
r.name = line['name']
r.abrv = line['name']
r.pax = float(line['pax'])
r.club = club
r.save()
db.reset_queries()
race_class_map[line['id']] = r
index_class = RaceClass()
index_class.name = 'Index'
index_class.bump_class = True
index_class.abrv = "i"
index_class.pax = 1.0
index_class.club = club
index_class.save()
print "loading reg_types"
reg_type_map = {}
for line in csv.DictReader(open('old_data/regtype.csv')):
"""id","name","class_letters","reg_limit","index",
"description","club_name"""
for k, v in line.iteritems():
if v == "NULL":
line[k] = None
club = Club.objects.get(name=line['club_name'])
r = RaceClass()
r.pax_class = True
r.description = line['description']
r.name = line['name']
r.abrv = line['class_letters']
if line['reg_limit']:
r.user_reg_limit=line['reg_limit']
r.pax = 1.0
r.club = club
r.save()
reg_type_map[line['id']] = r
db.reset_queries()
print "loading registrations"
registration_map = {}
for line in csv.DictReader(open('old_data/registration.csv')):
"""id","number","paid","token","payer_id","transaction_id",
"price","class_points","index_points","index_flag","anon_f_name",
"anon_l_name","anon_car","driver_user_name","event_id","reg_type_id",
"car_id","race_class_id"""
for k, v in line.iteritems():
if v == "NULL":
line[k] = None
if not line['event_id']:
continue
rc = race_class_map[line['race_class_id']]
r = Registration()
r.paid = bool(line['paid'])
if line['price']!=None:
r.price = float(line['price'])
if line['driver_user_name']:
user = User.objects.get(username=line['driver_user_name'])
r.user_profile = user.get_profile()
r.number = int(line['number'])
r.paid = int(line['paid'])
r.index_points = line['index_points']
r.class_points = line['class_points']
if line['price']:
r.price = float(line['price'])
else:
r.price = 0.00
r.index_points = int(line['index_points'])
r.class_points = int(line['class_points'])
if line['anon_car']:
r._anon_car=line['anon_car'].strip()
if line['anon_l_name']:
r._anon_l_name=line['anon_l_name'].strip()
if line['anon_f_name']:
r._anon_f_name=line['anon_f_name'].strip()
r.race_class = rc
r.event = event_map[line['event_id']]
r.club = r.event.club
try:
if line['reg_type_id']:
r.pax_class=reg_type_map[line['reg_type_id']]
except:
pass
if line['index_flag']:
r.bump_class=index_class
try:
if line['car_id']:
r.car=Car.objects.get(pk=car_map[line['car_id']])
except:
pass
#TODO race_class_id
#TODO remove reg_detail class, and associate reg with UserProfile directly
#TODO registrations can be siblings for joint update
try:
r.save()
except:
r.number += 100
try:
r.save()
except:
continue
registration_map[line['id']] = r.pk
db.reset_queries()
print "loading sessions"
session_map = {}
for line in csv.DictReader(open('old_data/session.csv')):
"""id", "name", "event_id", "course_id"""
for k, v in line.iteritems():
if v == "NULL":
line[k] = None
try:
event_map[line['event_id']]
except:
continue
s = Session()
s.name = line['name']
s.event = event_map[line['event_id']]
s.club = s.event.club
s.save()
session_map[line['id']] = s
db.reset_queries()
print "loading results"
result_map = {}
for line in csv.DictReader(open('old_data/result.csv')):
"""id","registration_id","event_id","sess_id"""
for k, v in line.iteritems():
if v == "NULL":
line[k] = None
try:
registration_map[line['registration_id']]
session_map[line['sess_id']]
except:
continue
r = Result()
reg = Registration.objects.get(pk=registration_map[line['registration_id']])
r.reg = reg
r.session = session_map[line['sess_id']]
r.club = r.session.club
try:
r.save()
except Exception as err:
print registration_map[line['registration_id']].pk
print str(err)
print registration_map[line['registration_id']]
result_map[line['id']] = r.id
db.reset_queries()
print "loading runs"
for line in csv.DictReader(open('old_data/run.csv')):
"""id","base_time","calc_time","index_time","cones",
"penalty","result_id","result_2_id"""
for k, v in line.iteritems():
if v == "NULL":
line[k] = None
if line['base_time'] == None:
continue
try:
r = Run()
r.base_time = float(line['base_time'])
r.cones = int(line['cones'])
if line['penalty']:
r.penalty = line['penalty']
result = Result.objects.get(pk=result_map[line['result_id']])
r.result = result
r.club = r.result.club
r.save()
db.reset_queries()
except KeyError:
continue
print "calculating reg times"
for reg in Registration.objects.select_related('results').all().iterator():
reg.calc_times()
#House keeping
nora = Club.objects.get(safe_name='noraascc')
user = User.objects.get(username="******")
nora.group.user_set.add(user)
def handle(self, *args, **options):
clear_db()
s = Site.objects.all()[0]
s.domain = "dev.racerslane.com"
s.name = "Racerslane.com"
s.save()
print "Loading users"
reader = csv.DictReader(open('old_data/driver.csv', 'rU'))
for line in reader:
"""user_name","email","verified","activation_code",
"f_name","l_name","address","city","state",
"zip_code","_password"""
for k, v in line.iteritems():
if v == "NULL":
line[k] = None
u = User()
u.username = line['user_name']
u.email = line['email']
u.password = "******" % line['_password']
u.first_name = line['f_name']
u.last_name = line['l_name']
u.is_active = True
u.save()
up = u.get_profile()
up.address = line['address']
up.city = line['city']
up.state = line['state']
up.zip_code = line['zip_code']
up.save()
db.reset_queries()
#dev to make it so I can login to any account
justin = User.objects.get(username="******")
password = justin.password
#User.objects.all().update(password=password)
print "Loading Cars"
car_map = {}
reader = csv.DictReader(open('old_data/car.csv', 'rb'))
for line in reader:
"""id","nickname","make","model","color","avatar_file_loc",
"avatar_thumb_loc","year","owner_user_name"""
for k, v in line.iteritems():
if v == "NULL":
line[k] = None
if line['owner_user_name']:
try:
c = Car()
c.provisional = False
c.name = line['nickname']
c.make = line['make']
c.model = line['model']
if line['color']:
c.color = line['color']
c.year = line['year']
c.user_profile = User.objects.get(
username=line['owner_user_name']).get_profile()
s_car_id = (line['owner_user_name'], line['nickname'])
if exists('old_data/avatars/%s_%s_avatar' % s_car_id):
c.avatar.save(
'%s_%s_avatar' % s_car_id,
File(
open('old_data/avatars/%s_%s_avatar' %
s_car_id)))
c.thumb.save(
'%s_%s_thumb' % s_car_id,
File(
open('old_data/avatars/%s_%s_thumb' %
s_car_id)))
c.save()
db.reset_queries()
car_map[line['id']] = c.pk
except:
continue
print "Loading Clubs"
#read in clubs
club_map = {}
for line in csv.DictReader(open('old_data/club.csv', 'rU')):
""""name","web_link","process_payments","points_calc_type",
"membership_cost","renewal_cost","membership_terms","paypal_email",
"index_point_method","address","city","state","zip_code","""
for k, v in line.iteritems():
if v == "NULL":
line[k] = None
c = Club()
c.city = line['city']
c.address = line['address']
c.zip_code = line['zip_code']
c.state = line['state']
c.name = line['name']
c.process_payments = line['process_payments']
c.paypal_email = line['paypal_email']
c.membership_terms = line['membership_terms']
c.new_member_cost = line['membership_cost']
c.renew_cost = line['renewal_cost']
c.save()
club_map[line['name']] = c
db.reset_queries()
print "loading coupons"
for line in csv.DictReader(open('old_data/coupon.csv')):
"""coupon_code","club_name","discount_amount","uses_left","expires",
"permanent","driver_user_name","registration_id"""
for k, v in line.iteritems():
if v == "NULL":
line[k] = None
c = Coupon()
c.code = line['coupon_code']
c.permanent = bool(line['permanent'])
c.club = club_map[line['club_name']]
c.uses_left = line['uses_left']
c.discount_amount = line['discount_amount']
if line['expires']:
c.expires = datetime.datetime.strptime(line['expires'],
'%Y-%m-%d %H:%M:%S')
if line['driver_user_name']:
c.user_prof = User.objects.get(
username=line['driver_user_name']).get_profile()
c.save()
db.reset_queries()
print "loading memberships"
reader = csv.DictReader(open('old_data/membership.csv', 'rb'))
for line in reader:
""""id","club_name","number","start_date","valid_thru_date",
"price","paid","token","payer_id","transaction_id","anon_f_name",
"anon_l_name","driver_user_name"""
for k, v in line.iteritems():
if v == "NULL":
line[k] = None
m = Membership()
m.num = line['number']
m.start = datetime.datetime.strptime(line['start_date'],
'%Y-%m-%d %H:%M:%S')
m.valid_thru = datetime.datetime.strptime(line['valid_thru_date'],
'%Y-%m-%d %H:%M:%S')
m.paid = bool(line['paid'])
if line['price'] != None:
m.price = float(line['price'])
try:
m.user_prof = User.objects.get(
username=line['driver_user_name']).get_profile()
except User.DoesNotExist:
continue
m.club = club_map[line['club_name']]
m._anon_f_name = line['anon_f_name']
m._anon_l_name = line['anon_l_name']
m.save()
db.reset_queries()
print "loading locations"
location_map = {}
for line in csv.DictReader(open("old_data/location.csv")):
"""id","name","address","lat","lng","club_name"""
for k, v in line.iteritems():
if v == "NULL":
line[k] = None
club = Club.objects.get(name=line['club_name'])
l = Location()
l.name = line['name']
l.address = line['address']
#l.lat = line['lat']
#l.lon = line['lng']
l.club = club
l.save()
location_map[line['id']] = l
db.reset_queries()
print "loading seasons"
season_map = {}
for line in csv.DictReader(open('old_data/season.csv')):
s = Season()
""""id","year","active","club_name","drop_lowest_events"""
for k, v in line.iteritems():
if v == "NULL":
line[k] = None
club = Club.objects.get(name=line['club_name'])
s.club = club
s.year = int(line['year'])
s.drop_lowest_events = int(line['drop_lowest_events'])
s.save()
season_map[line['id']] = s
db.reset_queries()
print "loading events"
event_map = {}
for line in csv.DictReader(open('old_data/event.csv', 'rU')):
"""id","name","note","date","registration_close","member_cost",
"non_member_cost","pay_at_event_cost","location_id","club_name",
"season_id","multiplier","count_points"""
for k, v in line.iteritems():
if v == "NULL":
line[k] = None
e = Event()
e.name = line['name']
e.note = line['note']
e.date = datetime.datetime.strptime(line['date'],
'%Y-%m-%d %H:%M:%S')
e.reg_close = line['registration_close'] + "-05:00"
e.member_price = float(line['member_cost'])
e.non_member_price = float(line['non_member_cost'])
e.non_pre_pay_penalty = float(line['pay_at_event_cost'])
e.season = season_map[line['season_id']]
e.club = e.season.club
e.count_points = int(line['count_points'])
e.multiplier = int(line['multiplier'])
if line['location_id']:
e.location = location_map[line['location_id']]
e.save()
event_map[line['id']] = e
db.reset_queries()
print "loading race_classes"
race_class_map = {}
for line in csv.DictReader(open('old_data/raceclass.csv')):
"""id","pax","name","club_name"""
for k, v in line.iteritems():
if v == "NULL":
line[k] = None
club = Club.objects.get(name=line['club_name'])
r = RaceClass()
r.name = line['name']
r.abrv = line['name']
r.pax = float(line['pax'])
r.club = club
r.save()
db.reset_queries()
race_class_map[line['id']] = r
index_class = RaceClass()
index_class.name = 'Index'
index_class.bump_class = True
index_class.abrv = "i"
index_class.pax = 1.0
index_class.club = club
index_class.save()
print "loading reg_types"
reg_type_map = {}
for line in csv.DictReader(open('old_data/regtype.csv')):
"""id","name","class_letters","reg_limit","index",
"description","club_name"""
for k, v in line.iteritems():
if v == "NULL":
line[k] = None
club = Club.objects.get(name=line['club_name'])
r = RaceClass()
r.pax_class = True
r.description = line['description']
r.name = line['name']
r.abrv = line['class_letters']
if line['reg_limit']:
r.user_reg_limit = line['reg_limit']
r.pax = 1.0
r.club = club
r.save()
reg_type_map[line['id']] = r
db.reset_queries()
print "loading registrations"
registration_map = {}
for line in csv.DictReader(open('old_data/registration.csv')):
"""id","number","paid","token","payer_id","transaction_id",
"price","class_points","index_points","index_flag","anon_f_name",
"anon_l_name","anon_car","driver_user_name","event_id","reg_type_id",
"car_id","race_class_id"""
for k, v in line.iteritems():
if v == "NULL":
line[k] = None
if not line['event_id']:
continue
rc = race_class_map[line['race_class_id']]
r = Registration()
r.paid = bool(line['paid'])
if line['price'] != None:
r.price = float(line['price'])
if line['driver_user_name']:
user = User.objects.get(username=line['driver_user_name'])
r.user_profile = user.get_profile()
r.number = int(line['number'])
r.paid = int(line['paid'])
r.index_points = line['index_points']
r.class_points = line['class_points']
if line['price']:
r.price = float(line['price'])
else:
r.price = 0.00
r.index_points = int(line['index_points'])
r.class_points = int(line['class_points'])
if line['anon_car']:
r._anon_car = line['anon_car'].strip()
if line['anon_l_name']:
r._anon_l_name = line['anon_l_name'].strip()
if line['anon_f_name']:
r._anon_f_name = line['anon_f_name'].strip()
r.race_class = rc
r.event = event_map[line['event_id']]
r.club = r.event.club
try:
if line['reg_type_id']:
r.pax_class = reg_type_map[line['reg_type_id']]
except:
pass
if line['index_flag']:
r.bump_class = index_class
try:
if line['car_id']:
r.car = Car.objects.get(pk=car_map[line['car_id']])
except:
pass
#TODO race_class_id
#TODO remove reg_detail class, and associate reg with UserProfile directly
#TODO registrations can be siblings for joint update
try:
r.save()
except:
r.number += 100
try:
r.save()
except:
continue
registration_map[line['id']] = r.pk
db.reset_queries()
print "loading sessions"
session_map = {}
for line in csv.DictReader(open('old_data/session.csv')):
"""id", "name", "event_id", "course_id"""
for k, v in line.iteritems():
if v == "NULL":
line[k] = None
try:
event_map[line['event_id']]
except:
continue
s = Session()
s.name = line['name']
s.event = event_map[line['event_id']]
s.club = s.event.club
s.save()
session_map[line['id']] = s
db.reset_queries()
print "loading results"
result_map = {}
for line in csv.DictReader(open('old_data/result.csv')):
"""id","registration_id","event_id","sess_id"""
for k, v in line.iteritems():
if v == "NULL":
line[k] = None
try:
registration_map[line['registration_id']]
session_map[line['sess_id']]
except:
continue
r = Result()
reg = Registration.objects.get(
pk=registration_map[line['registration_id']])
r.reg = reg
r.session = session_map[line['sess_id']]
r.club = r.session.club
try:
r.save()
except Exception as err:
print registration_map[line['registration_id']].pk
print str(err)
print registration_map[line['registration_id']]
result_map[line['id']] = r.id
db.reset_queries()
print "loading runs"
for line in csv.DictReader(open('old_data/run.csv')):
"""id","base_time","calc_time","index_time","cones",
"penalty","result_id","result_2_id"""
for k, v in line.iteritems():
if v == "NULL":
line[k] = None
if line['base_time'] == None:
continue
try:
r = Run()
r.base_time = float(line['base_time'])
r.cones = int(line['cones'])
if line['penalty']:
r.penalty = line['penalty']
result = Result.objects.get(pk=result_map[line['result_id']])
r.result = result
r.club = r.result.club
r.save()
db.reset_queries()
except KeyError:
continue
print "calculating reg times"
for reg in Registration.objects.select_related(
'results').all().iterator():
reg.calc_times()
#House keeping
nora = Club.objects.get(safe_name='noraascc')
user = User.objects.get(username="******")
nora.group.user_set.add(user)
