pub/mongoengine
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

run black to please ci

Browse Source
This commit is contained in:
Bastien Gérard 2020-08-11 21:55:22 +02:00
parent a5302b870b
commit 7116dec74a
2 changed files with 5 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mongoengine/fields.py
View File

@ -774,7 +774,7 @@ class EmbeddedDocumentField(BaseField):
def prepare_query_value(self, op, value): def prepare_query_value(self, op, value):
if value is not None and not isinstance(value, self.document_type): if value is not None and not isinstance(value, self.document_type):
# Short circuit for special operators, returning them as is # Short circuit for special operators, returning them as is
if isinstance(value, dict) and all(k.startswith('$') for k in value.keys()): if isinstance(value, dict) and all(k.startswith("$") for k in value.keys()):
return value return value
try: try:
value = self.document_type._from_son(value) value = self.document_type._from_son(value)

13
tests/queryset/test_transform.py
View File

@ -352,24 +352,19 @@ class TestTransform(unittest.TestCase):
class Drink(EmbeddedDocument): class Drink(EmbeddedDocument):
id = StringField() id = StringField()
meta = { meta = {"strict": False}
'strict': False
}
class Shop(Document): class Shop(Document):
drinks = EmbeddedDocumentListField(Drink) drinks = EmbeddedDocumentListField(Drink)
Shop.drop_collection() Shop.drop_collection()
drinks = [Drink(id='drink_1'), Drink(id='drink_2')] drinks = [Drink(id="drink_1"), Drink(id="drink_2")]
Shop.objects.create(drinks=drinks) Shop.objects.create(drinks=drinks)
q_obj = transform.query( q_obj = transform.query(
Shop, Shop, drinks__all=[{"$elemMatch": {"_id": x.id}} for x in drinks]
drinks__all=[{'$elemMatch': {'_id': x.id}} for x in drinks]
) )
assert q_obj == { assert q_obj == {
'drinks': { "drinks": {"$all": [{"$elemMatch": {"_id": x.id}} for x in drinks]}
'$all': [{'$elemMatch': {'_id': x.id}} for x in drinks]
}
} }
Shop.drop_collection() Shop.drop_collection()