Ограничение на журналы (тем, уроков, курсов)

csv/load_perm.py

@@ -46,17 +46,21 @@ if __name__ == '__main__':
 
             pivots = json.loads(row['success'])
             for pivot in pivots:
-                pv = PivotProgressVertex.objects.create(
+                try:
-                    progress=p,
+                    pv = PivotProgressVertex.objects.create(
-                    vertex=Vertex.objects.get(id=pivot['id']).token,
+                        progress=p,
-                    teacher=get_user_model().objects.get(id=pivot['teacher']),
+                        vertex=Vertex.objects.get(id=pivot['id']).token,
-                    status=1,
+                        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':
+                except Vertex.DoesNotExist:
-                    pv.date = datetime.strptime(pivot['date'].split('.')[0], '%Y-%m-%d %H:%M:%S')
+                    pass
-                    pv.status = 2
-                    pv.save()
 
             g = Group.objects.get(name='students')
             g.user_set.add(user)