From a79ad052d782c18b9b099a50c7cfb99ec6e7d792 Mon Sep 17 00:00:00 2001
From: messense <messense@icloud.com>
Date: Thu, 26 May 2022 22:47:59 +0800
Subject: [PATCH] Fix python interpreter detection on arm64 Windows

---
 src/python_interpreter/mod.rs | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/src/python_interpreter/mod.rs b/src/python_interpreter/mod.rs
index 4b76c8fb6..2c24c9f0a 100644
--- a/src/python_interpreter/mod.rs
+++ b/src/python_interpreter/mod.rs
@@ -98,27 +98,29 @@ fn find_all_windows(target: &Target, min_python_minor: usize) -> Result<Vec<Stri
     // which is used to find the installed interpreters
     let execution = Command::new("py").arg("-0").output();
     if let Ok(output) = execution {
-        let expr = Regex::new(r" -(\d).(\d+)-(\d+)(?: .*)?").unwrap();
+        // x86_64: ' -3.10-64 *'
+        // arm64:  ' -V:3.11-arm64 * Python 3.11 (ARM64)
+        let expr = Regex::new(r" -(V:)?(\d).(\d+)-(arm)?(\d+)(?: .*)?").unwrap();
         let lines = str::from_utf8(&output.stdout).unwrap().lines();
         for line in lines {
             if let Some(capture) = expr.captures(line) {
                 let context = "Expected a digit";
 
                 let major = capture
-                    .get(1)
+                    .get(2)
                     .unwrap()
                     .as_str()
                     .parse::<usize>()
                     .context(context)?;
                 let minor = capture
-                    .get(2)
+                    .get(3)
                     .unwrap()
                     .as_str()
                     .parse::<usize>()
                     .context(context)?;
                 if !versions_found.contains(&(major, minor)) {
                     let pointer_width = capture
-                        .get(3)
+                        .get(5)
                         .unwrap()
                         .as_str()
                         .parse::<usize>()