From 2f2fb8fe60e57fa787c42e4fb3ea4110763e58bc Mon Sep 17 00:00:00 2001
From: James Sharpe <james.sharpe@zenotech.com>
Date: Tue, 22 Jul 2025 10:20:44 +0000
Subject: [PATCH] Make python SDK targets configurable

---
 build_defs.bzl | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/build_defs.bzl b/build_defs.bzl
index 5c88f1c..f8c23e6 100644
--- a/build_defs.bzl
+++ b/build_defs.bzl
@@ -22,7 +22,6 @@ PYBIND_FEATURES = [
 
 PYBIND_DEPS = [
     Label("@pybind11//:pybind11"),
-    "@rules_python//python/cc:current_py_cc_headers",
 ]
 
 def pybind_extension(
@@ -32,6 +31,7 @@ def pybind_extension(
         linkopts = [],
         tags = [],
         deps = [],
+        python_headers_target = "@rules_python//python/cc:current_py_cc_headers",
         **kwargs):
     """Builds a Python extension module using pybind11.
 
@@ -42,6 +42,7 @@ def pybind_extension(
       linkopts: Linker options for building the module.
       tags: Tags for the module.
       deps: Dependencies required for building the module.
+      python_headers_target: The cc_library target that exposes the python C SDK headers
       **kwargs: Additional keyword arguments.
 
     This can be directly used in Python with the import statement.
@@ -72,7 +73,7 @@ def pybind_extension(
         }),
         linkshared = 1,
         tags = tags,
-        deps = deps + PYBIND_DEPS,
+        deps = deps + PYBIND_DEPS + [python_headers_target],
         **kwargs
     )
 
@@ -100,6 +101,7 @@ def pybind_library(
         features = [],
         tags = [],
         deps = [],
+        python_headers_target = "@rules_python//python/cc:current_py_cc_headers",
         **kwargs):
     """Builds a pybind11 compatible library. This can be linked to a pybind_extension."""
 
@@ -111,7 +113,7 @@ def pybind_library(
         copts = copts + PYBIND_COPTS,
         features = features + PYBIND_FEATURES,
         tags = tags,
-        deps = deps + PYBIND_DEPS,
+        deps = deps + PYBIND_DEPS + [python_headers_target],
         **kwargs
     )
 
@@ -121,6 +123,8 @@ def pybind_library_test(
         features = [],
         tags = [],
         deps = [],
+        python_headers_target = "@rules_python//python/cc:current_py_cc_headers",
+        python_library_target = "@rules_python//python/cc:current_py_cc_libs",
         **kwargs):
     """Builds a C++ test for a pybind_library."""
 
@@ -133,7 +137,8 @@ def pybind_library_test(
         features = features + PYBIND_FEATURES,
         tags = tags,
         deps = deps + PYBIND_DEPS + [
-            "@rules_python//python/cc:current_py_cc_libs",
+            python_headers_target,
+            python_library_target,
         ],
         **kwargs
     )