diff --git a/csv/load_perm.py b/csv/load_perm.py index 59c91a9..8ffa6fc 100644 --- a/csv/load_perm.py +++ b/csv/load_perm.py @@ -46,17 +46,21 @@ if __name__ == '__main__': 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, - ) + try: + 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'] and not pivot['date'] == 'None': + pv.date = datetime.strptime(pivot['date'].split('.')[0], '%Y-%m-%d %H:%M:%S') + pv.status = 2 + pv.save() - if pivot['date'] and not pivot['date'] == 'None': - pv.date = datetime.strptime(pivot['date'].split('.')[0], '%Y-%m-%d %H:%M:%S') - pv.status = 2 - pv.save() + except Vertex.DoesNotExist: + pass g = Group.objects.get(name='students') g.user_set.add(user)