Enable python compilation toolchain

.bazelrc

@@ -7,3 +7,6 @@ test --test_output=errors
 # creating (possibly empty) __init__.py files and adding them to the srcs of
 # Python targets as required.
 build --incompatible_default_to_explicit_init_py
+
+# Enable protobuf toolchain.
+common --incompatible_enable_proto_toolchain_resolution

MODULE.bazel

@@ -11,6 +11,9 @@ bazel_dep(
     name = "rules_python",
     version = "0.31.0",
 )
+
+register_toolchains("//tools/protoc:all")
+
 bazel_dep(
     name = "rules_oci",
     version = "1.7.5",

src/python/requirements.lock.txt

@@ -494,6 +494,19 @@ pre-commit==3.6.2 \
     --hash=sha256:ba637c2d7a670c10daedc059f5c49b5bd0aadbccfcd7ec15592cf9665117532c \
     --hash=sha256:c3ef34f463045c88658c5b99f38c1e297abdcc0ff13f98d3370055fbbfabc67e
     # via -r src/python/requirements.txt
+protobuf==5.26.1 \
+    --hash=sha256:38aa5f535721d5bb99861166c445c4105c4e285c765fbb2ac10f116e32dcd46d \
+    --hash=sha256:3c388ea6ddfe735f8cf69e3f7dc7611e73107b60bdfcf5d0f024c3ccd3794e23 \
+    --hash=sha256:7ee014c2c87582e101d6b54260af03b6596728505c79f17c8586e7523aaa8f8c \
+    --hash=sha256:8ca2a1d97c290ec7b16e4e5dff2e5ae150cc1582f55b5ab300d45cb0dfa90e51 \
+    --hash=sha256:9b557c317ebe6836835ec4ef74ec3e994ad0894ea424314ad3552bc6e8835b4e \
+    --hash=sha256:b9ba3ca83c2e31219ffbeb9d76b63aad35a3eb1544170c55336993d7a18ae72c \
+    --hash=sha256:d693d2504ca96750d92d9de8a103102dd648fda04540495535f0fec7577ed8fc \
+    --hash=sha256:da612f2720c0183417194eeaa2523215c4fcc1a1949772dc65f05047e08d5932 \
+    --hash=sha256:e6039957449cb918f331d32ffafa8eb9255769c96aa0560d9a5bf0b4e00a2a33 \
+    --hash=sha256:f7417703f841167e5a27d48be13389d52ad705ec09eade63dfc3180a959215d7 \
+    --hash=sha256:fbfe61e7ee8c1860855696e3ac6cfd1b01af5498facc6834fcc345c9684fb2ca
+    # via -r src/python/requirements.txt
 pylint==3.1.0 \
     --hash=sha256:507a5b60953874766d8a366e8e8c7af63e058b26345cfcb5f91f89d987fd6b74 \
     --hash=sha256:6a69beb4a6f63debebaab0a3477ecd0f559aa726af4954fc948c51f7a2549e23

src/python/requirements.txt

@@ -24,6 +24,7 @@ pathspec
 pg8000
 platformdirs
 pre-commit
+protobuf
 pylint
 pytest
 pyyaml

tools/protoc/BUILD.bazel

@@ -0,0 +1,17 @@
+load("@py_deps//:requirements.bzl", "requirement")
+load("@rules_proto//proto:defs.bzl", "proto_lang_toolchain")
+
+# Configure protoc to have the right arguments for generating Python stubs.
+proto_lang_toolchain(
+    name = "protoc_py_toolchain",
+    command_line = "--python_out=%s",
+    progress_message = "Generating Python proto_library %{label}",
+    runtime = requirement("protobuf"),
+)
+
+# Adapters to the register_toolchains call, adding the toolchain_type to above
+toolchain(
+    name = "protoc_py_toolchain.registration",
+    toolchain = ":protoc_py_toolchain",
+    toolchain_type = "@rules_python//python/proto:toolchain_type",
+)