Пользовательское поле заказа с помощью Django Rest Framework

У меня есть модель типа

class MyModel(models.Model):
    COLOR_CODES = ['RED', 'YELLOW', 'GREEN']
    name = models.CharField(db_column='name', max_length=200, blank=False, null=False, unique=True)
    colorCode = EnumField(db_column='color_code', choices=COLOR_CODES, null=False, default='GREEN')
    

    class Meta:
        managed = True
        db_table = 'MyModel'
        ordering = ['colorCode']

Я хочу упорядочить набор предметов по цветовому коду, но не как ЗЕЛЕНЫЙ, КРАСНЫЙ, ЖЕЛТЫЙ предметы. Я хочу сделать порядок как КРАСНЫЙ, ЖЕЛТЫЙ, ЗЕЛЕНЫЙ предметы.

Возможно ли это? Есть ли что-то похожее на Java Comparator? Заранее спасибо.

Вы могли бы использовать функцию Python sorted, но чтобы воспользоваться преимуществами запросов Django:

from django.db.models import Value, IntegerField, When, Case
# Don' t set the default order in the model (this also improves performance), simply write this query when you need this particular order
MyModel.objects.annotate(order=Case(When(colorCode='RED', then=0), When(colorCode='YELLOW', then=1), default=Value(2), output_field=IntegerField())).order_by('order')

Иначе вы могли бы изменить свою модель:

from enum import Enum

class MyModel(models.Model):
   class ColorCodes(Enum):
      # In the admin interface you will see 'red' and not '0r' (so don't worry about ugly names), the second value is stored in the database and is used for order the queryset
      red = ('red', '0r')
      yellow = ('yellow', '1y')
      green  = ('green', '2g')
      
      @classmethod
      def get_value(cls, member):
         return cls[member].value[0]
  

   colorCode = CharField(db_column='color_code', null=False, default=ColorCodes.get_value('green'), max_length=2, choices=[x.value for x in ColorCodes])
    
   class Meta:
      managed = True
      db_table = 'MyModel'
      ordering = ['colorCode']
Вернуться на верх