Как отфильтровать с помощью enum или string и получить такой же результат?

Question

Мне трудно использовать Enums в Django.

Это моя заявка модель:

class RequestStatuses(Enum):
    new = 'new'
    sent = 'sent'
    done = 'done'


class Request(BaseModel):


    request_number =    models.PositiveIntegerField(default=0)
    type =              models.CharField(max_length=31, blank=True, null=True)

    status =            models.CharField(
        max_length=31,
        choices=[(a.name, a.value) for a in RequestStatuses],
        default=RequestStatuses.new
    )

    sensor =            models.ForeignKey(Sensor, on_delete=models.SET_NULL, blank=True, null=True)
    device =            models.ForeignKey(Device, on_delete=models.SET_NULL, blank=True, null=True)
    user =              models.ForeignKey(User, on_delete=models.SET_NULL, blank=True, null=True)
    payload =           models.TextField(blank=True, null=True)

Есть разница, когда я создаю запись со строкой или типом enum, что действительно раздражает ...

Это прекрасно работает:

    device = Device.objects.create(serial_number=1)
    request = Request(
        device=self.device,
        status=RequestStatuses.sent
    )
    request.save()
    try:
        request = device.request_set.filter(
            status=RequestStatuses.sent
        )[0]
    except IndexError:
        print(device.request_set.all()[0].status)
        pass

Но это исключение

    device = Device.objects.create(serial_number=1)
    request = Request(
        device=device,
        status='sent'
    )
    request.save()
    try:
        request = device.request_set.filter(
            status=RequestStatuses.sent
        )[0]
    except IndexError:
        print(device.request_set.all()[0].status)
        pass

Когда я пытаюсь фильтровать по status=RequestStatuses.sent.value или просто по sent строке, первый пример вызывает исключение, а второй работает.

Какой смысл перечислений, когда вы не можете отфильтровать их по строке или наоборот? Как я могу заставить его работать с API - который будет передавать строку в фильтр? Или это просто проблема с кешем?

Answer 1

Честно говоря, я не знаю причину и почему это можно сделать, но добавление функций __repr__ и __str__ в enum решило проблему.

class RequestStatuses(Enum):
    new = 'new'
    sent = 'sent'
    done = 'done'

    def __repr__(self):
        return self.name

    def __str__(self):
        return self.name