python - 如何从 Django 用户模型中调用自定义数据

Question

我使用 django 的内置用户模型并添加了一些自定义字段，其中一个是用户类型，我想根据 e_type 设置条件语句，这是我的代码。

模型.py

from django.db import models
from django.core.validators import MinValueValidator, MaxValueValidator
from django.contrib import auth
from django.conf import settings

class User(auth.models.User,auth.models.PermissionsMixin,models.Model):
    EMPLOYEE='em'
    MANAGER='mn'
    USER_TYPE=[
        (EMPLOYEE,'employee'),
        (MANAGER,'manager'),
     ]
    e_type=models.CharField(max_length=2,choices=USER_TYPE,default=EMPLOYEE)
    no_of_leaves=models.IntegerField(null=False,validators=[MinValueValidator(1),
                                   MaxValueValidator(24)],default=24)

    def __str__(self):
        return "@{}".format(self.username)

视图.py

from django.shortcuts import render
from django.contrib.auth import views as auth_views
from .models import *
from django.views.generic import TemplateView


def home(request):
    user=request.user.e_type
    return render(request,"leaveApp/home.html")

标签： pythondjango