如何使用Django 1.7用户模型处理多个用户类型

时间:2015-03-23 22:34:14

标签: python django

我是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

1 个答案:

答案 0 :(得分:1)

您应该从Django User模型扩展,并添加您需要的额外字段:

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只会为您提供超级字段...为了获得您需要的扩展课程

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

如果您不添加任何auth_user ,则最新版本只会包含额外字段,其ID将与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)

但你可以用群组

来做到这一点