Django按顺序连接两个查询集

时间:2017-09-25 15:15:18

标签: python database django-queryset

所以我有一个带有选择字段的ModelForm。在此选择字段中,大约有100个条目。在这些条目中,我有许多条目,例如"组织科罗拉多 - 丹佛",我希望在列表的顶部有这样的条目,以便所有条目与"组织科罗拉多"位于列表的顶部,其他所有内容都按字典顺序排序。

我尝试过制作两个单独的查询集(这似乎是一个坏主意,但只需100个左右的条目即可管理)。似乎有很多方法可以组合这两个查询集,但不保持顺序(这是重点)。我试过这个:

class CreateContactForm(ModelForm):
    ...

    def __init__(self, *args, **kwargs):
        super(CreateContactForm, self).__init__(*args, **kwargs)
        p = models.ConstantContactList.objects.filter(
            name__startswith=settings.PREF_ORGANIZATION_PREFIX
        )
        np = models.ConstantContactList.objects.filter(
            name__regex=r'^(?!{})'.format(settings.PREF_ORGANIZATION_PREFIX)
        ).order_by('-name')

        self.fields['cc_lists'].queryset = list(p) + list(np)

这不起作用,虽然可能,如果有某种方法将该列表转换回查询集,或者是否有办法绕过查询集可能?我不确定。任何人都可以提供我应该做什么的线索吗?

2 个答案:

答案 0 :(得分:3)

我建议不要尝试对查询集进行排序,只是处理渲染层(模板或表单)中的排序。这样,如果您想要本地化代码,则无需更改查询。

假设您使用forms.Select小部件。您可能希望从此窗口小部件继承并覆盖render_menu逻辑,以便自己构建它并自己处理排序。您可以访问已呈现或未呈现的选项,因此从那时起它不应成为问题。

答案 1 :(得分:0)

行。我已经提出了一个适用于此处的解决方案,并且我已经将它提供给任何有相同需求的人。

from django.forms.widgets import Select
import re

class CustomOrderingWidget(Select):
    def __init__(self, priority_regex, sort='+', attrs=None):
        super(CustomOrderingWidget, self).__init__(attrs)
        self.regex = re.compile(priority_regex)
        self.sort = sort
        self.template_name = 'django/forms/widgets/select.html'

    def render(self, name, value, attrs=None, renderer=None):
        context = self.get_context(name, value, attrs)
        optgroups = context.get('widget').get('optgroups')
        firsts, others = [], []
        for grp in optgroups:
            if self.regex.search(grp[1][0].get('label')):
                firsts.append(grp)
            else:
                others.append(grp)
        if self.sort == '+':
            kfn = lambda x: x[1][0].get('label')
            context['widget']['optgroups'] = sorted(firsts, key=kfn) +\
                                               sorted(others, key=kfn)
        elif self.sort == '-':
            kfn = lambda x: x[1][0].get('label')
            context['widget']['optgroups'] =\
                sorted(firsts, key=kfn, reverse=True) +\
                sorted(others, key=kfn, reverse=True)
        else:
            context['widget']['optgroups'] = firsts + others
        return self._render(self.template_name, context, renderer)

然后你可以将它插入像这样的ModelForm ......

import settings # YOUR personal stuffz!

class CreateContactForm(ModelForm):
    ...
    class Meta:
       ...
        widgets = {
            # Just an example, make your own regex string!  
            'cc_lists': CustomOrderingWidget("^{0}".format(
                settings.PREF_ORGANIZATION_PREFIX
            ))
        }