python - 将静态 CASE WHEN 语句转换为动态以提供给 .annotate()

Question

我有这个运行良好的 ORM 查询。我测试了以下内容以查看输出，它实际上符合预期:

members=Members.objects.all().annotate(age_groups=Case(
                When(birthdate__year__range=(2007,2017), then=Value('0-10')),
                When(birthdate__year__range=(1997,2006), then=Value('11-20')),
                When(birthdate__year__range=(1987,1996), then=Value('21-30')),
                When(birthdate__year__range=(1977,1986), then=Value('31-40')),
                When(birthdate__year__range=(1967,1976), then=Value('41-50')),
                When(birthdate__year__range=(1957,1966), then=Value('51-60')),
                When(birthdate__year__range=(1947,1956), then=Value('61-70')),
                When(birthdate__year__range=(1937,1946), then=Value('71-80')),
                When(birthdate__year__range=(1927,1936), then=Value('81-90')),
                When(birthdate__year__range=(1917,1926), then=Value('91-100')),
                default=Value('100+'),  output_field=CharField())
                ).values('age_groups',).annotate(total=Count('age_groups'))

但我想将其移动到一个函数中，以便用户可以指定他们想要使用的日期字段并进行他们喜欢的分组(例如 0-20、21-40 等)。为此，我编写了以下函数，它在调用时将 CASE WHEN 语句 简单地输出为字符串。问题是当我将它传递给 annotate() 时，我得到 str has no resolve error。

希望你能得到我想做的事。我需要将以下函数的输出传递给 annoate():

from datetime import date
def AgeGrouping(field_name,required_min_age=0,required_max_age=100,jump_by=10):
now_year=date.today().year

reply='Case('
just_started=True
for i in range(now_year-required_min_age,now_year-required_max_age,-jump_by):

    if just_started==False:
        min_age=max_age + 1
        max_age=min_age + 9

    else:
        min_age= now_year - i
        max_age= min_age + 10

    min_year=now_year- min_age
    max_year= now_year - max_age

    reply="\n".join([reply,"When(" + field_name + "__year__range=(" + str(max_year) + "," + str(min_year) + "), then=Value('" + str(min_age) + "-" + str(max_age) + "')),"])

    just_started=False

reply="\n".join([reply, "default=Value('100+'),  output_field=models.CharField())"])

return reply

然后这样调用它:

   print(AgeGrouping('birthdate',required_min_age=0,required_max_age=100,jump_by=10))

Answer 1

您当前的方法是创建一个包含所需代码的字符串。可以将字符串作为代码求值，但这可能存在安全问题。

更好的方法是使用循环创建 When() 对象列表。然后，您可以在注释中调用 Case() 时解压缩列表。

def age_grouping(field_name, required_min_age=0, required_max_age=100, jump_by=10):
    ...
    whens = []
    for i in range(now_year-required_min_age,now_year-required_max_age,-jump_by):
       # Set min_age, max_age, min_year, max_year
       ...
       description = '%s-%s' % (min_age, max_age)  # e.g. '0-10'
       # create dict of kwargs for When()
       kwargs = {'%s__year__range' % field_name: (min_year, max_year)}
       whens.append(When(then=Value(description), **kwargs))

    # Return the Case() object that you can pass to annotate()
    return Case(*whens, default=Value('100+'), output_field=CharField())

然后你应该可以调用

Members.objects.all().annotate(age_groups=age_grouping(...))