Assign __module__ properly for py3 enums

Author: aldanor

include/pybind11/pybind11.h

@@ -1259,11 +1259,19 @@ class py3_enum {
 public:
     using underlying_type = typename std::underlying_type<T>::type;
 
-    py3_enum(handle scope, const char* enum_name)
-    : name(enum_name),
-      parent(scope),
+    py3_enum(handle scope, const char* name)
+    : name(name),
+      scope(scope),
       ctor(module::import("enum").attr("IntEnum")),
       unique(module::import("enum").attr("unique")) {
+        kwargs["value"] = cast(name);
+        kwargs["names"] = entries;
+        if (scope) {
+            if (hasattr(scope, "__module__"))
+                kwargs["module"] = scope.attr("__module__");
+            else if (hasattr(scope, "__name__"))
+                kwargs["module"] = scope.attr("__name__");
+        }
         update();
     }
 
@@ -1275,14 +1283,15 @@ class py3_enum {
 
 private:
     const char *name;
-    handle parent;
+    handle scope;
     dict entries;
     object ctor;
     object unique;
+    dict kwargs;
 
     void update() {
-        object type = unique(ctor(name, entries));
-        setattr(parent, name, type);
+        object type = unique(ctor(**kwargs));
+        setattr(scope, name, type);
         detail::type_caster<T>::bind(type, entries);
     }
 };

tests/test_enum.py

@@ -135,6 +135,7 @@ def test_py3_enum():
     for tp, entries in expected.items():
         assert issubclass(tp, IntEnum)
         assert sorted(tp.__members__.items()) == entries
+        assert tp.__module__ == 'pybind11_tests'
 
     assert make_py3_enum(True) is Py3EnumScoped.X
     assert make_py3_enum(False) is Py3EnumScoped.Y