diff --git a/csv/load_perm.py b/csv/load_perm.py index 1d70bf3..46c3d38 100644 --- a/csv/load_perm.py +++ b/csv/load_perm.py @@ -27,28 +27,29 @@ if __name__ == '__main__': try: user = get_user_model().objects.get(id=row['user']) - p, c = Progress.objects.get_or_create( - user=user, - route=route, - teacher=get_user_model().objects.get(id=row['teacher']), + except get_user_model().DoesNotExist: + print(row['user']) + break + + p = Progress.objects.create( + user=user, + route=route, + teacher=get_user_model().objects.get(id=row['teacher']), + ) + + pivots = json.loads(row['success']) + for pivot in pivots: + pv = PivotProgressVertex.objects.create( + progress=p, + vertex=Vertex.objects.get(id=pivot['id']).token, + teacher=get_user_model().objects.get(id=pivot['teacher']), + status=1, ) - pivots = json.loads(row['success']) - for pivot in pivots: - pv = PivotProgressVertex.objects.create( - progress=p, - vertex=Vertex.objects.get(id=pivot['id']).token, - teacher=get_user_model().objects.get(id=pivot['teacher']), - status=1, - ) - - if pivot['date']: - pv.date = datetime.strptime(pivot['date'].split('.')[0], '%Y-%m-%d %H:%M:%S') - pv.status = 2 - pv.save() + if pivot['date']: + pv.date = datetime.strptime(pivot['date'].split('.')[0], '%Y-%m-%d %H:%M:%S') + pv.status = 2 + pv.save() - g = Group.objects.get(name='students') - g.user_set.add(user) - - except get_user_model().DoesNotExist: - print(row['user']) + g = Group.objects.get(name='students') + g.user_set.add(user)