Django User 模块之 AbstractUser 扩展

最近在写博客，刚好写到用户注册注销模块，觉得这一方面还是挺有趣的。当尝试掀开 Django 的源代码时一切 API 就不会变得那么摸不着。顺着读Django 的各模块源码，我们可以更灵活地更改代码以实现自己想要的功能。

现在，思考一个问题，主需求是实现博客中用户的注册登录登出功能。如果只满足于用户注册时只登记其邮箱或是用户名，Django 自带的 User 模块既可以实现。但实际上，一个普遍的要求是注册用户应该能够修改自己的头像信息，邮箱信息，昵称信息等其他更灵活的需求。

可以先看一下 Django User 模块的源码

class User(AbstractUser):
    """
    Users within the Django authentication system are represented by this
    model.

    Username, password and email are required. Other fields are optional.
    """
    class Meta(AbstractUser.Meta):
        swappable = 'AUTH_USER_MODEL'

注意：如果你的是 Anaconda 管理，可以在路径 C:\Users\User\Anaconda3\Lib\site-packages\django\contrib\auth\models.py 查看

Django 中的 User 模块实际上继承了 AbstractUser 模块，AbstractUser 模块下有：

username
first_name
last_name
email
date_joined
…

你可以看出，User 模块继承了 AbstractUser 抽象基类，而仅仅只是继承了，并没有对 AbstractUser 进行任何扩展。所以，对于一个需要更多需求的 User 模块信息来说，我们可以继承 AbstractUser 并根据自己的需求进行扩展。

现在，我们对用户属性添加一些需求，比如支持用户修改头像、支持用户昵称、qq、wechat 以及网站链接等。

class User(AbstractUser):
    nickname = models.CharField(max_length=30, blank=True, null=True, verbose_name='昵称')
    qq = models.CharField(max_length=20, blank=True, null=True, verbose_name='QQ号码')
    url = models.URLField(max_length=100, blank=True, null=True, verbose_name='个人网页地址')
    avatar = ProcessedImageField(upload_to='avatar',
default='avatar/default.png', verbose_name='头像')

    class Meta:
        verbose_name = '用户'
        verbose_name_plural = verbose_name
        ordering = ['-id']

    def __str__(self):
        return self.username

我们给自定义的用户模块增加 nickname（昵称）， qq, url（网站链接），avatar（头像）属性。

注意：为了让 Django 能够识别使用自定义的用户模型，必须要在 settings.py 中设置自定义模块位置，如在 settings.py 上添加

AUTH_USER_MODEL = 'blog.user'

其中，blog 为你对应的应用 app 信息，user 为 blog 应用下的 user 模块，在这里 blog 和 user 大小写无关。

如果在你现在执行数据库迁移命令，可能会出现 blog 不存在 user 模块 的提示，而无法重新进行数据迁移。

ValueError: The field account.EmailAddress.user was declared with a lazy reference to 'blog.user', but app 'blog' doesn't provide model 'user'.
The field admin.LogEntry.user was declared with a lazy reference to 'blog.user', but app 'blog' doesn't provide model 'user'.
The field blog.Article.author was declared with a lazy reference to 'blog.user', but app 'blog' doesn't provide model 'user'.
The field easy_comment.Comment.user was declared with a lazy reference to 'blog.user', but app 'blog' doesn't provide model 'user'.
The field easy_comment.Like.user was declared with a lazy reference to 'blog.user', but app 'blog' doesn't provide model 'user'.
The field notifications.Notification.recipient was declared with a lazy reference to 'blog.user', but app 'blog' doesn't provide model 'user'.
The field online_status.OnlineStatus.user was declared with a lazy reference to 'blog.user', but app 'blog' doesn't provide model 'user'.
The field socialaccount.SocialAccount.user was declared with a lazy reference to 'blog.user', but app 'blog' doesn't provide model 'user'.

所以，如果之前是使用例如 AUTH_USER_MODEL = auth.user 的用户模型，并重新将其自定义为 AUTH_USER_MODEL = blog.user 请删掉 migrations 目录下的所有文件以及数据库文件。

删除之后，重新进行数据库的迁移

$ python manage.py makemigrations myapp
$ python manage.py migrate

这个时候，所使用的用户即为自定义后的用户了。

  File "C:\Users\Micky\Anaconda3\lib\site-packages\django\db\backends\utils.py", line 85, in _execute
    return self.cursor.execute(sql, params)
  File "C:\Users\Micky\Anaconda3\lib\site-packages\django\db\backends\sqlite3\base.py", line 303, in execute
    return Database.Cursor.execute(self, query, params)
django.db.utils.OperationalError: no such table: blog_user

这里可以在模板中指定数据库 db_table = 'user'

Django User 模块之 AbstractUser 扩展

猜你喜欢