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

Use set comprehensions for existing_fields & existing_db_fields

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

4
mongoengine/base/metaclasses.py
View File

@ -431,8 +431,8 @@ class TopLevelDocumentMetaclass(DocumentMetaclass):
of ('auto_id_X', '_auto_id_X') if the default name is already taken.
"""
id_name, id_db_name = ('id', '_id')
existing_fields = new_class._fields
existing_db_fields = (v.db_field for v in new_class._fields.values())
existing_fields = {field_name for field_name in new_class._fields}
existing_db_fields = {v.db_field for v in new_class._fields.values()}
if (
id_name not in existing_fields and
id_db_name not in existing_db_fields