Merge branch 'add_automatic_stubgen' into add_stubgen_autogeneration

Author: bakpaul

.github/workflows/ci.yml

@@ -83,14 +83,9 @@ jobs:
           #Setup env 
           echo "PYTHONPATH=$WORKSPACE_ARTIFACT_PATH/lib/python3/site-packages" | tee -a $GITHUB_ENV
           
-          if [[ "$RUNNER_OS" == "Windows" ]]; then
-            cmd //c "${{ steps.sofa.outputs.vs_vsdevcmd }} \
-              && cd /d ${{ env.WORKSPACE_INSTALL_PATH }}/lib/python3/site-packages/ \
-              && bash ${{ env.WORKSPACE_SRC_PATH }}/scripts/generate_stubs.sh ${{ env.WORKSPACE_INSTALL_PATH }}/lib/python3/site-packages/"
-          else
-            cd ${{ env.WORKSPACE_INSTALL_PATH }}/lib/python3/site-packages/
-            /bin/bash ${{ env.WORKSPACE_SRC_PATH }}/scripts/generate_stubs.sh ${{ env.WORKSPACE_INSTALL_PATH }}/lib/python3/site-packages/
-          fi
+          #For now use pybind11. This might be parametrized as an input of this action
+          ${{ steps.sofa.outputs.python_exe }} ${{ env.WORKSPACE_SRC_PATH }}/scripts/generate_stubs.py -d /home/paul/dev/build/sofa/lib/python3/site-packages -m Sofa --use_pybind11
+
 
       - name: Set env vars for artifacts
         shell: bash

scripts/generate_stubs.py

@@ -0,0 +1,35 @@
+import sys
+import argparse
+from utils import generate_module_stubs, generate_component_stubs
+
+
+def main(site_package_dir,modules_name,use_pybind11 = False):
+
+    work_dir = site_package_dir
+    modules = modules_name
+
+    #Generate stubs using either pybind11-stubgen or mypy version of stubgen
+
+    print(f"Generating stubgen for modules: {modules}")
+
+    for module_name in modules:
+        generate_module_stubs(module_name, work_dir,use_pybind11)
+
+    #Generate stubs for components using the factory
+    target_name="Sofa.Component"
+    generate_component_stubs(work_dir,target_name)
+
+
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(
+                    prog='generate_stubs',
+                    description='Generates python stubs for SOFA modules')
+
+    parser.add_argument('--use_pybind11',action='store_true',help='If flag is present, will use pybind11-stubgen instead of mypy stugen')
+    parser.add_argument('-d','--site_package_dir',nargs=1,help='Path to the site-package folder containing the SOFA modules')
+    parser.add_argument('-m','--modules_name',nargs='+',help='List of modules names to generate stubs for')
+
+    args = parser.parse_args()
+
+    main(args.site_package_dir,args.modules_name,args.use_pybind11)