build.rs checks for CPython

ijl · ijl · commit 61ab0e37bb9c · 2025-07-10T14:56:57.000Z
diff --git a/build.rs b/build.rs
@@ -10,6 +10,8 @@ fn main() {
     println!("cargo:rerun-if-env-changed=LDFLAGS");
     println!("cargo:rerun-if-env-changed=ORJSON_DISABLE_YYJSON");
     println!("cargo:rerun-if-env-changed=RUSTFLAGS");
+    println!("cargo:rustc-check-cfg=cfg(CPython)");
+    println!("cargo:rustc-check-cfg=cfg(GraalPy)");
     println!("cargo:rustc-check-cfg=cfg(intrinsics)");
     println!("cargo:rustc-check-cfg=cfg(optimize)");
     println!("cargo:rustc-check-cfg=cfg(Py_3_10)");
@@ -20,12 +22,19 @@ fn main() {
     println!("cargo:rustc-check-cfg=cfg(Py_3_15)");
     println!("cargo:rustc-check-cfg=cfg(Py_3_9)");
     println!("cargo:rustc-check-cfg=cfg(Py_GIL_DISABLED)");
+    println!("cargo:rustc-check-cfg=cfg(PyPy)");
 
     let python_config = pyo3_build_config::get();
     for cfg in python_config.build_script_outputs() {
         println!("{cfg}");
     }
 
+    if python_config.implementation == pyo3_build_config::PythonImplementation::CPython {
+        println!("cargo:rustc-cfg=CPython");
+    } else {
+        panic!("orjson only supports CPython")
+    }
+
     #[allow(unused_variables)]
     let is_64_bit_python = matches!(python_config.pointer_width, Some(64));
 
diff --git a/src/str/pyunicode_new.rs b/src/str/pyunicode_new.rs
@@ -5,6 +5,7 @@ use pyo3_ffi::{PyASCIIObject, PyCompactUnicodeObject, PyObject};
 
 macro_rules! validate_str {
     ($ptr:expr) => {
+        #[cfg(CPython)]
         debug_assert!(ffi!(_PyUnicode_CheckConsistency($ptr.cast::<PyObject>(), 1)) == 1);
     };
 }

Original file line number	Diff line number	Diff line change
`@@ -5,6 +5,7 @@ use pyo3_ffi::{PyASCIIObject, PyCompactUnicodeObject, PyObject};`
`5`	`5`
`6`	`6`	`macro_rules! validate_str {`
`7`	`7`	`($ptr:expr) => {`
	`8`	`+ #[cfg(CPython)]`
`8`	`9`	`debug_assert!(ffi!(_PyUnicode_CheckConsistency($ptr.cast::<PyObject>(), 1)) == 1);`
`9`	`10`	`};`
`10`	`11`	`}`