Фильтрация по пользовательскому полю UUIDField нарушена при обновлении Django 3.2

У меня есть проект Django, который я недавно обновил с Django 2.2 до 3.2. В этом проекте я использую пользовательский UUIDField, который сохраняет UUID в MySQL как char(36) в следующем формате: 12345678-1234-5678-1234-567812345678.

import uuid

from django.db import models


class UUIDField(models.UUIDField):
    """
    Overrides Django UUIDField to store full UUID's including dashes.
    """
    def __init__(self, verbose_name=None, **kwargs):
        super().__init__(verbose_name, **kwargs)
        self.max_length = 36

    def get_internal_type(self):
        return "CharField"

    def get_db_prep_value(self, value, connection, prepared=False):
        if value is None:
            return None
        if not isinstance(value, uuid.UUID):
            try:
                value = uuid.UUID(value)
            except AttributeError:
                raise TypeError(self.error_messages['invalid'] % {'value': value})

        if connection.features.has_native_uuid_field:
            return value
        return str(value)

После обновления я заметил, что поиск по полному UUID больше не работает. Если я предоставляю только первую часть UUID (до первого символа после первого дефиса), все работает как ожидалось.

Python 3.6.9 (default, Mar 15 2022, 13:55:28)
[GCC 8.4.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
(InteractiveConsole)
>>> from foobar.foo.models import Foo
>>>
>>> Foo.objects.all()
<QuerySet [<Foo: Foo object (34c46fe8-caf0-11ec-bdb9-482ae362a4c0)>]>
>>>
>>> Foo.objects.filter(id__icontains='34c46fe8-caf0-11ec-bdb9-482ae362a4c0')
<QuerySet []>
>>>
>>> Foo.objects.filter(id__icontains='34c46fe8-')
<QuerySet [<Foo: Foo object (34c46fe8-caf0-11ec-bdb9-482ae362a4c0)>]>
>>>
>>> Foo.objects.filter(id__icontains='34c46fe8-c')
<QuerySet []>
>>>

Я играл с методами UUIDField, но не могу понять, что пошло не так. Вот ссылка на Gist с использованием упрощенной модели, откуда я взял вышеприведенный пример оболочки.

После довольно долгого pdb сеанса отладки мне удалось найти, в чем проблема. Я надеялся найти по пути необработанные фрагменты SQL, но вместо этого мое внимание привлек объект под названием WhereNode.

> /home/milanb/temp/django_custom_uuid_field/django_playground/lib/python3.6/site-packages/django/db/models/sql/query.py(1399)build_filter()
-> return clause, used_joins if not require_outer else ()
(Pdb) n
--Return--
> /home/milanb/temp/django_custom_uuid_field/django_playground/lib/python3.6/site-packages/django/db/models/sql/query.py(1399)build_filter()->(<WhereNode: (...f9c4f880be0>)>, {'foo_foo'})
-> return clause, used_joins if not require_outer else ()
(Pdb) retval
(<WhereNode: (AND: <django.db.models.lookups.UUIDIContains object at 0x7f9c4f880be0>)>, {'foo_foo'})
(Pdb) pp locals()
{'__return__': (<WhereNode: (AND: <django.db.models.lookups.UUIDIContains object at 0x7f9c4f880be0>)>,
                {'foo_foo'}),
 'alias': 'foo_foo',
 'allow_joins': True,
 'allow_many': True,
 'arg': 'id__icontains',
 'branch_negated': False,
 'can_reuse': {'foo_foo'},
 'check_filterable': True,
 'clause': <WhereNode: (AND: <django.db.models.lookups.UUIDIContains object at 0x7f9c4f880be0>)>,
 'col': Col(foo_foo, foo.Foo.id),
 'condition': <django.db.models.lookups.UUIDIContains object at 0x7f9c4f880be0>,
 'current_negated': False,
 'filter_expr': ('id__icontains', '34c46fe8-caf0-11ec-bdb9-482ae362a4c0'),
 'join_info': JoinInfo(final_field=<foobar.foo.fields.UUIDField: id>, targets=(<foobar.foo.fields.UUIDField: id>,), opts=<Options for Foo>, joins=['foo_foo'], path=[], transform_function=<function Query.setup_joins.<locals>.final_transformer at 0x7f9c4f89ca60>),
 'join_list': ['foo_foo'],
 'lookup_type': 'icontains',
 'lookups': ['icontains'],
 'opts': <Options for Foo>,
 'parts': ['id'],
 'pre_joins': {},
 'reffed_expression': False,
 'require_outer': False,
 'reuse_with_filtered_relation': False,
 'self': <django.db.models.sql.query.Query object at 0x7f9c4f902c18>,
 'split_subq': True,
 'targets': (<foobar.foo.fields.UUIDField: id>,),
 'used_joins': {'foo_foo'},
 'value': '34c46fe8-caf0-11ec-bdb9-482ae362a4c0'}
(Pdb) pp clause
<WhereNode: (AND: <django.db.models.lookups.UUIDIContains object at 0x7f9c4f880be0>)>

Здесь я заметил эту конкретную строку:

 'clause': <WhereNode: (AND: <django.db.models.lookups.UUIDIContains object at 0x7f9c4f880be0>)>,

Я не знал, что такое объект django.db.models.lookups.UUIDIContains, поэтому сразу обратился к исходному коду Django, чтобы узнать. Это пустой класс, состоящий из двух миксинов IContains и UUIDTextMixin. Последний был виновником в моем случае, так как он удалял все дефисы из значения поиска. Это было введено в качестве исправления после версии 2.2, которую я использовал до обновления, для поддержки бэкендов баз данных без собственного типа UUID (например, MySQL 5.7 в моем случае).

Исправление было очень простым, мне просто нужно было зарегистрировать пользовательский класс UUIDIContains без UUIDTextMixin для моего пользовательского класса UUIDField.

Мой новый fields.py: 

import uuid

from django.db import models
from django.db.models.lookups import IContains


class UUIDField(models.UUIDField):
    """
    Overrides Django UUIDField to store full UUID's including dashes.
    """
    def __init__(self, verbose_name=None, **kwargs):
        super().__init__(verbose_name, **kwargs)
        self.max_length = 36

    def get_internal_type(self):
        return "CharField"

    def get_db_prep_value(self, value, connection, prepared=False):
        if value is None:
            return None
        if not isinstance(value, uuid.UUID):
            try:
                value = uuid.UUID(value)
            except AttributeError:
                raise TypeError(self.error_messages['invalid'] % {'value': value})

        if connection.features.has_native_uuid_field:
            return value
        return str(value)


@UUIDField.register_lookup
class UUIDIContains(IContains):
    pass
Вернуться на верх