Browse Source
Merge pull request #787 from cgill95/enhancement/avoid_username_is_taken
Enhancement/avoid username is taken
pull/862/head
Hiroki Nakayama
5 years ago
committed by
GitHub
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with
6 additions and
2 deletions
-
app/server/management/commands/create_admin.py
|
|
@ -20,12 +20,16 @@ class Command(createsuperuser.Command): |
|
|
|
|
|
|
|
try: |
|
|
|
super().handle(*args, **options) |
|
|
|
except IntegrityError: |
|
|
|
self.stderr.write(f'User {username} already exists.') |
|
|
|
except Exception as err: |
|
|
|
if 'is already taken' in str(err): |
|
|
|
self.stderr.write(f'User {username} already exists.') |
|
|
|
else: |
|
|
|
raise |
|
|
|
|
|
|
|
if password: |
|
|
|
database = options.get('database') |
|
|
|
db = self.UserModel._default_manager.db_manager(database) |
|
|
|
user = db.get(username=username) |
|
|
|
user.set_password(password) |
|
|
|
self.stderr.write(f'Setting password for User {username}.') |
|
|
|
user.save() |