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

improved string operation code

Browse Source
This commit is contained in:
Bastien Gérard 2019-03-08 17:09:39 +01:00
parent b407c0e6c6
commit 82e28dec43
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
mongoengine/fields.py
View File

@ -13,6 +13,8 @@ import pymongo
import six
from six import iteritems
from mongoengine.queryset.transform import STRING_OPERATORS
try:
import dateutil
except ImportError:
@ -106,11 +108,11 @@ class StringField(BaseField):
if not isinstance(op, six.string_types):
return value
if op.lstrip('i') in ('startswith', 'endswith', 'contains', 'exact'):
flags = 0
if op.startswith('i'):
flags = re.IGNORECASE
op = op.lstrip('i')
if op in STRING_OPERATORS:
case_insensitive = op.startswith('i')
op = op.lstrip('i')
flags = re.IGNORECASE if case_insensitive else 0
regex = r'%s'
if op == 'startswith':