Generate stubbed package with shell script

Author: Thadah

.gitignore

@@ -13,4 +13,7 @@ docs/
 demoData/
 dist/
 openfhe/openfhe.so
-openfhe.egg-info/
+openfhe/*.pyi
+openfhe.egg-info/
+stubs/
+.venv/

build_package.sh

@@ -0,0 +1,51 @@
+#!/bin/bash
+# Exit on any error
+set -e
+
+# Find the venv directory
+if [ -d ".venv" ]; then
+    VENV_DIR=".venv"
+elif [ -d "../.venv" ]; then
+    VENV_DIR="../.venv"
+else
+    echo "The virtual environment does not exist. Please run 'python -m venv .venv' to create it." >&2
+    exit 1
+fi
+
+# Activate the virtual environment
+source $VENV_DIR/bin/activate
+
+# Install pybind11-stubgen
+if ! pip show pybind11-stubgen > /dev/null; then
+    pip install pybind11-stubgen
+fi
+
+# Check if the virtual environment has the openfhe package installed
+if ! pip show openfhe > /dev/null; then
+    echo "The openfhe package is not installed in the virtual environment. Please run 'pip install -e .' to install it." >&2
+    exit 1
+fi
+
+# Generate stub files using pybind11-stubgen
+echo "Generating stub files..."
+pybind11-stubgen openfhe
+
+# Check if stub generation was successful
+if [ $? -eq 0 ]; then
+    echo "Stub files generated successfully."
+else
+    echo "Stub generation failed." >&2
+    exit 1
+fi
+
+# Move the generated stub files to the openfhe package directory
+echo "Moving the generated stub files to the openfhe package directory..."
+mv stubs/openfhe/* openfhe/
+rm -r -d stubs
+
+# Build the source distribution and wheel distribution
+echo "Building the sdist and bdist_wheel..."
+python setup.py sdist bdist_wheel
+
+# Indicate where the distributions were saved
+echo "The distributions have been built and are located in the 'dist' directory. You can install the package using 'pip install dist/<distribution_file>'."

setup.py

@@ -66,13 +66,13 @@ def run(self):
     url='https://github.com/openfheorg/openfhe-python',
     license='BSD-2-Clause',
     packages=['openfhe'],
-    package_data={'openfhe': ['*.so']},
+    package_data={'openfhe': ['*.so', '*.pyi']},
     ext_modules=[CMakeExtension('openfhe', sourcedir='')],
     cmdclass={
         'build_ext': CMakeBuild,
         'sdist': SDist
     },
     include_package_data=True,
     python_requires=">=3.6",
-    install_requires=['pybind11', 'pybind11-global']
+    install_requires=['pybind11', 'pybind11-global', 'pybind11-stubgen']
 )