Django将ForeinKey建模为抽象类

时间:2016-11-13 21:31:20

标签: django django-models

我正在制作简单的RPG浏览器游戏,我想这样做:

#Basic class
class AbstractClass(models.Model):
    health = models.IntegerField(default=10)
    mana = models.IntegerField(default=10)

我没有像

这样的专业课程
class WarriorClass(AbstractClass):
    strength = models.IntegerField(default=20)
    intelligence = models.IntegerField(default=10)

class MageClass(AbstractClass):
    strength = models.IntegerField(default=10)
    intelligence = models.IntegerField(default=20)

在UserProfile模型中

class UserProfile(models.Model):
    user = models.OneToOneField(
        User, on_delete=models.CASCADE, related_name='profile'
    )
    profession = #??? 

而且我不知道我应该在职业领域做些什么。我想要像ForeingKey这样的东西(但在创建新实例时我想指定哪个类(mage或warrior)应该是这个ForeignKey。

我该怎么做?或者也许你们有更好的想法做这样的迷你系统?

最佳

2 个答案:

答案 0 :(得分:1)

您可以使用文档中的GenericForeignKey

from django.db import models
from django.contrib.contenttypes.fields import GenericForeignKey
from django.contrib.contenttypes.models import ContentType

class TaggedItem(models.Model):
    tag = models.SlugField()
    content_type = models.ForeignKey(ContentType, on_delete=models.CASCADE)
    object_id = models.PositiveIntegerField()
    content_object = GenericForeignKey('content_type', 'object_id')

    def __str__(self):
        return self.tag

然后:

>>> from django.contrib.auth.models import User
>>> guido = User.objects.get(username='Guido')
>>> t = TaggedItem(content_object=guido, tag='bdfl')
>>> t.save()
>>> t.content_object
<User: Guido>

但是这个解决方案将来可能会出现问题。 更简单的解决方案

  1. 将职业作为选择领域
  2. 将你的法术力,力量等放入UserProfile模型
  3. 依赖于选择设置适当的值
  4. 您可以覆盖保存方法,如果用户选择Warrior设置强度为20等,

答案 1 :(得分:0)

在类似情况下检查this answer

基于此,您可以定义:

class UserProfile(models.Model):
user = models.OneToOneField(
    User, on_delete=models.CASCADE, related_name='profile'
)
profession = models.ForeignKey(WarriorClass)