用于模型选择的

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

主要问题是，您正在尝试初始化类实例之外的某些内容，以便您可以这样做

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 2

要扩展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()所做的那样。