def update_existing_family(already, family): if not already.hid: already.hid = family.accessid if not already.phone: already.phone = family.home if not already.email: already.email = family.email if not already.address: already.address = Addresses.create( line1=family.street, line2='Apt. ' + family.apt if family.apt else '', city=family.city.title(), state=family.state.upper(), zipcode=family.zip, ) if not already.mother: already.mother = Parents.create( first=family.mfirst, # alt_first = family.mnick, alt_last='' if family.mlast == family.family else family.mlast, sex='F', alt_phone=0 if family.mcell == family.home else family.mcell, family_id=already.id, ) if not already.father: already.father = Parents.create( first=family.ffirst, # alt_first = family.fnick, alt_last='' if family.flast == family.family else family.flast, sex='M', alt_phone=0 if family.fcell == family.home else family.fcell, family_id=already.id, ) for student in OldStudents.filter(familyid=family.accessid): alreadyStudent = find_existing_student(already, student) if alreadyStudent: update_existing_student(alreadyStudent, student) else: stu = Students.create( oid=student.id, hid=student.studentid, created_at=student.creationdate, updated_at=student.moddate, family=already, first=namecase(student.first), alt_last='' if (not already) or already.last == student.last else student.last, alt_email='' if (not already) or already.email == student.email else student.email, sex=student.sex, birthday=student.dob if student.dob else '2020-12-31', grad_year=est_grad(student), tshirt=sub(student.tshirt, {'AXL': 'XL'}), needs=student.needsdescribe)
def load_post(request): start_import = datetime.now() nUsers = 0 nFamilies = 0 nStudents = 0 nParents = 0 nAddresses = 0 nVenues = 0 nCourseTrads = 0 nCourses = 0 nEnrollments = 0 json_dump = request.POST['json_dump'] request.session['json_dump'] = '' data = json.loads(json_dump) for ct in data['coursetrads']: if 'alias_id' not in ct: already = CourseTrads.fetch(id=ct['id']) if already: print 'Exists: '+ct['title'].upper() else: print 'Importing '+ct['title'].upper() ct.setdefault('default','enrolled' if ct['after_tuit'] == 0 else 'need_pay') CourseTrads.create(**ct) nCourseTrads += 1 for ct in data['coursetrads']: if 'alias_id' in ct: already = CourseTrads.fetch(id=ct['id']) if already: print 'Exists: '+ct['title'].upper() else: print 'Importing '+ct['title'].upper() ct.setdefault('default','--------') alias = CourseTrads.get(id=ct.pop('alias_id')) ct['alias'] = alias CourseTrads.create(**ct) nCourseTrads += 1 # cts = CourseTrads.filter(e=True) # for year in data['years']: # for ct in cts: # ct.make(year) # nCourses += 1 for fam in data['families']: print fam['last'] family = copy(fam,['last','phone','email']) if 'address' in fam: address = copy(fam['address']) if 'zipcode' not in address: address['zipcode'] = 00000 address = Addresses.create(**address) nAddresses += 1 family['address'] = address family = Families.create(**family) family.update_name_num() nFamilies += 1 user_obj = fam.get('user') if user_obj: Users.create( owner_type = 'Family', owner_id = family.id, username = user_obj['username'], password = user_obj['password'], permission = user_obj['permission'], ) nUsers += 1 if 'mother' in fam: mother = copy(fam['mother']) mother['family_id'] = family.id family.mother = Parents.create(**mother) nParents += 1 if 'father' in fam: father = copy(fam['father']) father['family_id'] = family.id family.father = Parents.create(**father) nParents += 1 family.save() for stu in fam['students']: if 'first' in stu: print ' '+stu['first'] student = copy(stu) enrollments = student.pop('enrollments') if 'enrollments' in student else [] student['family'] = family newStudent = Students.create(**student) for enrollment in enrollments: rolled = type(enrollment) in [object,dict] course_id = enrollment['course_id'] if rolled else enrollment print ' '+course_id course = Courses.fetch(id=course_id) if not course: course = Courses.create_by_id(course_id) nCourses += 1 enrollment_kwargs = { 'student': newStudent, 'course' : course, 'status' : enrollment.setdefault('status','enrolled') if hasattr(enrollment,'setdefault') else 'enrolled' } if rolled: enrollment_kwargs['role'] = enrollment['role'] enrollment_kwargs['role_type'] = enrollment['role_type'] Enrollments.create(**enrollment_kwargs) nEnrollments += 1 # if type(enrollment) in [str,unicode]: # Courses.get(id=enrollment).sudo_enroll(newStudent) # else: # course = Courses.get(id=enrollment['course_id']) # Enrollments.create(course=course, student=newStudent, role=enrollment['role'], role_type=enrollment['role_type']) nStudents += 1 print "- assign name_num's" Each(Families.all()).update_name_num() print '- order_coursetrads' order_coursetrads() # year = getyear() # print '- make {}'.format(year) # nCourses += make(year) # print '- assign developer: Wolf Elkan' # elkan = Families.fetch(last='Elkan') # if elkan: # Users.create( # username='******', # password='******', # permission=7, # owner_type='Family', # owner_id=elkan.id # ) import_duration = datetime.now() - start_import print '\nIMPORT COMPLETED' print 'Users: ' + str(nUsers).rjust(5) print 'Families: ' + str(nFamilies).rjust(5) print 'Students: ' + str(nStudents).rjust(5) print 'Parents: ' + str(nParents).rjust(5) print 'Addresses: ' + str(nAddresses).rjust(5) print 'Venues: ' + str(nVenues).rjust(5) print 'Traditions:' + str(nCourseTrads).rjust(5) print 'Courses: ' + str(nCourses).rjust(5) print 'Enrollments:'+ str(nEnrollments).rjust(4) print 'Time: ' + str(import_duration) return redirect('/seed/load/')