dir show module content

palaviv · palaviv · commit d24f79b17d41 · 2019-03-04T21:27:03.000+02:00
diff --git a/vm/src/obj/objtype.rs b/vm/src/obj/objtype.rs
@@ -2,8 +2,8 @@ use super::objdict;
 use super::objstr;
 use super::objtype; // Required for arg_check! to use isinstance
 use crate::pyobject::{
-    AttributeProtocol, IdProtocol, PyAttributes, PyContext, PyFuncArgs, PyObject, PyObjectPayload,
-    PyObjectRef, PyResult, TypeProtocol,
+    AttributeProtocol, DictProtocol, IdProtocol, PyAttributes, PyContext, PyFuncArgs, PyObject,
+    PyObjectPayload, PyObjectRef, PyResult, TypeProtocol,
 };
 use crate::vm::VirtualMachine;
 use std::cell::RefCell;
@@ -277,6 +277,13 @@ pub fn get_attributes(obj: &PyObjectRef) -> PyAttributes {
             attributes.insert(name.to_string(), value.clone());
         }
     }
+
+    // Get module attributes:
+    if let PyObjectPayload::Module { ref scope, .. } = &obj.payload {
+        for (name, value) in scope.locals.get_key_value_pairs().iter() {
+            attributes.insert(objstr::get_value(name).to_string(), value.clone());
+        }
+    }
     attributes
 }
 
