fix enum for traitlets 5

Author: vidartf

js/scripts/templates/py_enums.mustache

@@ -15,6 +15,9 @@ class EnumNamespace:
     def __contains__(self, key):
         return key in self.__dict__
 
+    def __iter__(self):
+        yield from filter(lambda e: not e.startswith('_'), dir(self))
+
     def __repr__(self):
         return str(list(filter(lambda e: not e.startswith('_'), dir(self))))
 

pythreejs/enums.py

@@ -15,6 +15,9 @@ def __init__(self, **kwargs):
     def __contains__(self, key):
         return key in self.__dict__
 
+    def __iter__(self):
+        yield from filter(lambda e: not e.startswith('_'), dir(self))
+
     def __repr__(self):
         return str(list(filter(lambda e: not e.startswith('_'), dir(self))))