allow to use sets in field choices (#1482)

2017-02-15 08:51:47 -05:00
parent ed34c2ca68
commit b27c7ce11b
6 changed files with 68 additions and 35 deletions
@@ -193,7 +193,8 @@ class BaseField(object):
        EmbeddedDocument = _import_class('EmbeddedDocument')

        choice_list = self.choices
-        if isinstance(choice_list[0], (list, tuple)):
+        if isinstance(next(iter(choice_list)), (list, tuple)):
+            # next(iter) is useful for sets
            choice_list = [k for k, _ in choice_list]

        # Choices which are other types of Documents