shields/services/pypi/pypi-python-versions.service.js

import semver from 'semver'
import PypiBase, { pypiGeneralParams } from './pypi-base.js'
import { parseClassifiers } from './pypi-helpers.js'

export default class PypiPythonVersions extends PypiBase {
  static category = 'platform-support'

  static route = this.buildRoute('pypi/pyversions')

  static openApi = {
    '/pypi/pyversions/{packageName}': {
      get: {
        summary: 'PyPI - Python Version',
        parameters: pypiGeneralParams,
      },
    },
  }

  static defaultBadgeData = { label: 'python' }

  static render({ versions }) {
    const versionSet = new Set(versions)
    // We only show v2 if eg. v2.4 does not appear.
    // See https://github.com/badges/shields/pull/489 for more.
    ;['2', '3'].forEach(majorVersion => {
      if (Array.from(versions).some(v => v.startsWith(`${majorVersion}.`))) {
        versionSet.delete(majorVersion)
      }
    })
    if (versionSet.size) {
      return {
        message: Array.from(versionSet)
          .sort((v1, v2) =>
            semver.compare(semver.coerce(v1), semver.coerce(v2)),
          )
          .join(' | '),
        color: 'blue',
      }
    } else {
      return {
        message: 'missing',
        color: 'red',
      }
    }
  }

  async handle({ egg }, { pypiBaseUrl }) {
    const packageData = await this.fetch({ egg, pypiBaseUrl })

    const versions = parseClassifiers(
      packageData,
      /^Programming Language :: Python :: ([\d.]+)$/,
    )
    // If no versions are found yet, check "X :: Only" as a fallback.
    if (versions.length === 0) {
      versions.push(
        ...parseClassifiers(
          packageData,
          /^Programming Language :: Python :: (\d+) :: Only$/,
        ),
      )
    }

    return this.constructor.render({ versions })
  }
}