Skip to main content
 首页 » 编程设计

python之如何使用 Django 1.7 用户模型处理多种用户类型

2025年05月04日24jackei

我是 Python 和 Django 的新手。我正在尝试为代理、经纪人、公司和客户等不同角色设置用户模型。这些角色中的每一个都可以作为用户注册到站点。然后代理人和经纪人可以拥有面向公众的个人资料页面。

我必须使用自定义用户模型还是内置用户模型才行?我有代理人和经纪人的其他属性,例如许可证、位置、语言、照片等。

class MyUser(AbstractBaseUser): 
    AGENTS = 'AG' 
    BROKERS = 'BR' 
    COMPANY = 'CP' 
    CUSTOMER = 'CM' 
 
    ROLE_IN_CHOICES = ( 
        (AGENTS, 'Agent'), 
        (BROKERS, 'Broker'), 
        (COMPANY, 'Company'), 
        (CUSTOMER, 'Customer')) 
 
    first_name = models.CharField(max_length=100, blank=True) 
    second_name = models.CharField(max_length=100, blank=True) 
    middle_name = models.CharField(max_length=100, blank=True) 
    dob = models.DateField(blank=True, null=True) 
    phone = models.CharField(max_length=10) 
    secondary_phone = models.CharField(max_length=10, blank=True, null=True) 

……

@property 
def is_agent(self): 
    return self.role in (self.AGENTS) 
 
@property 
def is_customer(self): 
    return self.role in (self.CUSTOMER) 
 
@property 
def is_broker(self):* 
    return self.role in (self.BROKER) 
 
@property 
def is_company(self): 
    return self.role in (self.COMPANY) 

....

我可以不使用基本用户模型并实现相同吗?我在写作轨道上吗?

如何为这些角色(代理、经纪人)创建面向公众的页面?

这是我第一次尝试使用 Python 和 Django。我正在使用 Django 1.7.7 和 Python 3.4

请您参考如下方法:

您应该改为从 Django 用户模型扩展并添加您需要的额外字段:

from django.contrib.auth.models import User as Auth_User 
 
class User(Auth_User): 
 
    # add your extra fields here like roles, etc 
    phone = CharField(max_length=20, null=True, blank=True) 
 
    # add your extra functions 
    def extra_user_function(self): 
         return "This is an extra function" 

这样你就有了自己的字段和 Django 用户字段...

迁移后,如果您检查数据库,您将拥有 auth_user 和 your_app_user 表。

请记住,request.user 只会给你 super 字段......为了获得你需要的扩展类

User.objects.get(id=request.user.id) 

最新的将只有额外的字段,它的 id 将与 auth.User 相同如果您不自己添加任何 auth_user。

注意:这很重要!

否则 request.user.id 和 your_app.User.id 不匹配,因此 User.objects.get(id=request.user.id) 获胜不起作用,您必须查询数据库才能找到 your_app.User.id

User.object.get(user_ptr_id = request.user.id) 

其他需要考虑的事情

这会起作用:

# you_app.User objects gets vars from auth.User 
user = User.objects.get(id=request.user.id) 
first_name = user.first_name 

但这行不通

# auth.User trying to get a function from your_app.User 
user =  request.user 
user.extra_user_function() 

所以用户模型可能是这样的:

import os 
from django.contrib.auth.models import User as Django_User 
from django.db.models import CharField, ImageField 
 
class User(Django_User): 
 
    phone = CharField(max_length=20, null=True, blank=True) 
    observations = CharField(max_length=2048, null=True, blank=True) 
    picture = ImageField(upload_to='users', default='default/avatar.jpg') 
 
    class Meta: 
        # adding extra permissions (default are: add_user, change_user, delete_user) 
        permissions = ( 
            ("access_user_list", "Can access user list"), 
            ("access_user", "Can access user"), 
        ) 
        ordering = ["-is_staff", "first_name"] 

Django 没有为用户创建角色,而是已经有了组,因此您应该使用它们。

这些组遵循相同的逻辑:

from django.contrib.auth.models import Group as Auth_Group 
from django.db import models 
 
class Group(Auth_Group): 
 
    observations = models.CharField(max_length=2048, null=True, blank=True) 
 
    def get_users_in_group(self): 
        return self.user_set.filter(is_active=1).order_by('first_name', 'last_name') 
 
    def count_users_in_group(self): 
        return self.user_set.count() 
 
    def __unicode__(self): 
        return self.name 
 
    class Meta: 
        permissions = ( 
             ("access_group_list", "Can access group list"), 
             ("access_group", "Can access group"), 
        ) 
        ordering = ["name"] 

您可以清除/添加用户到组:

user.groups.clear() 
user.groups.add(group) 

清除/添加组权限:

group.permissions.clear() 
group.permissions.add(permission) 

还有一个装饰器用来检查用户是否有权限

from django.contrib.auth.decorators import permission_required 
 
@permission_required(("users.change_user","users.access_user",)) 
def your_view(request): 
     ... 

我过去尝试过很多东西,但我想这是要走的路。

如果你真的需要角色,并且一个用户可以有多个角色,也许最好的办法是创建一个模型角色并将其添加到用户有一个 ManyToMany 字段

roles = ManyToManyField(Role) 

但是你可以用组来做