首页 > 解决方案 > Django:在views.py中导入我的自定义用户的最佳方法是什么?

问题描述

实际上我在models.py中创建了我自己的自定义用户(MyUser)

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

class MyUser(AbstractUser):
    country=models.CharField(max_length=20)

设置.py

AUTH_USER_MODEL = 'new_user.MyUser'

我在views.py 中创建了一个简单的函数来创建新用户。

from django.shortcuts import render,redirect
from django.http import HttpResponse
from django.contrib.auth import get_user_model
from django.conf import settings

def save_account(request):
    MyUser=get_user_model()
    if request.method == "POST":
        name=request.POST.get("name")
        email=request.POST.get("email")
        password1=request.POST.get("password1")
        password2=request.POST.get("confirm-password")
        country=request.POST.get("country")
        new_user=MyUser.objects.create_user(username=name,email=email,password=password1,country=country)
        new_user.save()
        return redirect('/')
    else:
        return HttpResponse("404 Not Found")

1.这里我使用 get_user_model() 函数来获取我当前的自定义用户。

2.我有第二个选择来获取我的自定义用户,只需从 models.py 导入 MyUser 模型,然后使用它。

from .models import MyUser

3.我有第三个选项来获取自定义用户,只需直接从 settings.py 导入AUTH_USER_MODEL并以这种方式使用它。

from django.conf import settings
MyUser=settings.AUTH_USER_MODEL

但是第三个选项返回一个字符串而不是返回我的自定义用户。但是为什么?

我只想知道哪个选项最好在views.py中导入自定义用户,为什么?

标签: pythondjangodjango-custom-user

解决方案


您的前两个选项都可以。Usingget_user_model可以更轻松地重用应用程序,而无需更改导入。然而,许多 Django 项目并不意味着被重用,在这种情况下,显式导入可以更清楚地从哪里导入用户模型。

settings.AUTH_USER_MODEL是一个字符串。你在做的时候将它设置为一个字符串AUTH_USER_MODEL = 'new_user.MyUser',当你访问它时,Django 不会自动将它更改为模型实例。它在外键中很有用,因为这意味着您不需要导入模型。例如:

class MyModel(models.Model):
    author = models.ForeignKey(
        settings.AUTH_USER_MODEL,
        on_delete=models.CASCADE,
    )

推荐阅读