自定义 Django 模型字段的字段查找功能

Question

我有一个自定义模型字段 YearWithSurenessField，它在 python 中由自定义数据类型 YearWithSureness 表示。 YearWithSureness 的构造函数为 YearWithSureness(year='', is_certain=False)，其中 year 为 '' 或四位数年份（作为字符串），并且 is_certain 是一个布尔值，表示我是否确定给定年份是正确的。这种类型的模型字段以year/is_certain的形式存储在我的数据库中，例如“2008/True”、“2011/False”、“/False”等。

例如，在Member 模型，我有一个字段 grad_year = YearWithSurenessField(...)，它存储会员的毕业年份以及我是否确定我存储的年份是正确的。

我希望能够做的是使用类似的方法

Member.objects.filter(grad_year__year=2011)

来获取 grad_year 为“2011/True”或“2011/False”的所有成员的 QuerySet 。同样，我希望能够使用类似的方法

Member.objects.filter(grad_year__range=(2000, 2011))

来获取 grad_year 在 2000 至 2011 范围内的所有成员的 QuerySet，无论 grad_year.is_certain 是否是真是假。

这可能吗？我知道我可以使用 Member.objects.filter(grad_year__contains="2011") 来获得第一个结果，但我希望能够使用 __year。

这是相关的类，删除了无关的代码：

class YearWithSureness(object):
    def __init__(self, year='', is_certain=False):
        # ...

    def __str__(self):
        return "{year}/{is_certain}".format(year=self.year,
                                            is_certain=self.is_certain)

class YearWithSurenessField(models.Field):
    __metaclass__ = models.SubfieldBase

    def __init__(self, *args, **kwargs):
        # ...

    def to_python(self, value):
        # ...

    def get_prep_value(self, value):
        # ...

    def get_prep_lookup(self, lookup_type, value):
        if lookup_type in ('month', 'day'):
            raise TypeError('Lookup type {0} not supported.'.format(lookup_type))
        else:
            return super(YearWithSurenessField, self).get_prep_lookup(lookup_type, value)

    def value_to_string(self, obj):
        # ...

Answer 1

这是我发现有用的东西：
在 Django 中创建自定义字段查找

更灵活的方法是编写自定义查询集以及自定义管理器。使用 ozan 的代码：

class PersonQuerySet(models.query.QuerySet):
    def in_age_range(self, min, max):
        return self.filter(age__gte=min, age__lt=max)

class PersonManager(models.Manager):
    def get_query_set(self):
         return PersonQuerySet(self.model)

    def __getattr__(self, name):
        return getattr(self.get_query_set(), name)

class Person(models.Model):
    age = #...
    objects = PersonManager()

这允许您链接自定义查询。所以这两个查询都是有效的：

Person.objects.in_age_range(20,30)
Person.objects.exclude(somefield = some_value).in_age_range(20, 30)

Answer 2

我不明白为什么你需要这样一个自定义字段。据我所知，“year”和“is_certain”非常适合存储在两个独立的字段中。通过这样做，首先，按年份或年份范围进行搜索会更容易。其次，搜索的效率也会显着提高，尤其是当数据量很大时。最后但并非最不重要的一点是，您不再需要费心如何正确实现自定义字段。

因此，我建议您解释为什么需要将这两种自然不同类型的数据存储到数据库表中的单个列中的根本原因。也许我们可以指出一种更简单的方法来实现您的真正目标。

Answer 3

您是否尝试过更改 get_prep_lookup 的行为，以在 Lookup_type == 'year' 时仅返回年份值？您可以返回 int(value.split('/')[0])

我不确定实现这样的自定义字段是最佳选择，是否有充分的理由避免将值拆分为两个单独的字段？