add example with stubs

Parskatt · Parskatt · commit 06f7a31737ff · 2024-05-15T16:37:22.000+02:00
diff --git a/python_example/_C/__init__.pyi b/python_example/_C/__init__.pyi
@@ -0,0 +1,3 @@
+def add(a: int, b: int) ->int:...
+
+def subtract(a: int, b: int) ->int:...
diff --git a/python_example/__init__.py b/python_example/__init__.py
@@ -0,0 +1 @@
+from python_example._C import *
diff --git a/setup.py b/setup.py
@@ -1,6 +1,6 @@
 # Available at setup time due to pyproject.toml
 from pybind11.setup_helpers import Pybind11Extension, build_ext
-from setuptools import setup
+from setuptools import setup, find_packages
 
 __version__ = "0.0.1"
 
@@ -15,7 +15,7 @@
 
 ext_modules = [
     Pybind11Extension(
-        "python_example",
+        "python_example/_C",
         ["src/main.cpp"],
         # Example: passing in the version to the compiled code
         define_macros=[("VERSION_INFO", __version__)],
@@ -30,8 +30,10 @@
     url="https://github.com/pybind/python_example",
     description="A test project using pybind11",
     long_description="",
+    packages=find_packages(include="python_example/*"),
     ext_modules=ext_modules,
     extras_require={"test": "pytest"},
+    package_data={"python_example": ["_C/*.pyi"]},
     # Currently, build_ext only provides an optional "highest supported C++
     # level" feature, but in the future it may provide more features.
     cmdclass={"build_ext": build_ext},
