def upload(request):
stats = {
'total': 0,
'new_mentors': 0,
'new_users': 0,
}
if request.method == "POST":
form = CSVImportForm(request.POST, request.FILES)
if form.is_valid():
# form is valid, start parsing CSV.
sessions = {
'2 August': DojoSession.objects.get(date = datetime.date(2014, 8, 2)),
'9 August': DojoSession.objects.get(date = datetime.date(2014, 8, 9)),
'16 August': DojoSession.objects.get(date = datetime.date(2014, 8, 16)),
'23 August': DojoSession.objects.get(date = datetime.date(2014, 8, 23)),
'30 August': DojoSession.objects.get(date = datetime.date(2014, 8, 30)),
'6 September': DojoSession.objects.get(date = datetime.date(2014, 9, 6)),
'13 September': DojoSession.objects.get(date = datetime.date(2014, 9, 13)),
'20 September': DojoSession.objects.get(date = datetime.date(2014, 9, 20)),
}
# First detect dialect
csvfile = request.FILES['csv']
dialect = csv.Sniffer().sniff(csvfile.read(2048))
csvfile.seek(0)
# Open reader.
reader = csv.DictReader(csvfile, dialect=dialect)
# Parse each row.
for row in reader:
# first, check if mentor already exists.
same = Mentor.objects.filter(user__email = row['Email address'])
if (same):
m = same[0]
stats
else:
print "New mentor %s" % row['Full name']
m = Mentor()
stats['new_mentors'] += 1
# Check to see if a non-mentor user exists with the mentor's email
same = get_user_model().objects.filter(email = row['Email address'])
if (same):
m.user = same[0]
else:
pwd = get_user_model().objects.make_random_password()
print "Password for %s: %s" % (row['Full name'], pwd)
name_array = row['Full name'].split(' ', 1)
m.user = get_user_model().objects.create_user(row['Email address'], name_array[0], name_array[1], pwd)
m.user.save()
stats['new_users'] += 1
# Add contact number
m.user.phone_number = '0' + row['Mobile number']
# Add new user to mentors group
group = Group.objects.get_or_create(name = 'Mentors')
m.user.groups.add(group[0])
m.user.save()
m.uni = csv_tools.none_catch(row['University'])
m.uni_study = csv_tools.none_catch(row['Study'])
m.work = csv_tools.none_catch(row['Work'])
# shirt size mapping
ts = row['T-shirt size']
if ts == 'I already have a CoderDojo WA tshirt':
m.needs_shirt = False
elif ts == 'Male S':
m.shirt_size = Mentor.MALE_SMALL
elif ts == 'Male M':
m.shirt_size = Mentor.MALE_MEDIUM
elif ts == 'Male L':
m.shirt_size = Mentor.MALE_LARGE
elif ts == 'Male XL':
m.shirt_size = Mentor.MALE_EXTRA_LARGE
elif ts == 'Female XS':
m.shirt_size = Mentor.FEMALE_EXTRA_SMALL
elif ts == 'Female S':
m.shirt_size = Mentor.FEMALE_SMALL
elif ts == 'Female M':
m.shirt_size = Mentor.FEMALE_MEDIUM
elif ts == 'Female L':
m.shirt_size = Mentor.FEMALE_LARGE
elif ts == 'Female XL':
m.shirt_size = Mentor.FEMALE_EXTRA_LARGE
# WWCC
m.wwcc = csv_tools.none_catch(row['WWCC Number'])
# Curtin status
curtin_status = row['Curtin Status']
if curtin_status == 'Associate':
m.curtin_status = Mentor.ASSOCIATE
elif curtin_status == 'Staff':
m.curtin_status = Mentor.STAFF
elif curtin_status == 'Neither/Not Sure':
m.curtin_status = Mentor.NEITHER
m.curtin_id = csv_tools.none_catch(row['Curtin Associate/Staff ID'])
# Coding experience
coding_xp = row['Coding Experience']
if coding_xp == 'I know nothing but am keen to learn!':
m.coding_experience = Mentor.NOTHING
elif coding_xp == 'I know some basics':
m.coding_experience = Mentor.SOMETHING
elif coding_xp == 'I know a great deal':
m.coding_experience = Mentor.EVERYTHING
# Child-related experience
children_xp = row['Child-Related Experience']
if children_xp == 'I know nothing but am keen to learn!':
m.children_experience = Mentor.NOTHING
elif children_xp == 'I know some basics':
m.children_experience = Mentor.SOMETHING
elif children_xp == 'I know a great deal':
m.children_experience = Mentor.EVERYTHING
# Gotta save before you can do M2M relations.
m.save()
# Roles
for role in Role.objects.all():
if role.name in row['Roles'].decode('utf-8') and role not in m.roles_desired.all():
m.roles_desired.add(role)
# Shift Availabilities
for session in sessions:
if row['Availability [Saturday %s]' % session] == 'Available':
m.shift_availabilities.add(sessions[session])
m.save()
stats['total'] += 1
if stats['total'] > 0:
return render(request, 'mentors/upload/success.html', {
'stats': stats,
})
else:
return render(request, 'mentors/upload/failure.html')
