Generate Unique Indices for SortedListField and EmbeddedDocumentListFields

2019-03-15 13:20:40 +00:00
parent 88c4aa2d87
commit ba6a37f315
4 changed files with 76 additions and 2 deletions
--- a/mongoengine/base/document.py
+++ b/mongoengine/base/document.py
@@ -883,7 +883,8 @@ class BaseDocument(object):
                index = {'fields': fields, 'unique': True, 'sparse': sparse}
                unique_indexes.append(index)

-            if field.__class__.__name__ == 'ListField':
+            if field.__class__.__name__ in {'EmbeddedDocumentListField',
+                                            'ListField', 'SortedListField'}:
                field = field.field

            # Grab any embedded document field unique indexes