Makes sure to inject the module name

Author: Tpt

noxfile.py

@@ -1439,7 +1439,16 @@ def test_introspection(session: nox.Session):
             "import pyo3_pytests; print(pyo3_pytests.pyo3_pytests.__file__)",
             silent=True,
         ).strip()
-        _run_cargo(session, "run", "-p", "pyo3-introspection", "--", lib_file, stub_dir)
+        _run_cargo(
+            session,
+            "run",
+            "-p",
+            "pyo3-introspection",
+            "--",
+            lib_file,
+            "pyo3_pytests",
+            stub_dir,
+        )
         _run(session, "ruff", "format", stub_dir)
         _ensure_directory_equals(Path(stub_dir), Path("pytests/stubs"))
 

pyo3-introspection/src/main.rs

@@ -6,8 +6,8 @@ use std::path::Path;
 use std::{env, fs};
 
 fn main() -> Result<()> {
-    let [_, binary_path, output_path] = env::args().collect::<Vec<_>>().try_into().map_err(|_| anyhow!("pyo3-introspection takes two arguments, the path of the binary to introspect and the path of the directory to write the stub to"))?;
-    let module = introspect_cdylib(&binary_path, "pyo3_pytests")
+    let [_, binary_path, module_name, output_path] = env::args().collect::<Vec<_>>().try_into().map_err(|_| anyhow!("pyo3-introspection takes three arguments, the path of the binary to introspect, the name of the python module to introspect and and the path of the directory to write the stub to"))?;
+    let module = introspect_cdylib(&binary_path, &module_name)
         .with_context(|| format!("Failed to introspect module {binary_path}"))?;
     let actual_stubs = module_stub_files(&module);
     for (path, module) in actual_stubs {