模型选择类

Question

如何在 Django 中使用类来选择模型？

这是我想要的：

class ChoicesCls(object):
    def __init__(self):
        self.OPTION_ONE = 'O1'
        self.OPTION_ONE_SHORT = 'Opt 1'
        self.OPTION_ONE_LONG = 'Option 1'
        self.OPTION_TWO = 'O2'
        self.OPTION_TWO_SHORT = 'Opt 2'
        self.OPTION_TWO_LONG = 'Option 2'

    def get_choices(self):
        return (
            (self.OPTION_ONE, self.OPTION_ONE_LONG),
            (self.OPTION_TWO, self.OPTION_TWO_LONG)
        )


class TestModel(models.Model):
    choices_obj = ChoicesCls()
    field = models.CharField(choices=choices_obj.get_choices)

我想要这样的主要原因是在另一个地方我可以做这样的事情

option = "ONE"
getattr(TestModel.choices_obj, "OPTION_{}_SHORT".format(option))

Answer 1

试试这样的

FakeChoices 类： def __init__(self): self.OPTION_ONE = 'O1' self.OPTION_ONE_SHORT = '选择 1' self.OPTION_ONE_LONG = '选项 1' self.OPTION_TWO = 'O2' self.OPTION_TWO_SHORT = '选择 2' self.OPTION_TWO_LONG = '选项 2' def __get_tuple(self): 返回 （ (self.OPTION_ONE, self.OPTION_ONE_LONG), （self.OPTION_TWO，self.OPTION_TWO_LONG） ) def __getitem__(self, key): 返回 self.__get_tuple()[key] def __iter__(self): 返回迭代器（self.__get_tuple（）） ... 类测试模型（模型。模型）： 字段 = models.CharField(choices=FakeChoices())

正如您可能想象的那样，您可以将各种疯狂的逻辑放入 FakeChoices 类中。你应该能够做到

选项=“一个” getattr(TestModel._meta.get_field('field').choices, "OPTION_{}_SHORT".format(option))

Answer 2

主要问题是你试图在类实例之外初始化一些东西，所以你可以这样做

field = models.CharField(choices=ChoicesCls().get_choices())

但这仍然不是很有用（在我看来）

我发现从模块中导入一个元组列表并将其用作选择会更好，不需要为此生成一个类。

这有一个额外的好处，它允许你做你想做的第二件事，因为你也在模块级别定义了选项

OPTION_ONE_WITH_A_GOOD_NAME = '1'
OPTION_TWO_WITH_A_GOOD_NAME = '2'

MY_GOOD_NAME_CHOICES = [(OPTION_ONE_WITH_A_GOOD_NAME,OPTION_ONE_WITH_A_GOOD_NAME),
                        (OPTION_TWO_WITH_A_GOOD_NAME,OPTION_TWO_WITH_A_GOOD_NAME)]

Answer 3

要扩展Mad Wombat's answer，您可以创建一个自定义字段：

class ChoicesCharField(CharField):
    def contribute_to_class(self, cls, name, virtual_only=False):
        super().contribute_to_class(cls, name, virtual_only)

        def get_FOO_display_short(s):
            value = getattr(s, self.name)
            return self.choices.get_short_name(value)
        get_FOO_display_short.__name__ = 'get_{}_display_short'.format(self.name)
        get_FOO_display_short.short_description = self.verbose_name
        get_FOO_display_short.admin_order_field = self.name

        setattr(cls, get_FOO_display_short.__name__, get_FOO_display_short)

这样您就可以使用my_model_instance.get_field_display_short()，就像您通常使用my_model_instance.get_field_display() 一样。