关于继承先前模型的意见

时间:2018-06-17 20:18:11

标签: python django inheritance model

我正在查看我的代码并看到这两个模型有多个相似的字段。我想知道我的ParentProfile模型是否可以继承我的User模型以缩短代码并删除重叠。这是一个非常初学的问题,并希望有人就此事发表意见。谢谢!

models.py

class User(AbstractBaseUser):
    email = models.EmailField(max_length=255, unique=True)
    first_name = models.CharField(max_length=255)
    last_name = models.CharField(max_length=255)
    child_first_name = models.CharField(max_length=255)
    timestamp = models.DateTimeField(auto_now_add=True)
    student = models.BooleanField(default=False)
    parent = models.BooleanField(default=False)
    teacher = models.BooleanField(default=False)
    active = models.BooleanField(default=True) # can login
    staff = models.BooleanField(default=False) # staff user, not superuser
    admin = models.BooleanField(default=False) # superuser

    objects = UserManager()

    # takes email as username | removes email
    USERNAME_FIELD = 'email'
    REQUIRED_FIELDS = ['first_name', 'last_name']

    def __str__(self):
        return self.email

    def has_perm(self, perm, onj=None):
        "Does the user have a specific permission?"
        return True

    def has_module_perms(self, app_label):
        "Does the user have permissions to view the app 'app_label'?"
        return True

    @property
    def is_staff(self):
        return self.staff

    @property
    def is_admin(self):
        return self.admin

    @property
    def is_active(self):
        return self.active

class ParentProfile(models.Model):
    user = models.OneToOneField(User, on_delete=models.CASCADE)
    email = models.EmailField(max_length=255, unique=True)
    first_name = models.CharField(max_length=255)
    last_name = models.CharField(max_length=255)
    child_first_name = models.CharField(max_length=255)
    timestamp = models.DateTimeField(auto_now_add=True)
    student = models.BooleanField(default=False)
    parent = models.BooleanField(default=False)
    teacher = models.BooleanField(default=False)
    active = models.BooleanField(default=True) # can login
    staff = models.BooleanField(default=False) # staff user, not superuser
    admin = models.BooleanField(default=False) # superuser

    objects = UserManager()

    # takes email as username | removes email
    USERNAME_FIELD = 'email'
    REQUIRED_FIELDS = ['first_name', 'last_name', 'child_first_name']

    def __str__(self):
        return self.email

    def has_perm(self, perm, onj=None):
        "Does the user have a specific permission?"
        return True

    def has_module_perms(self, app_label):
        "Does the user have permissions to view the app 'app_label'?"
        return True

    @property
    def is_staff(self):
        return self.staff

    @property
    def is_admin(self):
        return self.admin

    @property
    def is_active(self):
        return self.active

1 个答案:

答案 0 :(得分:1)

您可以使用公共字段创建抽象模型,并在两者中继承

class CommonFieldModel(models.Model):
    email = models.EmailField(max_length=255, unique=True)
    first_name = models.CharField(max_length=255)
    last_name = models.CharField(max_length=255)
    child_first_name = models.CharField(max_length=255)
    timestamp = models.DateTimeField(auto_now_add=True)
    student = models.BooleanField(default=False)
    parent = models.BooleanField(default=False)
    teacher = models.BooleanField(default=False)
    active = models.BooleanField(default=True) # can login
    staff = models.BooleanField(default=False) # staff user, not superuser
    admin = models.BooleanField(default=False) # superuser

    class Meta:
        abstract = True

Meta类定义它是一个抽象模型,Django不会为此创建任何表。现在,您可以在其他模型中继承此模型。实施例

class ParentProfile(CommonFieldModel):
    user = models.OneToOneField(User, on_delete=models.CASCADE)
class User(CommonFieldModel, AbstractBaseUser):
    user = models.OneToOneField(User, on_delete=models.CASCADE)