From 274b4fa5d3c8c0abd2725eb14ca8f1146a48aa42 Mon Sep 17 00:00:00 2001
From: messense <messense@icloud.com>
Date: Fri, 15 Apr 2022 22:15:44 +0800
Subject: [PATCH] Remove `PYO3_NO_PYTHON` hack for `pyo3 >= 0.16.4`

---
 Changelog.md   |  2 ++
 src/compile.rs | 33 ++++++++++++++++++++++++++++++---
 2 files changed, 32 insertions(+), 3 deletions(-)

diff --git a/Changelog.md b/Changelog.md
index 74cc61e59..6af36f9ec 100644
--- a/Changelog.md
+++ b/Changelog.md
@@ -7,6 +7,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+* Stop setting `PYO3_NO_PYTHON` environment variable for pyo3 0.16.4 and later in [#875](https://github.com/PyO3/maturin/pull/875)
+
 ## [0.12.12] - 2022-04-07
 
 * Migrate docker image to GitHub container registry in [#845](https://github.com/PyO3/maturin/pull/845)
diff --git a/src/compile.rs b/src/compile.rs
index d1cf8c6a8..153d41574 100644
--- a/src/compile.rs
+++ b/src/compile.rs
@@ -11,6 +11,10 @@ use std::path::{Path, PathBuf};
 use std::process::Stdio;
 use std::str;
 
+/// The first version of pyo3 that supports building Windows abi3 wheel
+/// without `PYO3_NO_PYTHON` environment variable
+const PYO3_ABI3_NO_PYTHON_VERSION: (u64, u64, u64) = (0, 16, 4);
+
 /// Builds the rust crate into a native module (i.e. an .so or .dll) for a
 /// specific python version. Returns a mapping from crate type (e.g. cdylib)
 /// to artifact location.
@@ -218,9 +222,13 @@ fn compile_target(
             .map(|p| p.interpreter_kind == InterpreterKind::PyPy)
             .unwrap_or(false);
         if !is_pypy && !target.is_windows() {
-            // This will make pyo3's build script only set some predefined linker
-            // arguments without trying to read any python configuration
-            build_command.env("PYO3_NO_PYTHON", "1");
+            let pyo3_ver = pyo3_version(&context.cargo_metadata)
+                .context("Failed to get pyo3 version from cargo metadata")?;
+            if pyo3_ver < PYO3_ABI3_NO_PYTHON_VERSION {
+                // This will make old pyo3's build script only set some predefined linker
+                // arguments without trying to read any python configuration
+                build_command.env("PYO3_NO_PYTHON", "1");
+            }
         }
     }
 
@@ -386,3 +394,22 @@ pub fn warn_missing_py_init(artifact: &Path, module_name: &str) -> Result<()> {
 
     Ok(())
 }
+
+fn pyo3_version(cargo_metadata: &cargo_metadata::Metadata) -> Option<(u64, u64, u64)> {
+    let packages: HashMap<&str, &cargo_metadata::Package> = cargo_metadata
+        .packages
+        .iter()
+        .filter_map(|pkg| {
+            let name = &pkg.name;
+            if name == "pyo3" || name == "pyo3-ffi" {
+                Some((name.as_ref(), pkg))
+            } else {
+                None
+            }
+        })
+        .collect();
+    packages
+        .get("pyo3")
+        .or_else(|| packages.get("pyo3-ffi"))
+        .map(|pkg| (pkg.version.major, pkg.version.minor, pkg.version.patch))
+}