querysetでextraを使って新たに定義したものについてorder_byする方法

2019年12月31日2020年1月3日

今回はDjangoのquerysetでextraを使って新たに定義したものについてorder_byする方法についてご紹介いたします。

querysetでextraを使って新たに定義したものについてorder_byする方法

最初に結論を言ってしまうと、以下のソースを使うことでquerysetでextraを使って新たに定義したものについて昇順で整列させることができます。

ratings_tuples = [(r.id, 'extraで新たに定義した変数') for r in instances]
ratings_list = sorted(ratings_tuples, key=lambda x: x[1])
pk_list = [idx for idx, rating in ratings_list]
from django.db.models import Case, When
preserved = Case(*[When(pk=pk, then=pos) for pos, pk in enumerate(pk_list)])
instances = instances.filter(pk__in=pk_list).order_by(preserved)

ratings_tuples = [(r.id, 'extraで新たに定義した変数') for r in instances]

ratings_list = sorted(ratings_tuples, key=lambda x: x[1])

pk_list = [idx for idx, rating in ratings_list]

from django.db.models import Case, When

preserved = Case(*[When(pk=pk, then=pos) for pos, pk in enumerate(pk_list)])

instances = instances.filter(pk__in=pk_list).order_by(preserved)

例として次のBookモデルがあるとします。

from django.db import models

# Create your models here.
class Book(models.Model):
    title = models.CharField(max_length=120) # max_length=required
    author = models.CharField(max_length=120) # max_length=required
    price = models.DecimalField(decimal_places=2, max_digits=100)

from django.db import models

# Create your models here.

class Book(models.Model):

title = models.CharField(max_length=120) # max_length=required

author = models.CharField(max_length=120) # max_length=required

price = models.DecimalField(decimal_places=2, max_digits=100)

今Bookオブジェクトは３つあり、それぞれ保持する値は次の通りだとします。

>>> instances = Book.objects.all()
>>> for a in instances:
...     a.title
...     a.author
...     a.price
... 
'title1'
'author1'
Decimal('400.00')
'title2'
'author2'
Decimal('100.00')
'title3'
'author3'
Decimal('500.00')

>>> instances = Book.objects.all()

>>> for a in instances:

... a.title

... a.author

... a.price

...

'title1'

'author1'

Decimal('400.00')

'title2'

'author2'

Decimal('100.00')

'title3'

'author3'

Decimal('500.00')

Bookモデルに関して、extraを使って新たに税込みの値段を表す’taxPrice’を定義するとします。
extraを使ってpriceに10パーセントの税率をかけたtaxPriceを新たに定義します。

Book.objects.extra(select={'taxPrice': 'price * 1.1'})

1	Book.objects.extra(select={'taxPrice': 'price * 1.1'})

このtaxPriceでソートをかけたいとします。（taxPriceを使わずともそのままpriceでソートをかければ同じ結果になりますが、あくまでも例です。）

冒頭のソースをいじって、次のソースを準備します。

ratings_tuples = [(r.id, r.taxPrice) for r in instances]
ratings_list = sorted(ratings_tuples, key=lambda x: x[1])
pk_list = [idx for idx, rating in ratings_list]
from django.db.models import Case, When
preserved = Case(*[When(pk=pk, then=pos) for pos, pk in enumerate(pk_list)])
instances = instances.filter(pk__in=pk_list).order_by(preserved)

ratings_tuples = [(r.id, r.taxPrice) for r in instances]

ratings_list = sorted(ratings_tuples, key=lambda x: x[1])

pk_list = [idx for idx, rating in ratings_list]

from django.db.models import Case, When

preserved = Case(*[When(pk=pk, then=pos) for pos, pk in enumerate(pk_list)])

instances = instances.filter(pk__in=pk_list).order_by(preserved)

上記ソースを実行すると次のようになります。

>>> instances =  Book.objects.extra(select={'taxPrice': 'price * 1.1'})
>>> ratings_tuples = [(r.id, r.taxPrice) for r in instances]
>>> ratings_list = sorted(ratings_tuples, key=lambda x: x[1])
>>> pk_list = [idx for idx, rating in ratings_list]
>>> from django.db.models import Case, When
>>> preserved = Case(*[When(pk=pk, then=pos) for pos, pk in enumerate(pk_list)]) 
>>> instances = instances.filter(pk__in=pk_list).order_by(preserved)
>>> instances
<QuerySet [<Book: Book object (2)>, <Book: Book object (1)>, <Book: Book object (3)>]>
>>> for a in instances:
...     a.taxPrice
... 
110.00000000000001
440.00000000000006
550.0