def setUp(self): self.u1 = KanriUser.objects.create_user( email = '*****@*****.**', first_name = 'John', last_name = 'Smith', password = '******', ) self.u1.phone_number = '0400000000' self.u1.save() self.m1 = Mentor( uni = 'Notre Dame', uni_study = 'Basket Weaving', work = 'CEO', shirt_size = Mentor.FEMALE_EXTRA_LARGE, needs_shirt = True, curtin_status = Mentor.NEITHER, coding_experience = knowledge.NOTHING, children_experience = knowledge.NOTHING, user = self.u1 ) self.m1.save()
def dashboard(request):
return render(request, 'mentors/dashboard/index.html', {
'mentor': Mentor.get_mentor_for_user(request.user)
})
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')
class NinjaTestCase(TestCase): def setUp(self): self.u1 = KanriUser.objects.create_user( email = '*****@*****.**', first_name = 'John', last_name = 'Smith', password = '******', ) self.u1.phone_number = '0400000000' self.u1.save() self.m1 = Mentor( uni = 'Notre Dame', uni_study = 'Basket Weaving', work = 'CEO', shirt_size = Mentor.FEMALE_EXTRA_LARGE, needs_shirt = True, curtin_status = Mentor.NEITHER, coding_experience = knowledge.NOTHING, children_experience = knowledge.NOTHING, user = self.u1 ) self.m1.save() def test_name(self): self.assertEqual(self.m1.name(), 'John S') def test_get_call_url(self): self.assertEqual(self.m1.get_call_url(), 'tel:0400000000') def test_get_email_url(self): self.assertEqual(self.m1.get_email_url(), 'mailto:[email protected]') def test_wwcc_status_correct(self): # Just WWCC number self.m1.wwcc = 1234 self.assertEqual(self.m1.get_wwcc_status(), 'Recorded (1234)') # Just WWCC receipt self.m1.wwcc = None self.m1.wwcc_receipt = 5678 self.assertEqual(self.m1.get_wwcc_status(), 'Processing (5678)') # Number and receipt self.m1.wwcc = 1234 self.m1.wwcc_receipt = 5678 self.assertEqual(self.m1.get_wwcc_status(), 'Recorded (1234)') def test_get_curtin_status_correct(self): # Neither self.assertEqual(self.m1.get_curtin_status(), 'Neither/not sure') # Staff (No ID) self.m1.curtin_status = Mentor.STAFF self.assertEqual(self.m1.get_curtin_status(), 'Staff') # Associate (No ID) self.m1.curtin_status = Mentor.ASSOCIATE self.assertEqual(self.m1.get_curtin_status(), 'Associate') # Associate (ID) self.m1.curtin_id = '123456J' self.assertEqual(self.m1.get_curtin_status(), 'Associate (123456J)') # Staff (ID) self.m1.curtin_status = Mentor.STAFF self.assertEqual(self.m1.get_curtin_status(), 'Staff (123456J)')