Django模型中外鍵的​​動態選擇

Question

我目前在為模型中的字段實現動態選擇時遇到一些困難。 有問題的模型如下。

class Item(models.Model):
    # Fields
    ...
    item_type = models.ForeignKey(Type, on_delete=models.PROTECT)

型號型號

class Type(models.Model):
    # Fields
    group = models.CharField(max_length=50, editable=False)
    name = models.CharField(max_length=30, help_text="Enter category name")
    ...

    # Methods
    def __str__(self):
        return str(self.name)

我想要限制此字段返回的Type的數量。 我的每個“ Type條目都有一個“組”字段，我只想返回與當前用戶所在的組具有相同組的“ Type條目。例如，創建Item時“ G1”組中的用戶只能在組字段中選擇帶有“ g1”的Type 。

我已經簽出limit_choices_to ，但是我不確定如何使用戶組打開表單，以便將其傳遞給limit_choices_to 。 如果有人可以提供建議或任何其他方法來解決此問題，將不勝感激。

Answer 1

您可以覆蓋默認對象管理器或添加自定義對象管理器。 在您的情況下將是這樣。

class TypesManager(models.Manager):
    def get_queryset(self, user_group):
        qs = super(TypesManager,self).get_queryset().filter(group_id=user_group.id)

class Type(models.Model):
    group = models.ForeignKey(Group, verbose_name='Group')
    name = models.CharField(max_length=30, help_text="Enter category name")

    # default_manager
    objects = TypesManager()

    # custom_manager
    custom_manager = TypesManager()

    # Methods
    def __str__(self):
        return str(self.name)'

您可以在這里閱讀有關Manager的更多信息https://docs.djangoproject.com/en/2.0/topics/db/managers/

或者，如果您只想限制特定表單的類型，則可以在表單init中過濾初始數據