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

black reformat

Browse Source
This commit is contained in:
Ido Shraga
2021-10-01 07:48:01 +03:00
parent 3d6b650592
commit 7a0a58c163
2 changed files with 15 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
mongoengine/fields.py
View File

@@ -163,7 +163,7 @@ class StringField(BaseField):
regex = value regex = value
# escape unsafe characters which could lead to a re.error # escape unsafe characters which could lead to a re.error
if op == 'regex': if op == "regex":
value = re.compile(regex, flags) value = re.compile(regex, flags)
else: else:
value = re.escape(value) value = re.escape(value)
@@ -1093,9 +1093,7 @@ class DictField(ComplexBaseField):
return DictField(db_field=member_name) return DictField(db_field=member_name)
def prepare_query_value(self, op, value): def prepare_query_value(self, op, value):
match_operators = [ match_operators = [*STRING_OPERATORS]
*STRING_OPERATORS
]
if op in match_operators and isinstance(value, str): if op in match_operators and isinstance(value, str):
return StringField().prepare_query_value(op, value) return StringField().prepare_query_value(op, value)

27
tests/queryset/test_queryset.py
View File

@@ -872,17 +872,15 @@ class TestQueryset(unittest.TestCase):
self.Person.objects.create(name="Foo", age=11) self.Person.objects.create(name="Foo", age=11)
bob = self.Person.objects.as_pymongo().first() bob = self.Person.objects.as_pymongo().first()
assert 'age' in bob assert "age" in bob
assert bob['age'] == 11 assert bob["age"] == 11
self.Person.objects(name="Foo").update( self.Person.objects(name="Foo").update(rename__age="person_age")
rename__age='person_age'
)
bob = self.Person.objects.as_pymongo().first() bob = self.Person.objects.as_pymongo().first()
assert 'age' not in bob assert "age" not in bob
assert 'person_age' in bob assert "person_age" in bob
assert bob['person_age'] == 11 assert bob["person_age"] == 11
def test_save_and_only_on_fields_with_default(self): def test_save_and_only_on_fields_with_default(self):
class Embed(EmbeddedDocument): class Embed(EmbeddedDocument):
@@ -1257,7 +1255,6 @@ class TestQueryset(unittest.TestCase):
obj = self.Person.objects(name__iexact="gUIDO VAN rOSSU").first() obj = self.Person.objects(name__iexact="gUIDO VAN rOSSU").first()
assert obj is None assert obj is None
# Test wholeword # Test wholeword
obj = self.Person.objects(name__wholeword="Guido").first() obj = self.Person.objects(name__wholeword="Guido").first()
assert obj == person assert obj == person
@@ -1370,12 +1367,14 @@ class TestQueryset(unittest.TestCase):
person.save() person.save()
people = self.Person.objects people = self.Person.objects
people = people.filter(name__startswith="Gui")\ people = (
.filter(name__not__endswith="tum")\ people.filter(name__startswith="Gui")
.filter(name__icontains="VAN")\ .filter(name__not__endswith="tum")
.filter(name__regex="^Guido")\ .filter(name__icontains="VAN")
.filter(name__wholeword="Guido")\ .filter(name__regex="^Guido")
.filter(name__wholeword="Guido")
.filter(name__wholeword="van") .filter(name__wholeword="van")
)
assert people.count() == 1 assert people.count() == 1
def assertSequence(self, qs, expected): def assertSequence(self, qs, expected):