From 02a165922e46e5ed6dd3ed2446141cd0922a7c54 Mon Sep 17 00:00:00 2001
From: Sutou Kouhei <kou@clear-code.com>
Date: Fri, 20 Dec 2024 20:16:51 +0900
Subject: [PATCH] GH-45076: [CI][Packaging][Python] Simplify
 dev/tasks/python-wheels/github.linux.yml (#45077)
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

### Rationale for this change

We can use GitHub Actions' standard features not Jinja2 features to generate test image names.

### What changes are included in this PR?

Use `GITHUB_ENV` instead of Jinja2.

### Are these changes tested?

Yes.

### Are there any user-facing changes?

No.

* GitHub Issue: https://github.com/apache/arrow/issues/45076

Authored-by: Sutou Kouhei <kou@clear-code.com>
Signed-off-by: Raúl Cumplido <raulcumplido@gmail.com>
---
 dev/tasks/python-wheels/github.linux.yml | 25 ++++++++++++------------
 1 file changed, 12 insertions(+), 13 deletions(-)

diff --git a/dev/tasks/python-wheels/github.linux.yml b/dev/tasks/python-wheels/github.linux.yml
index 8ddd0a23099df..ec5b9b31da8e1 100644
--- a/dev/tasks/python-wheels/github.linux.yml
+++ b/dev/tasks/python-wheels/github.linux.yml
@@ -19,16 +19,6 @@
 
 {{ macros.github_header() }}
 
-# Testing free-threaded wheels uses a different Docker setup
-{% set test_imports_image = (
-    'python-free-threaded-wheel-manylinux-test-imports' if python_abi_tag == 'cp313t'
-    else 'python-wheel-manylinux-test-imports')
-%}
-{% set test_unittests_image = (
-    'python-free-threaded-wheel-manylinux-test-unittests' if python_abi_tag == 'cp313t'
-    else 'python-wheel-manylinux-test-unittests')
-%}
-
 permissions:
   packages: write
 
@@ -60,6 +50,15 @@ jobs:
       {{ macros.github_install_archery()|indent }}
       {{ macros.github_login_dockerhub()|indent }}
 
+      - name: Prepare
+        run: |
+          if [ "${PYTHON_ABI_TAG}" = "cp313t" ]; then
+            test_image_prefix=python-free-threaded
+          else
+            test_image_prefix=python
+          fi
+          echo "TEST_IMAGE_PREFIX=${test_image_prefix}" >> ${GITHUB_ENV}
+
       - name: Build wheel
         shell: bash
         env:
@@ -85,8 +84,8 @@ jobs:
         shell: bash
         run: |
           source arrow/ci/scripts/util_enable_core_dumps.sh
-          archery docker run {{ test_imports_image }}
-          archery docker run {{ test_unittests_image }}
+          archery docker run ${TEST_IMAGE_PREFIX}-wheel-manylinux-test-imports
+          archery docker run ${TEST_IMAGE_PREFIX}-wheel-manylinux-test-unittests
 
       - name: Test wheel on AlmaLinux 8
         shell: bash
@@ -158,5 +157,5 @@ jobs:
         shell: bash
         run: |
           archery docker push python-wheel-manylinux-{{ manylinux_version }}
-          archery docker push {{ test_unittests_image }}
+          archery docker push ${TEST_IMAGE_PREFIX}-wheel-manylinux-test-unittests
       {% endif %}