zoukankan      html  css  js  c++  java
  • django ORM排序

    I thought it would be an easy one, but found myself lost with 34 opened tabs on stackoverflow...

    The problem : keep it ordered

    Usually, obtaining a QuerySet from a list is quite simple :

    >>> queryset = Theme.objects.filter(pk__in=[1, 2, 10])
    >>> type(queryset)
    <class 'django.db.models.query.QuerySet'>
    >>> queryset
    [<Theme: Fauna>, <Theme: Flora>, <Theme: Refuge>]
    

    The problem is that the list order is ignored :

    >>> Theme.objects.filter(pk__in=[10, 2, 1])
    [<Theme: Fauna>, <Theme: Flora>, <Theme: Refuge>]
    

    If obtaining a QuerySet is not a requirement, it's rather easy to get a list sorted according to another :

    pks_list = [10, 2, 1]
    themes = list(Theme.objects.filter(pk__in=pks_list))
    themes.sort(key=lambda t: pks_list.index(t.pk))
    

    In my case, I want a QuerySet, a brave lazy one, with proper filter()exclude()values() ...

    Fallback to SQL

    AFAIK, most database engines ignore order of records, until you specify an ordering column. In our case, the list is arbitrary, and does not map to any existing attribute, thus db column.

    If you use MySQL (who does?!), there is a FIELD() function that provides custom input for the sort method :

    SELECT *
    FROM theme
    ORDER BY FIELD(`id`, 10, 2, 1);
    

    Using the ORM, it gives us (thanks Daniel Roseman)

    pk_list = [10, 2, 1]
    ordering = 'FIELD(`id`, %s)' % ','.join(str(id) for id in pk_list)
    queryset = Theme.objects.filter(pk__in=[pk_list]).extra(
               select={'ordering': ordering}, order_by=('ordering',))
    

    Well, good news it can be ported to PostgreSQL. But if possible, I would prefer native SQL.

    And it looks like the magnificient syntax of SQL provides ORDER BY CASE WHEN ... END !

    SELECT *
    FROM theme
    ORDER BY
      CASE
        WHEN id=10 THEN 0
        WHEN id=2 THEN 1
        WHEN id=1 THEN 2
      END;
    

    Using the ORM, it gives us :

    pk_list = [10, 2, 1]
    clauses = ' '.join(['WHEN id=%s THEN %s' % (pk, i) for i, pk in enumerate(pk_list)])
    ordering = 'CASE %s END' % clauses
    queryset = Theme.objects.filter(pk__in=pk_list).extra(
               select={'ordering': ordering}, order_by=('ordering',))
    

    I wonder how it behaves with zillions of records though ;)

    One more thing: before Django 1.6, there was a bug with calling values_list() on a queryset ordered by an extra column. Use this :

    values = queryset.values('ordering', 'label')
    labels = [value['label'] for value in values]
    

    Good luck ! Please share your advices or critics ;)

    转发自:http://blog.mathieu-leplatre.info/django-create-a-queryset-from-a-list-preserving-order.html

  • 相关阅读:
    博客园样式设置
    最坏情况为线性时间的选择算法
    棋盘覆盖
    矩阵乘法的Strassen算法及时间复杂度
    大整数乘法及算法时间复杂度
    全排列问题的递归算法(Perm)
    python的lambda
    python的zip函数
    python操作队列
    mysql基础命令
  • 原文地址:https://www.cnblogs.com/bara/p/8022044.html
Copyright © 2011-2022 走看看