1 files changed, 38 insertions, 0 deletions
diff --git a/libs/plugin_python/typings/generate.py b/libs/plugin_python/typings/generate.py
new file mode 100644
index 0000000..1983186
--- /dev/null
+++ b/libs/plugin_python/typings/generate.py
@@ -0,0 +1,38 @@
+import os
+import site
+import sys
+from pathlib import Path
+from typing import cast
+
+import pybind11_stubgen as py11stubs
+
+typings_dir = Path(__file__).parent
+mobase_tests_dir = Path(__file__).parent.parent.joinpath(
+    "vsbuild", "tests", "python", "pylibs", "mobase_tests"
+)
+
+site.addsitedir(str(mobase_tests_dir.parent))
+
+os.add_dll_directory(str(Path(cast(str, os.getenv("QT_ROOT"))).joinpath("bin")))
+os.add_dll_directory(str(os.getenv("UIBASE_PATH")))
+
+from PyQt6.QtWidgets import QApplication  # noqa: E402
+
+app = QApplication(sys.argv)
+
+args = py11stubs.arg_parser().parse_args(["dummy"], namespace=py11stubs.CLIArgs())
+
+parser = py11stubs.stub_parser_from_args(args)
+printer = py11stubs.Printer(invalid_expr_as_ellipses=True)  # type: ignore
+
+for path in mobase_tests_dir.glob("*.pyd"):
+    name = path.name.split(".")[0]
+    py11stubs.run(
+        parser,
+        printer,
+        f"mobase_tests.{name}",
+        typings_dir.joinpath("mobase_tests"),
+        sub_dir=None,
+        dry_run=False,
+        writer=py11stubs.Writer(stub_ext="pyi"),  # type: ignore
+    )