Wait for PyPi package to be available (#25)

* Wait for PyPi package to be available Fixes #24 * Limit PyPi waiting to 15 minutes
pulumi · Jul 26, 2024 · 7fa110c · 7fa110c
1 parent ae81691
commit 7fa110c
Show file tree

Hide file tree

Showing 4 changed files with 45 additions and 2 deletions.
diff --git a/badges/coverage.svg b/badges/coverage.svg
diff --git a/dist/index.js b/dist/index.js
diff --git a/dist/index.js.map b/dist/index.js.map
diff --git a/src/verifyRelease.ts b/src/verifyRelease.ts
@@ -117,6 +117,20 @@ async function installPipPackageVersion(
     core.debug(`Failed to uninstall ${packageRef}`)
   }
 
+  // Wait for up to 15 minutes for the package to be available on PyPI
+  const startTime = Date.now()
+  while (!isPypiPackageAvailable(cwd, pip, packageRef, opts.packageVersion)) {
+    core.debug(
+      `Waiting for ${packageRef}==${opts.packageVersion} to be available on PyPI`
+    )
+    if (Date.now() - startTime > 15 * 60 * 1000) {
+      throw new Error(
+        `Timed out waiting for ${packageRef}==${opts.packageVersion} to be available on PyPI`
+      )
+    }
+    await new Promise(resolve => setTimeout(resolve, 5000)) // 5 seconds
+  }
+
   const packageVersionRef = `${packageRef}==${opts.packageVersion}`
   const installCmd = `${pip} install ${packageVersionRef}`
   core.debug(`Installing pip package: ${installCmd}`)
@@ -131,6 +145,19 @@ async function installPipPackageVersion(
   }
 }
 
+function isPypiPackageAvailable(
+  cwd: string,
+  pip: string,
+  packageRef: string,
+  packageVersion: string
+): boolean {
+  const versions = shell.exec(`${pip} index versions --pre "${packageRef}"`, {
+    cwd,
+    silent: true
+  }).stdout
+  return versions.includes(packageVersion)
+}
+
 async function installDotnetPackageVersion(
   cwd: string,
   opts: VerifyReleaseOptions