Skip to content

Commit 327f0d1

Browse files
jlarizaqcoumes
authored andcommitted
Maps PositiveBigIntegerField field. #55
1 parent 88bb79b commit 327f0d1

File tree

1 file changed

+22
-4
lines changed

1 file changed

+22
-4
lines changed

django_opensearch_dsl/documents.py

Lines changed: 22 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,7 @@
3333
models.ImageField: fields.FileField,
3434
models.IntegerField: fields.IntegerField,
3535
models.NullBooleanField: fields.BooleanField,
36+
models.PositiveBigIntegerField: fields.LongField,
3637
models.PositiveIntegerField: fields.IntegerField,
3738
models.PositiveSmallIntegerField: fields.ShortField,
3839
models.SlugField: fields.KeywordField,
@@ -60,10 +61,18 @@ def __init__(self, related_instance_to_ignore=None, **kwargs):
6061
def search(cls, using=None, index=None):
6162
"""Return a `Search` object parametrized with the index' information."""
6263
return Search(
63-
using=cls._get_using(using), index=cls._default_index(index), doc_type=[cls], model=cls.django.model
64+
using=cls._get_using(using),
65+
index=cls._default_index(index),
66+
doc_type=[cls],
67+
model=cls.django.model,
6468
)
6569

66-
def get_queryset(self, filter_: Optional[Q] = None, exclude: Optional[Q] = None, count: int = None) -> QuerySet:
70+
def get_queryset(
71+
self,
72+
filter_: Optional[Q] = None,
73+
exclude: Optional[Q] = None,
74+
count: int = None,
75+
) -> QuerySet:
6776
"""Return the queryset that should be indexed by this doc type."""
6877
qs = self.django.model.objects.all()
6978

@@ -144,7 +153,10 @@ def init_prepare(self):
144153
if prep_func:
145154
fn = prep_func
146155
else:
147-
fn = partial(field.get_value_from_instance, field_value_to_ignore=self._related_instance_to_ignore)
156+
fn = partial(
157+
field.get_value_from_instance,
158+
field_value_to_ignore=self._related_instance_to_ignore,
159+
)
148160

149161
preparers.append((name, field, fn))
150162

@@ -233,4 +245,10 @@ def update(self, thing, action, *args, refresh=None, using=None, **kwargs): # n
233245
else:
234246
object_list = thing
235247

236-
return self._bulk(self._get_actions(object_list, action), *args, refresh=refresh, using=using, **kwargs)
248+
return self._bulk(
249+
self._get_actions(object_list, action),
250+
*args,
251+
refresh=refresh,
252+
using=using,
253+
**kwargs,
254+
)

0 commit comments

Comments
 (0)