2017-01-12 34 views
1

我修改Django中的用戶模型後出現錯誤。createsuperuser沒有要求用戶名

當我要創建一個超級用戶時,它沒有提示輸入用戶名,而是跳過它,反正對象propery用戶名仍然需要,導致用戶創建失敗。

import jwt 
from django.db import models 
from django.contrib.auth.models import (AbstractBaseUser, BaseUserManager, 
PermissionsMixin) 
from datetime import datetime, timedelta 
from django.conf import settings 
# from api.core.models import TimeStampedModel 


class AccountManager(BaseUserManager): 

    def create_user(self, username, email, password=None, **kwargs): 
     if not email: 
      raise ValueError('Please provide a valid email address') 

     if not kwargs.get('username'): 
      # username = 'what' 
      raise ValueError('User must have a username') 

     account = self.model(
        email=self.normalize_email(email), username=kwargs.get('username')) 

     account.set_password(password) 
     account.save() 

     return account 

    def create_superuser(self,username, email, password, **kwargs): 
     account = self.create_user(username, email, password, **kwargs) 

     account.is_admin = True 
     account.save() 

     return account 

class Account(AbstractBaseUser): 
    email = models.EmailField(unique=True) 
    username = models.CharField(max_length=40, unique=True) 

    first_name = models.CharField(max_length=40) 
    last_name = models.CharField(max_length =40, blank=True) 

    tagline = models.CharField(max_length=200, blank=True) 

    is_admin = models.BooleanField(default=False) 

    USERNAME_FIELD = 'email' 
    REQUIRED_FIELD = ['username'] 


    objects = AccountManager() 

    def __str__(self): 
     return self.username 

    @property 
    def token(self): 
     return generate_jwt_token() 

    def generate_jwt_token(self): 
     dt = datetime.now + datetime.timedelta(days=60) 

     token = jwt.encode({ 
      'id': self.pk, 
      'exp': int(dt.strftime('%s')) 
      }, settings.SECRET_KEY, algorithm='HS256') 
     return token.decode('utf-8') 

    def get_full_name(self): 
     return ' '.join(self.first_name, self.last_name) 

    def get_short_name(self): 
     return self.username 

它的結果是:

manage.py createsuperuser 
Running 'python /home/gema/A/PyProject/janet_x/janet/manage.py createsuperuser' command: 
Email: [email protected] 
Password: 
Password (again): 
Traceback (most recent call last): 
    File "/home/gema/A/PyProject/janet_x/janet/manage.py", line 22, in <module> 
    execute_from_command_line(sys.argv) 
    File "/home/gema/.virtualenvs/JANET/lib/python3.5/site-packages/django/core/management/__init__.py", line 367, in execute_from_command_line 
    utility.execute() 
    File "/home/gema/.virtualenvs/JANET/lib/python3.5/site-packages/django/core/management/__init__.py", line 359, in execute 
    self.fetch_command(subcommand).run_from_argv(self.argv) 
    File "/home/gema/.virtualenvs/JANET/lib/python3.5/site-packages/django/core/management/base.py", line 294, in run_from_argv 
    self.execute(*args, **cmd_options) 
    File "/home/gema/.virtualenvs/JANET/lib/python3.5/site-packages/django/contrib/auth/management/commands/createsuperuser.py", line 63, in execute 
    return super(Command, self).execute(*args, **options) 
    File "/home/gema/.virtualenvs/JANET/lib/python3.5/site-packages/django/core/management/base.py", line 345, in execute 
    output = self.handle(*args, **options) 
    File "/home/gema/.virtualenvs/JANET/lib/python3.5/site-packages/django/contrib/auth/management/commands/createsuperuser.py", line 183, in handle 
    self.UserModel._default_manager.db_manager(database).create_superuser(**user_data) 
TypeError: create_superuser() missing 1 required positional argument: 'username' 

當我使用shell:

Account.objects.create(u'administer', u'[email protected]', u'password123') 

返回:

return getattr(self.get_queryset(), name)(*args, **kwargs) 
TypeError: create() takes 1 positional argument but 3 were given 

怎麼可能是錯的? 謝謝。

+0

嘗試在此處僅提供用戶名..def create_superuser(self,username,** kwargs): –

回答

3

REQUIRED_FIELD應該是REQUIRED_FIELDS(複數),否則你將不會被提示輸入用戶名(或任何其他必填字段),因爲Django在REQUIRED_FIELDS中找不到任何內容。

舉個例子,我用這個的UserManager在我的項目之一:

class UserManager(BaseUserManager): 
    def create_user(self, email, password=None, first_name=None, last_name=None, **extra_fields): 
     if not email: 
      raise ValueError('Enter an email address') 
     if not first_name: 
      raise ValueError('Enter a first name') 
     if not last_name: 
      raise ValueError('Enter a last name') 
     email = self.normalize_email(email) 
     user = self.model(email=email, first_name=first_name, last_name=last_name, **extra_fields) 
     user.set_password(password) 
     user.save(using=self._db) 
     return user 

    def create_superuser(self, email, password, first_name, last_name): 
     user = self.create_user(email, password=password, first_name=first_name, last_name=last_name) 
     user.is_superuser = True 
     user.is_staff = True 
     user.save(using=self._db) 
     return user 

USERNAME_FIELD設置爲emailREQUIRED_FIELDS('first_name', 'last_name')。你應該能夠適應這個例子來適應你的需求。

+0

謝謝,我認爲你是對的,但我該如何重置這個配置?因爲在我將其改爲複數後,它仍然沒有詢問用戶名。 –

+0

@BarackObama我已經用一些示例代碼更新了我的答案,您應該能夠適應您的需求。 –

+0

感謝它給我啓迪。 –

1

此位沒有意義:

USERNAME_FIELD = 'email' 
REQUIRED_FIELD = ['username'] 

你爲什麼要設置USERNAME_FIELD到 「電子郵件」?當然,它應該是「用戶名」。

+0

我試圖改變它,但仍不會詢問用戶名。你知道是否有任何manage.py命令的直接配置嗎? –

相關問題