''' import csv from classData import user from classData import userprofile as uprofile from classData import course_enrollment as ce if __name__ == '__main__': csv.field_size_limit(1000000) uIn = csv.reader(open('users.csv', 'r')) uDict = user.builddict(uIn) upIn = csv.reader(open('profiles.csv', 'r')) upDict = uprofile.builddict(upIn) ceIn = csv.reader(open('enrollment.csv', 'r')) ceDict = ce.builddict(ceIn) of = csv.writer(open('userDiffs.csv', 'w')) of.writerow(['ids in user file, not in profiles file']) for u in iter(uDict): if u not in upDict: of.writerow([u]) of.writerow(['ids in profiles file, not in user file']) for p in iter(upDict): if p not in uDict: of.writerow([p]) of.writerow(['ids in user file, not in enrollment file']) for u in iter(uDict): if u not in ceDict: of.writerow([u])
week1 = wk1 + '/' + ck_course week2 = wk2 + '/' + ck_course userFile = '/users.csv' certFile = '/certificates.csv' enroll = '/enrollment.csv' uf1 = csv.reader(open(week2 + userFile, 'r')) uf2 = csv.reader(open(week1 + userFile, 'r')) cf1 = csv.reader(open(week2 + certFile, 'r')) cf2 = csv.reader(open(week1 + certFile, 'r')) ef1 = csv.reader(open(week2 + enroll, 'r')) ef2 = csv.reader(open(week1 + enroll, 'r')) u1dict = user.builddict(uf1) u2dict = user.builddict(uf2) c1dict = certificates.builddict(cf1) c2dict = certificates.builddict(cf2) e1dict = course_enrollment.builddict(ef1) e2dict = course_enrollment.builddict(ef2) OneNotTwo = compareUsers(u1dict, u2dict) TwoNotOne = compareUsers(u2dict, u1dict) for u in iter(OneNotTwo): if u in c1dict and c1dict[u].status == 'downloadable': OneNotTwo[u] = 'y' for u in iter(TwoNotOne): if u in c2dict and c2dict[u].status == 'downloadable': TwoNotOne[u] = 'y' outfile = csv.writer(open('userDiff06020616' + ck_course + '.csv', 'w')) outfile.writerow(['Users in ' + wk1 + ' list but not in ' + wk2 + ' list']) outfile.writerow(['User id', 'Certificate granted', 'Date enrolled']) for u in iter(OneNotTwo):