Add check for roles

server/core/management/commands/create_teacher.py

@@ -1,8 +1,4 @@
-import csv
-
 from django.core.management import BaseCommand
-import os
-from django.conf import settings
 
 from users.models import User, SchoolClass, Role, UserRole
 
@@ -28,7 +24,11 @@ class Command(BaseCommand):
             user.save()
 
             self.stdout.write("Assigning teacher role")
-            teacher = Role.objects.get(key='teacher')
+            try:
+                teacher = Role.objects.get(key='teacher')
+            except Role.DoesNotExist:
+                Role.objects.create_default_roles()
+                teacher = Role.objects.get(key='teacher')
             UserRole.objects.get_or_create(user=user, role=teacher)
 
             school, _ = SchoolClass.objects.get_or_create(name=school_class_name)

server/core/management/commands/export_assignments.py

@@ -1,6 +1,6 @@
 from django.core.management import BaseCommand
 
-from assignments.models import Assignment, StudentSubmission
+from assignments.models import Assignment
 import random
 import json
 from django.db.models import Q

server/core/management/commands/generate_codes.py

@@ -1,6 +1,5 @@
 from django.core.management import BaseCommand
 
-from notes.models import ContentBlockBookmark, ModuleBookmark, ChapterBookmark
 from users.models import SchoolClass