Django Extending the Existing User model

Posted on  Mar 27, 2017  in  Django  by  Amo Chen  ‐ 1 min read

Django 已經是一套相當成熟的 Web 框架,該有的功能也都一應俱全,但有些時候不免需要進行客製化。例如 Django 內建的 User model 雖已堪用,但難免需要額外增加欄位或修改 Primary Key ,本篇將介紹如何擴充 Django 既有的 User model 。

執行環境

本文的執行環境:

  • Python 3
  • Django 1.10.6

新增共用的 Django App

一般共用的元件,我都會放在一個稱為 core 的 app 中集中管理:

$ ./manage.py startapp core

繼承 AbstractUser

擴充既有的 User model 很簡單,只要繼承 AbstractUser 即可,例如以下為修改 Primary Key 的範例:

# core/models.py
import uuid

from django.db.models import UUIDField
from django.contrib.auth.models import AbstractUser

class MyUser(AbstractUser):
    id = UUIDField(primary_key=True, default=uuid.uuid4, editable=False)

修改 settings.py

擴充完既有的 User model 之後必須於 settings.py 中增加一個設定 AUTH_USER_MODEL ,將認證的 model 改為新的 User model 。

例如改為 core app 底下的 MyUser ,同時把 core app 加到 INSTALLED_APPS 中:

# settings.py
AUTH_USER_MODEL = 'core.MyUser'

INSTALLED_APPS = [
    # skipped
    'core',
]

接著 migrate DB:

$ ./manage.py makemigrations core
$ ./manage.py migrate core

這樣就修改完畢了!

使用新的 User model

新的 User model 在使用時就不再是透過原有的 import 方式進行使用,若在其他 model 中需要用到的話,則需改用 settings.AUTH_USER_MODEL ,例如:

# mytestapp/models.py
from django.conf import settings
from django.db import models
from django.db.models import OneToOneField


class MyTestModel(models.Model):
    user = OneToOneField(settings.AUTH_USER_MODEL, on_delete=CASCADE)

此外, Django 也有提供 get_user_model() 來取得設定好的 User model ,但前提是必須讓 Django import 完所有 models 才能夠使用,否則就會出現 django.core.exceptions.AppRegistryNotReady: Models aren't loaded yet. 錯誤訊息。

以上本文結束!

對抗久坐職業傷害

研究指出每天增加 2 小時坐著的時間,會增加大腸癌、心臟疾病、肺癌的風險,也造成肩頸、腰背疼痛等常見問題。

然而對抗這些問題,卻只需要工作時定期休息跟伸展身體即可!

你想輕鬆改變現狀嗎?試試看我們的 PomodoRoll 番茄鐘吧! PomodoRoll 番茄鐘會根據你所設定的專注時間,定期建議你 1 項辦公族適用的伸展運動,幫助你打敗久坐所帶來的傷害!

贊助我們的創作

看完這篇文章了嗎? 休息一下,喝杯咖啡吧!

如果你覺得 MyApollo 有讓你獲得實用的資訊,希望能看到更多的技術分享,邀請你贊助我們一杯咖啡,讓我們有更多的動力與精力繼續提供高品質的文章,感謝你的支持!