Function run_file

src/lib.rs:160–193

pymain_run_file_obj in Modules/main.c

(vm: &VirtualMachine, scope: Scope, path: &str)

// pymain_run_file_obj in Modules/main.c
fn run_file(vm: &VirtualMachine, scope: Scope, path: &str) -> PyResult<()> {
    // Check if path is a package/directory with __main__.py
    if let Some(_importer) = get_importer(path, vm)? {
        vm.insert_sys_path(vm.new_pyobj(path))?;
        let runpy = vm.import("runpy", 0)?;
        let run_module_as_main = runpy.get_attr("_run_module_as_main", vm)?;
        run_module_as_main.call((vm::identifier!(vm, __main__).to_owned(), false), vm)?;
        return Ok(());
    }

    // Add script directory to sys.path[0]
    if !vm.state.config.settings.safe_path {
        let dir = std::path::Path::new(path)
            .parent()
            .and_then(|p| p.to_str())
            .unwrap_or("");
        vm.insert_sys_path(vm.new_pyobj(dir))?;
    }

    #[cfg(feature = "host_env")]
    {
        vm.run_any_file(scope, path)
    }
    #[cfg(not(feature = "host_env"))]
    {
        // In sandbox mode, the binary reads the file and feeds source to the VM.
        // The VM itself has no filesystem access.
        let path = if path.is_empty() { "???" } else { path };
        match std::fs::read_to_string(path) {
            Ok(source) => vm.run_string(scope, &source, path.to_owned()).map(drop),
            Err(err) => Err(vm.new_os_error(err.to_string())),
        }
    }
}

fn get_importer(path: &str, vm: &VirtualMachine) -> PyResult<Option<PyObjectRef>> {
    use rustpython_vm::builtins::PyDictRef;