You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
143 lines
7.0 KiB
143 lines
7.0 KiB
# Ultralytics YOLO 🚀, AGPL-3.0 license |
|
# Publish pip package to PyPI https://pypi.org/project/ultralytics/ |
|
|
|
name: Publish to PyPI |
|
|
|
on: |
|
push: |
|
branches: [main] |
|
workflow_dispatch: |
|
inputs: |
|
pypi: |
|
type: boolean |
|
description: Publish to PyPI |
|
|
|
jobs: |
|
publish: |
|
if: github.repository == 'ultralytics/ultralytics' && github.actor == 'glenn-jocher' |
|
name: Publish |
|
runs-on: ubuntu-latest |
|
steps: |
|
- name: Checkout code |
|
uses: actions/checkout@v4 |
|
with: |
|
token: ${{ secrets.PERSONAL_ACCESS_TOKEN || secrets.GITHUB_TOKEN }} # use your PAT here |
|
- name: Git config |
|
run: | |
|
git config --global user.name "UltralyticsAssistant" |
|
git config --global user.email "web@ultralytics.com" |
|
- name: Set up Python environment |
|
uses: actions/setup-python@v5 |
|
with: |
|
python-version: "3.x" |
|
cache: "pip" # caching pip dependencies |
|
- name: Install dependencies |
|
run: | |
|
python -m pip install --upgrade pip wheel |
|
pip install requests build twine toml |
|
- name: Check PyPI version |
|
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 |
|
|
|
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 |
|
- name: Publish new tag |
|
if: (github.event_name == 'push' || github.event.inputs.pypi == 'true') && steps.check_pypi.outputs.increment == 'True' |
|
run: | |
|
git tag -a "${{ steps.check_pypi.outputs.current_tag }}" -m "$(git log -1 --pretty=%B)" # i.e. "v0.1.2 commit message" |
|
git push origin "${{ steps.check_pypi.outputs.current_tag }}" |
|
- name: Publish new release |
|
if: (github.event_name == 'push' || github.event.inputs.pypi == 'true') && steps.check_pypi.outputs.increment == 'True' |
|
env: |
|
OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }} |
|
GITHUB_TOKEN: ${{ secrets.PERSONAL_ACCESS_TOKEN || secrets.GITHUB_TOKEN }} |
|
CURRENT_TAG: ${{ steps.check_pypi.outputs.current_tag }} |
|
PREVIOUS_TAG: ${{ steps.check_pypi.outputs.previous_tag }} |
|
run: | |
|
curl -s "https://raw.githubusercontent.com/ultralytics/actions/main/utils/summarize_release.py" | python - |
|
shell: bash |
|
- name: Publish to PyPI |
|
continue-on-error: true |
|
if: (github.event_name == 'push' || github.event.inputs.pypi == 'true') && steps.check_pypi.outputs.increment == 'True' |
|
env: |
|
PYPI_TOKEN: ${{ secrets.PYPI_TOKEN }} |
|
run: | |
|
python -m build |
|
python -m twine upload dist/* -u __token__ -p $PYPI_TOKEN |
|
- name: Extract PR Details |
|
env: |
|
GH_TOKEN: ${{ secrets.PERSONAL_ACCESS_TOKEN || secrets.GITHUB_TOKEN }} |
|
run: | |
|
# Check if the event is a pull request or pull_request_target |
|
if [ "${{ github.event_name }}" = "pull_request" ] || [ "${{ github.event_name }}" = "pull_request_target" ]; then |
|
PR_NUMBER=${{ github.event.pull_request.number }} |
|
PR_TITLE=$(gh pr view $PR_NUMBER --json title --jq '.title') |
|
else |
|
# Use gh to find the PR associated with the commit |
|
COMMIT_SHA=${{ github.event.after }} |
|
PR_JSON=$(gh pr list --search "${COMMIT_SHA}" --state merged --json number,title --jq '.[0]') |
|
PR_NUMBER=$(echo $PR_JSON | jq -r '.number') |
|
PR_TITLE=$(echo $PR_JSON | jq -r '.title') |
|
fi |
|
echo "PR_NUMBER=$PR_NUMBER" >> $GITHUB_ENV |
|
echo "PR_TITLE=$PR_TITLE" >> $GITHUB_ENV |
|
- name: Notify on Slack (Success) |
|
if: success() && github.event_name == 'push' && steps.check_pypi.outputs.increment == 'True' |
|
uses: slackapi/slack-github-action@v1.27.0 |
|
with: |
|
payload: | |
|
{"text": "<!channel> GitHub Actions success for ${{ github.workflow }} ✅\n\n\n*Repository:* https://github.com/${{ github.repository }}\n*Action:* https://github.com/${{ github.repository }}/actions/runs/${{ github.run_id }}\n*Author:* ${{ github.actor }}\n*Event:* NEW '${{ github.repository }} ${{ steps.check_pypi.outputs.current_tag }}' pip package published 😃\n*Job Status:* ${{ job.status }}\n*Pull Request:* <https://github.com/${{ github.repository }}/pull/${{ env.PR_NUMBER }}> ${{ env.PR_TITLE }}\n"} |
|
env: |
|
SLACK_WEBHOOK_URL: ${{ secrets.SLACK_WEBHOOK_URL_YOLO }} |
|
- name: Notify on Slack (Failure) |
|
if: failure() |
|
uses: slackapi/slack-github-action@v1.27.0 |
|
with: |
|
payload: | |
|
{"text": "<!channel> GitHub Actions error for ${{ github.workflow }} ❌\n\n\n*Repository:* https://github.com/${{ github.repository }}\n*Action:* https://github.com/${{ github.repository }}/actions/runs/${{ github.run_id }}\n*Author:* ${{ github.actor }}\n*Event:* ${{ github.event_name }}\n*Job Status:* ${{ job.status }}\n*Pull Request:* <https://github.com/${{ github.repository }}/pull/${{ env.PR_NUMBER }}> ${{ env.PR_TITLE }}\n"} |
|
env: |
|
SLACK_WEBHOOK_URL: ${{ secrets.SLACK_WEBHOOK_URL_YOLO }}
|
|
|