From 822e5a5e70b120d852e278a35a41de32ad7d7f12 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Wed, 16 Oct 2024 17:53:12 +0200
Subject: [PATCH] Update publish.yml (#16967)

---
 .github/workflows/publish.yml | 42 ++---------------------------------
 1 file changed, 2 insertions(+), 40 deletions(-)

diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml
index e82aecc5a0..024eb8567d 100644
--- a/.github/workflows/publish.yml
+++ b/.github/workflows/publish.yml
@@ -41,49 +41,11 @@ jobs:
         shell: python
         run: |
           import os
-          import requests
-          import toml
-
-          # Load version and package name from pyproject.toml
-          pyproject = toml.load('pyproject.toml')
-          package_name = pyproject['project']['name']
-          local_version = pyproject['project'].get('version', 'dynamic')
-
-          # If version is dynamic, extract it from the specified file
-          if local_version == 'dynamic':
-              version_attr = pyproject['tool']['setuptools']['dynamic']['version']['attr']
-              module_path, attr_name = version_attr.rsplit('.', 1)
-              with open(f"{module_path.replace('.', '/')}/__init__.py") as f:
-                  local_version = next(line.split('=')[1].strip().strip("'\"") for line in f if line.startswith(attr_name))
-
-          print(f"Local Version: {local_version}")
-
-          # Get online version from PyPI
-          response = requests.get(f"https://pypi.org/pypi/{package_name}/json")
-          online_version = response.json()['info']['version'] if response.status_code == 200 else None
-          print(f"Online Version: {online_version or 'Not Found'}")
-
-          # Determine if a new version should be published
-          publish = False
-          if online_version:
-              local_ver = tuple(map(int, local_version.split('.')))
-              online_ver = tuple(map(int, online_version.split('.')))
-              major_diff = local_ver[0] - online_ver[0]
-              minor_diff = local_ver[1] - online_ver[1]
-              patch_diff = local_ver[2] - online_ver[2]
-
-              publish = (
-                  (major_diff == 0 and minor_diff == 0 and 0 < patch_diff <= 2) or
-                  (major_diff == 0 and minor_diff == 1 and local_ver[2] == 0) or
-                  (major_diff == 1 and local_ver[1] == 0 and local_ver[2] == 0)
-              )
-          else:
-              publish = True  # First release
-
+          from actions.utils import check_pypi_version
+          local_version, online_version, publish = check_pypi_version()
           os.system(f'echo "increment={publish}" >> $GITHUB_OUTPUT')
           os.system(f'echo "current_tag=v{local_version}" >> $GITHUB_OUTPUT')
           os.system(f'echo "previous_tag=v{online_version}" >> $GITHUB_OUTPUT')
-
           if publish:
               print('Ready to publish new version to PyPI ✅.')
         id: check_pypi