Skip to content

Commit 7ed5b91

Browse files
committed
fix enum lookup
1 parent f45586c commit 7ed5b91

File tree

2 files changed

+2
-5
lines changed

2 files changed

+2
-5
lines changed

django_mongodb/features.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -85,9 +85,6 @@ class DatabaseFeatures(BaseDatabaseFeatures):
8585
# contains with expressions/subqueries doesn't work.
8686
"model_fields_.test_arrayfield.TestQuerying.test_contains_including_expression",
8787
"model_fields_.test_arrayfield.TestQuerying.test_contains_subquery",
88-
# Add FieldGetDbPrepValueMixin to lookups?
89-
# https://github.com/django/django/commit/5a36c81f58b8ff45d8dac052343722c54e3fa521#diff-10cd50a6d8761600a1aea96d413184582481a278388b4ad38a788030a21cbd45R5
90-
"model_fields_.test_arrayfield.TestQuerying.test_enum_lookup",
9188
# Unsupported conversion from array to string in $convert
9289
"model_fields_.test_arrayfield.TestQuerying.test_icontains",
9390
# Field 'field' expected a number but got Value(1).

django_mongodb/fields/array.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
from django.core import checks, exceptions
55
from django.db.models import DecimalField, Field, Func, IntegerField, Transform, Value
66
from django.db.models.fields.mixins import CheckFieldDefaultMixin
7-
from django.db.models.lookups import In, Lookup
7+
from django.db.models.lookups import FieldGetDbPrepValueMixin, In, Lookup
88
from django.utils.translation import gettext_lazy as _
99

1010
from django_mongodb.forms import SimpleArrayField
@@ -267,7 +267,7 @@ def _rhs_not_none_values(self, rhs):
267267

268268

269269
@ArrayField.register_lookup
270-
class ArrayContains(Lookup): # ArrayRHSMixin, lookups.DataContains):
270+
class ArrayContains(FieldGetDbPrepValueMixin, Lookup):
271271
lookup_name = "contains"
272272

273273
def as_mql(self, compiler, connection):

0 commit comments

Comments
 (0)