diff --git a/apps/jumpserver/settings.py b/apps/jumpserver/settings.py index 6d6eae6c6..7dfc1e603 100644 --- a/apps/jumpserver/settings.py +++ b/apps/jumpserver/settings.py @@ -96,7 +96,7 @@ TEMPLATES = [ }, ] -WSGI_APPLICATION = 'jumpserver.wsgi.application' +# WSGI_APPLICATION = 'jumpserver.wsgi.application' # Database # https://docs.djangoproject.com/en/1.10/ref/settings/#databases @@ -230,9 +230,6 @@ MEDIA_URL = '/media/' MEDIA_ROOT = os.path.join(BASE_DIR, 'media').replace('\\', '/') + '/' -# Custom User Auth model -AUTH_USER_MODEL = 'users.User' - # Use django-bootstrap-form to format template, input max width arg BOOTSTRAP_COLUMN_COUNT = 11 @@ -275,3 +272,7 @@ WS4REDIS_PREFIX = 'demo' SESSION_ENGINE = 'redis_sessions.session' SESSION_REDIS_PREFIX = 'session' + +# Custom User Auth model +AUTH_USER_MODEL = 'users.User' + diff --git a/apps/users/models.py b/apps/users/models.py index b086780d2..2681a2470 100644 --- a/apps/users/models.py +++ b/apps/users/models.py @@ -10,9 +10,8 @@ from django.db import models from django.contrib.auth.models import AbstractUser, Permission from django.db.models.signals import post_save from django.dispatch import receiver -from rest_framework.authtoken.models import Token from django.db import IntegrityError - +from rest_framework.authtoken.models import Token # class Role(models.Model): @@ -250,5 +249,8 @@ def generate_fake(): @receiver(post_save, sender=settings.AUTH_USER_MODEL) def create_auth_token(sender, instance=None, created=False, **kwargs): if created: - Token.objects.create(user=instance) + try: + Token.objects.create(user=instance) + except IntegrityError: + pass diff --git a/apps/users/templates/users/user_list.html b/apps/users/templates/users/user_list.html index 1c61d3cdf..e1c8da5cd 100644 --- a/apps/users/templates/users/user_list.html +++ b/apps/users/templates/users/user_list.html @@ -42,7 +42,7 @@