RustPython/vm/src/obj/objmodule.rs

use crate::frame::ScopeRef;
use crate::function::PyRef;
use crate::pyobject::{DictProtocol, PyContext, PyObjectPayload2, PyObjectRef, PyResult};
use crate::vm::VirtualMachine;

#[derive(Clone, Debug)]
pub struct PyModule {
    pub name: String,
    pub scope: ScopeRef,
}
pub type PyModuleRef = PyRef<PyModule>;

impl PyObjectPayload2 for PyModule {
    fn required_type(ctx: &PyContext) -> PyObjectRef {
        ctx.module_type()
    }
}

impl PyModuleRef {
    fn dir(self, vm: &mut VirtualMachine) -> PyResult {
        let keys = self
            .scope
            .locals
            .get_key_value_pairs()
            .iter()
            .map(|(k, _v)| k.clone())
            .collect();
        Ok(vm.ctx.new_list(keys))
    }
}

pub fn init(context: &PyContext) {
    extend_class!(&context, &context.module_type, {
        "__dir__" => context.new_rustfunc(PyModuleRef::dir)
    });
}