python: add pymanager Python download option

Signed-off-by: Adrian Vladu <avladu@cloudbasesolutions.com>

Author: ader1990

.github/workflows/build_test_cbsinit.yml

@@ -9,7 +9,7 @@ jobs:
       max-parallel: 100
       matrix:
         os: ['windows-2022']
-        download_official_python_msi: ["true", "false"]
+        download_official_python_msi: ["true"]
         cbsinit_repo: ['https://github.com/cloudbase/cloudbase-init']
         cbsinit_branch: ['master']
         python_version: ['3.14_4']
@@ -32,7 +32,7 @@ jobs:
             -CloudbaseInitRepoUrl ${{ matrix.cbsinit_repo }} ^
             -CloudbaseInitRepoBranch ${{ matrix.cbsinit_branch }} ^
             -InstallOfficialPythonMsi:$${{ matrix.download_official_python_msi }} ^
-            -OfficialPythonMsiChecksum "C10234D0D9BD89F6F6DD55BAE28EDE0F97EE0DF4"
+            -InstallOfficialPythonUsingPyManager:$true
     - uses: actions/upload-artifact@v7
       with:
         name: "CloudbaseInit_${{ matrix.platform }}_${{ matrix.os }}_MSI_${{ matrix.cbsinit_branch }}"

BuildAutomation/BuildCloudbaseInitSetup.ps1

@@ -12,7 +12,8 @@ Param(
   [string]$SignTimestampUrl = "http://timestamp.digicert.com?alg=sha256",
   [string]$VCVars = "2019",
   [switch]$InstallOfficialPythonMsi = $false,
-  [string]$OfficialPythonMsiChecksum = "C10234D0D9BD89F6F6DD55BAE28EDE0F97EE0DF4"
+  [string]$OfficialPythonMsiChecksum = "C10234D0D9BD89F6F6DD55BAE28EDE0F97EE0DF4",
+  [switch]$InstallOfficialPythonUsingPyManager = $false
 )
 
 $ErrorActionPreference = "Stop"
@@ -80,11 +81,16 @@ try
     $python_template_dir = join-path $cloudbaseInitInstallerDir "Python$($pythonversion.replace('.', ''))_${platform}_Template"
 
     if ($InstallOfficialPythonMsi) {
-        if (!$OfficialPythonMsiChecksum) {
-            throw "Please set a OfficialPythonMsiChecksum parameter value."
+        if ($InstallOfficialPythonUsingPyManager) {
+                Remove-Item -Recurse -Force $python_template_dir -ErrorAction SilentlyContinue
+                DownloadInstall-PythonUsingPyManager $platform $python_template_dir $pythonversion
+        } else {
+                if (!$OfficialPythonMsiChecksum) {
+                    throw "Please set a OfficialPythonMsiChecksum parameter value."
+                }
+                Remove-Item -Recurse -Force $python_template_dir -ErrorAction SilentlyContinue
+                DownloadInstall-PythonMsi $platform $python_template_dir $pythonversion $OfficialPythonMsiChecksum
         }
-        Remove-Item -Recurse -Force $python_template_dir -ErrorAction SilentlyContinue
-        DownloadInstall-PythonMsi $platform $python_template_dir $pythonversion $OfficialPythonMsiChecksum
     }
 
     CheckCopyDir $python_template_dir $python_dir

BuildAutomation/BuildUtils.ps1

@@ -460,3 +460,41 @@ function DownloadInstall-PythonMsi($platform, $python_template_dir, $pythonVersi
       }
 
 }
+
+function DownloadInstall-PythonUsingPyManager($platform, $python_template_dir, $pythonVersion) {
+    $pythonManagerUrl = "https://www.python.org/ftp/python/pymanager/python-manager-26.1.msix"
+    $pythonManagerPath = Join-Path (Resolve-Path "${python_template_dir}/..").Path "/python-manager.exe"
+
+    $pythonManagerPackage = Get-AppPackage -Name PythonSoftwareFoundation.PythonManager -ErrorAction SilentlyContinue
+
+    if (!$pythonManagerPackage) {
+        ExecRetry { DownloadFile $pythonManagerUrl $pythonManagerPath }
+        Add-AppPackage -Path $pythonManagerPath
+    }
+
+    $platformSuffix = ""
+    if ($platform -eq "x86") {
+        $platformSuffix = "-32"
+    }
+    if ($platform -eq "arm64") {
+        $platformSuffix = "-arm64"
+    }
+
+    if (Test-Path $python_template_dir) {
+        throw "$python_template_dir folder already exists"
+    }
+
+    $pythonVersionEscaped = $pythonVersion.replace("_",".") + $platformSuffix
+    pymanager.exe install --target=$python_template_dir --force --update $pythonVersionEscaped
+    if ($LASTEXITCODE) {
+        throw "Failed to install python in directory: ${python_template_dir}"
+    }
+    & "$python_template_dir/python.exe" --version
+    if ($LASTEXITCODE) {
+        throw "Failed to run python in directory: ${python_template_dir}"
+    }
+    if (!(Test-Path $python_template_dir)) {
+        throw "$python_template_dir has not been created"
+    }
+
+}