Skip to content

Commit 18afe4e

Browse files
committed
Edits
1 parent 5983a61 commit 18afe4e

File tree

2 files changed

+20
-20
lines changed

2 files changed

+20
-20
lines changed

django_mongodb_backend/expressions/search.py

Lines changed: 16 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -161,10 +161,10 @@ def set_source_expressions(self, exprs):
161161
def search_operator(self, compiler, connection):
162162
params = {
163163
"path": self.path.as_mql(compiler, connection, as_path=True),
164-
"value": self.value.as_mql(compiler, connection, as_path=True),
164+
"value": self.value.as_mql(compiler, connection),
165165
}
166166
if self.score is not None:
167-
params["score"] = self.score.as_mql(compiler, connection, as_path=True)
167+
params["score"] = self.score.as_mql(compiler, connection)
168168
return {"equals": params}
169169

170170

@@ -211,7 +211,7 @@ def set_source_expressions(self, exprs):
211211
def search_operator(self, compiler, connection):
212212
params = {
213213
"path": self.path.as_mql(compiler, connection, as_path=True),
214-
"value": self.value.as_mql(compiler, connection, as_path=True),
214+
"value": self.value.as_mql(compiler, connection),
215215
}
216216
if self.score is not None:
217217
params["score"] = self.score.definitions
@@ -234,19 +234,19 @@ def get_source_expressions(self):
234234
return [self.path, self.query, self.slop, self.synonyms]
235235

236236
def set_source_expressions(self, exprs):
237-
self.path, self.query, self.score, self.slop, self.synonyms = exprs
237+
self.path, self.query, self.slop, self.synonyms = exprs
238238

239239
def search_operator(self, compiler, connection):
240240
params = {
241241
"path": self.path.as_mql(compiler, connection, as_path=True),
242-
"query": self.query.as_mql(compiler, connection, as_path=True),
242+
"query": self.query.as_mql(compiler, connection),
243243
}
244244
if self.score is not None:
245-
params["score"] = self.score.as_mql(compiler, connection, as_path=True)
245+
params["score"] = self.score.as_mql(compiler, connection)
246246
if self.slop is not None:
247-
params["slop"] = self.slop.as_mql(compiler, connection, as_path=True)
247+
params["slop"] = self.slop.as_mql(compiler, connection)
248248
if self.synonyms is not None:
249-
params["synonyms"] = self.synonyms.as_mql(compiler, connection, as_path=True)
249+
params["synonyms"] = self.synonyms.as_mql(compiler, connection)
250250
return {"phrase": params}
251251

252252

@@ -268,8 +268,8 @@ def set_source_expressions(self, exprs):
268268

269269
def search_operator(self, compiler, connection):
270270
params = {
271-
"defaultPath": self.path,
272-
"query": self.query.as_mql(compiler, connection, as_path=True),
271+
"defaultPath": self.path.as_mql(compiler, connection, as_path=True),
272+
"query": self.query.as_mql(compiler, connection),
273273
}
274274
if self.score is not None:
275275
params["score"] = self.score.definitions
@@ -290,10 +290,10 @@ def get_search_fields(self, compiler, connection):
290290
return {self.path.as_mql(compiler, connection, as_path=True)}
291291

292292
def get_source_expressions(self):
293-
return [self.path, self.query, self.lt, self.lte, self.gt, self.gte]
293+
return [self.path, self.lt, self.lte, self.gt, self.gte]
294294

295295
def set_source_expressions(self, exprs):
296-
self.path, self.query, self.lt, self.lte, self.gt, self.gte = exprs
296+
self.path, self.lt, self.lte, self.gt, self.gte = exprs
297297

298298
def search_operator(self, compiler, connection):
299299
params = {
@@ -302,13 +302,13 @@ def search_operator(self, compiler, connection):
302302
if self.score is not None:
303303
params["score"] = self.score.definitions
304304
if self.lt is not None:
305-
params["lt"] = self.lt
305+
params["lt"] = self.lt.as_mql(compiler, connection)
306306
if self.lte is not None:
307-
params["lte"] = self.lte
307+
params["lte"] = self.lte.as_mql(compiler, connection)
308308
if self.gt is not None:
309-
params["gt"] = self.gt
309+
params["gt"] = self.gt.as_mql(compiler, connection)
310310
if self.gte is not None:
311-
params["gte"] = self.gte
311+
params["gte"] = self.gte.as_mql(compiler, connection)
312312
return {"range": params}
313313

314314

tests/queries_/test_search.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -97,14 +97,14 @@ def setUp(self):
9797
self.article = Article.objects.create(headline="cross", number=1, body="body")
9898
Article.objects.create(headline="other thing", number=2, body="body")
9999

100-
def test_search_equals(self):
101-
qs = Article.objects.annotate(score=SearchEquals(path="headline", value="cross"))
102-
self.wait_for_assertion(lambda: self.assertCountEqual(qs.all(), [self.article]))
103-
104100
def tearDown(self):
105101
self._tear_down(Article)
106102
super().tearDown()
107103

104+
def test_search_equals(self):
105+
qs = Article.objects.annotate(score=SearchEquals(path="headline", value="cross"))
106+
self.wait_for_assertion(lambda: self.assertCountEqual(qs.all(), [self.article]))
107+
108108

109109
class SearchAutocompleteTest(SearchUtilsMixin):
110110
def setUp(self):

0 commit comments

Comments
 (0)