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

Cleaner loop using itertools.count()

Browse Source
This commit is contained in:
Stefan Wojcik 2019-06-25 11:41:27 +02:00
parent b661afba01
commit 0578cdb62e
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mongoengine/base/metaclasses.py
View File

@ -1,3 +1,4 @@
import itertools
import warnings import warnings
import six import six
@ -440,7 +441,7 @@ class TopLevelDocumentMetaclass(DocumentMetaclass):
return id_name, id_db_name return id_name, id_db_name
id_basename, id_db_basename, i = ('auto_id', '_auto_id', 0) id_basename, id_db_basename, i = ('auto_id', '_auto_id', 0)
while True: for i in itertools.count():
id_name = '{0}_{1}'.format(id_basename, i) id_name = '{0}_{1}'.format(id_basename, i)
id_db_name = '{0}_{1}'.format(id_db_basename, i) id_db_name = '{0}_{1}'.format(id_db_basename, i)
if ( if (
@ -448,8 +449,6 @@ class TopLevelDocumentMetaclass(DocumentMetaclass):
id_db_name not in existing_db_fields id_db_name not in existing_db_fields
): ):
return id_name, id_db_name return id_name, id_db_name
else:
i += 1
class MetaDict(dict): class MetaDict(dict):