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

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)