fix: handle upgrade/user flags for pipx

src/python/python.ts

@@ -21,7 +21,7 @@ import { isBinUptoDate } from "../utils/setup/version"
 import { unique } from "../utils/std"
 import { MinVersions } from "../versions/default_versions"
 import { pathExists } from "path-exists"
-import { setupPipPackWithPython } from "../utils/setup/setupPipPack"
+import { hasPipx, setupPipPackWithPython } from "../utils/setup/setupPipPack"
 
 export async function setupPython(version: string, setupDir: string, arch: string): Promise<InstallationInfo> {
   const installInfo = await findOrSetupPython(version, setupDir, arch)
@@ -43,17 +43,19 @@ export async function setupPython(version: string, setupDir: string, arch: strin
 
 async function setupPipx(foundPython: string) {
   try {
-    try {
+    if (!(await hasPipx())) {
-      await setupPipPackWithPython(foundPython, "pipx", undefined, true)
+      try {
-    } catch (err) {
+        await setupPipPackWithPython(foundPython, "pipx", undefined, true)
-      if (isUbuntu()) {
+      } catch (err) {
-        await setupAptPack([{ name: "python3-pipx" }])
+        if (isUbuntu()) {
-      } else if (isArch()) {
+          await setupAptPack([{ name: "python3-pipx" }])
-        await setupPacmanPack("python-pipx")
+        } else if (isArch()) {
-      } else if (hasDnf()) {
+          await setupPacmanPack("python-pipx")
-        await setupDnfPack([{ name: "python3-pipx" }])
+        } else if (hasDnf()) {
-      } else {
+          await setupDnfPack([{ name: "python3-pipx" }])
-        throw err
+        } else {
+          throw err
+        }
       }
     }
     await execa(foundPython, ["-m", "pipx", "ensurepath"], { stdio: "inherit" })

src/utils/setup/setupPipPack.ts

@@ -22,15 +22,16 @@ export async function setupPipPackWithPython(
   upgrade = false,
   user = true,
 ): Promise<InstallationInfo> {
-  const pip = (await which("pipx", { nothrow: true })) !== null ? "pipx" : "pip"
+  const isPipx = await hasPipx()
+  const pip = isPipx ? "pipx" : "pip"
 
   info(`Installing ${name} ${version ?? ""} via ${pip}`)
 
   const nameAndVersion = version !== undefined && version !== "" ? `${name}==${version}` : name
-  const upgradeFlag = upgrade === true ? ["--upgrade"] : []
+  const upgradeFlag = upgrade ? (isPipx ? ["upgrade"] : ["install", "--upgrade"]) : ["install"]
-  const userFlag = user === true ? ["--user"] : []
+  const userFlag = !isPipx && user ? ["--user"] : []
 
-  execaSync(givenPython, ["-m", pip, "install", ...upgradeFlag, ...userFlag, nameAndVersion], {
+  execaSync(givenPython, ["-m", pip, ...upgradeFlag, ...userFlag, nameAndVersion], {
     stdio: "inherit",
   })
 
@@ -42,6 +43,10 @@ export async function setupPipPackWithPython(
   return { binDir }
 }
 
+export async function hasPipx() {
+  return (await which("pipx", { nothrow: true })) !== null
+}
+
 async function getPython_raw(): Promise<string> {
   const pythonBin = (await setupPython(getVersion("python", undefined, await ubuntuVersion()), "", process.arch)).bin
   if (pythonBin === undefined) {