Commit 3babf539 by cahrens

Handle the case of script being terminated before the user was deleted.

parent 42f71156
...@@ -2,13 +2,23 @@ from auth.authz import _grant_instructors_creator_access ...@@ -2,13 +2,23 @@ from auth.authz import _grant_instructors_creator_access
from django.core.management.base import BaseCommand from django.core.management.base import BaseCommand
from django.contrib.auth.models import User from django.contrib.auth.models import User
from django.db.utils import IntegrityError
class Command(BaseCommand): class Command(BaseCommand):
help = 'Grants all users with INSTRUCTOR role permission to create courses' help = 'Grants all users with INSTRUCTOR role permission to create courses'
def handle(self, *args, **options): def handle(self, *args, **options):
admin = User.objects.create_user('populate_creators_command', 'grant+creator+access@edx.org', 'foo') username = 'populate_creators_command'
admin.is_staff = True email = 'grant+creator+access@edx.org'
try:
admin = User.objects.create_user(username, email, 'foo')
admin.is_staff = True
admin.save()
except IntegrityError:
# If the script did not complete the last time it was run,
# the admin user will already exist.
admin = User.objects.get(username=username, email=email)
_grant_instructors_creator_access(admin) _grant_instructors_creator_access(admin)
admin.delete() admin.delete()
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment