feat: Complete PyPI Query MCP Server Implementation (#3)
Merge pull request implementing complete PyPI query MCP server with comprehensive features and CI/CD pipeline.
This commit is contained in:
parent
b1a1a6866d
commit
030b3a2607
60
.github/workflows/build-exe.yml
vendored
60
.github/workflows/build-exe.yml
vendored
@ -1,60 +0,0 @@
|
|||||||
name: Build Executable
|
|
||||||
|
|
||||||
on:
|
|
||||||
workflow_run:
|
|
||||||
workflows: ["Python Package"]
|
|
||||||
types:
|
|
||||||
- completed
|
|
||||||
branches:
|
|
||||||
- "main"
|
|
||||||
|
|
||||||
jobs:
|
|
||||||
build:
|
|
||||||
# Only run if the python-publish workflow succeeded
|
|
||||||
if: ${{ github.event.workflow_run.conclusion == 'success' }}
|
|
||||||
strategy:
|
|
||||||
fail-fast: false
|
|
||||||
matrix:
|
|
||||||
target:
|
|
||||||
- os: windows-2022
|
|
||||||
triple: x86_64-pc-windows-msvc
|
|
||||||
- os: ubuntu-22.04
|
|
||||||
triple: x86_64-unknown-linux-gnu
|
|
||||||
- os: macos-12
|
|
||||||
triple: x86_64-apple-darwin
|
|
||||||
runs-on: ${{ matrix.target.os }}
|
|
||||||
permissions:
|
|
||||||
contents: write
|
|
||||||
|
|
||||||
steps:
|
|
||||||
- uses: actions/checkout@v4
|
|
||||||
with:
|
|
||||||
fetch-depth: 0
|
|
||||||
|
|
||||||
- uses: olegtarasov/get-tag@v2.1.4
|
|
||||||
id: get_tag_name
|
|
||||||
with:
|
|
||||||
tagRegex: "v(?<version>.*)"
|
|
||||||
|
|
||||||
- name: Set up Python
|
|
||||||
uses: actions/setup-python@v5
|
|
||||||
with:
|
|
||||||
python-version: "3.10"
|
|
||||||
cache: pip
|
|
||||||
|
|
||||||
- name: Install dependencies
|
|
||||||
run: |
|
|
||||||
python -m pip install --upgrade pip
|
|
||||||
pip install nox
|
|
||||||
|
|
||||||
- name: Build exe
|
|
||||||
run: |
|
|
||||||
nox -s build-exe -- --release --version ${{ steps.get_tag_name.outputs.version }}
|
|
||||||
|
|
||||||
- uses: ncipollo/release-action@v1
|
|
||||||
with:
|
|
||||||
allowUpdates: true
|
|
||||||
updateOnlyUnreleased: false
|
|
||||||
omitBody: true
|
|
||||||
artifacts: ".zip/*.zip"
|
|
||||||
token: ${{ secrets.PERSONAL_ACCESS_TOKEN }}
|
|
23
.github/workflows/bumpversion.yml
vendored
23
.github/workflows/bumpversion.yml
vendored
@ -1,23 +0,0 @@
|
|||||||
name: Bump version
|
|
||||||
|
|
||||||
on:
|
|
||||||
push:
|
|
||||||
branches:
|
|
||||||
- main
|
|
||||||
|
|
||||||
jobs:
|
|
||||||
bump-version:
|
|
||||||
if: "!startsWith(github.event.head_commit.message, 'bump:')"
|
|
||||||
runs-on: ubuntu-latest
|
|
||||||
name: "Bump version and create changelog with commitizen"
|
|
||||||
steps:
|
|
||||||
- name: Check out
|
|
||||||
uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4
|
|
||||||
with:
|
|
||||||
fetch-depth: 0
|
|
||||||
token: '${{ secrets.PERSONAL_ACCESS_TOKEN }}'
|
|
||||||
- name: Create bump and changelog
|
|
||||||
uses: commitizen-tools/commitizen-action@master
|
|
||||||
with:
|
|
||||||
github_token: ${{ secrets.PERSONAL_ACCESS_TOKEN }}
|
|
||||||
branch: main
|
|
34
.github/workflows/codecov.yml
vendored
34
.github/workflows/codecov.yml
vendored
@ -1,34 +0,0 @@
|
|||||||
name: Codecov
|
|
||||||
on: [push, pull_request]
|
|
||||||
jobs:
|
|
||||||
run:
|
|
||||||
runs-on: ubuntu-latest
|
|
||||||
steps:
|
|
||||||
- name: Checkout
|
|
||||||
uses: actions/checkout@v4
|
|
||||||
with:
|
|
||||||
fetch-depth: 0
|
|
||||||
- name: Get repository name
|
|
||||||
id: repo-name
|
|
||||||
uses: MariachiBear/get-repo-name-action@v1.3.0
|
|
||||||
with:
|
|
||||||
with-owner: 'true'
|
|
||||||
string-case: 'uppercase'
|
|
||||||
- name: Set up Python 3.10
|
|
||||||
uses: actions/setup-python@v5
|
|
||||||
with:
|
|
||||||
python-version: '3.10'
|
|
||||||
- name: Install dependencies
|
|
||||||
run: |
|
|
||||||
python -m pip install -r requirements-dev.txt
|
|
||||||
poetry --version
|
|
||||||
- name: Run tests and collect coverage
|
|
||||||
run: |
|
|
||||||
nox -s pytest
|
|
||||||
- name: Upload coverage reports to Codecov
|
|
||||||
uses: codecov/codecov-action@v5
|
|
||||||
with:
|
|
||||||
slug: loonghao/${{ steps.repo-name.outputs.repository-name }}
|
|
||||||
files: 'coverage.xml'
|
|
||||||
env:
|
|
||||||
CODECOV_TOKEN: ${{ secrets.CODECOV_TOKEN }}
|
|
18
.github/workflows/issue-translator.yml
vendored
18
.github/workflows/issue-translator.yml
vendored
@ -1,18 +0,0 @@
|
|||||||
name: 'issue-translator'
|
|
||||||
on:
|
|
||||||
issue_comment:
|
|
||||||
types: [created]
|
|
||||||
issues:
|
|
||||||
types: [opened]
|
|
||||||
|
|
||||||
jobs:
|
|
||||||
build:
|
|
||||||
runs-on: ubuntu-latest
|
|
||||||
steps:
|
|
||||||
- uses: usthe/issues-translate-action@v2.7
|
|
||||||
with:
|
|
||||||
IS_MODIFY_TITLE: false
|
|
||||||
# not require, default false, . Decide whether to modify the issue title
|
|
||||||
# if true, the robot account @Issues-translate-bot must have modification permissions, invite @Issues-translate-bot to your project or use your custom bot.
|
|
||||||
CUSTOM_BOT_NOTE: Bot detected the issue body's language is not English, translate it automatically. 👯👭🏻🧑🤝🧑👫🧑🏿🤝🧑🏻👩🏾🤝👨🏿👬🏿
|
|
||||||
# not require. Customize the translation robot prefix message.
|
|
35
.github/workflows/mr-test.yml
vendored
35
.github/workflows/mr-test.yml
vendored
@ -1,35 +0,0 @@
|
|||||||
name: MR Checks
|
|
||||||
on: [ pull_request ]
|
|
||||||
|
|
||||||
jobs:
|
|
||||||
python-check:
|
|
||||||
strategy:
|
|
||||||
max-parallel: 3
|
|
||||||
matrix:
|
|
||||||
target:
|
|
||||||
- os: 'ubuntu-22.04'
|
|
||||||
triple: 'x86_64-unknown-linux-gnu'
|
|
||||||
- os: 'macos-12'
|
|
||||||
triple: 'x86_64-apple-darwin'
|
|
||||||
- os: 'windows-2022'
|
|
||||||
triple: 'x86_64-pc-windows-msvc'
|
|
||||||
python-version: ["3.10"]
|
|
||||||
fail-fast: false
|
|
||||||
runs-on: ${{ matrix.target.os }}
|
|
||||||
steps:
|
|
||||||
- name: Checkout
|
|
||||||
uses: actions/checkout@v4
|
|
||||||
- name: Set up Python ${{ matrix.python-version }}
|
|
||||||
uses: actions/setup-python@v5
|
|
||||||
with:
|
|
||||||
python-version: ${{ matrix.python-version }}
|
|
||||||
- name: Install dependencies
|
|
||||||
run: |
|
|
||||||
python -m pip install -r requirements-dev.txt
|
|
||||||
poetry --version
|
|
||||||
- name: lint
|
|
||||||
run: |
|
|
||||||
nox -s lint
|
|
||||||
- name: test build
|
|
||||||
run: |
|
|
||||||
nox -s build-exe -- --test
|
|
66
.github/workflows/python-publish.yml
vendored
66
.github/workflows/python-publish.yml
vendored
@ -1,66 +0,0 @@
|
|||||||
name: Upload Python Package
|
|
||||||
|
|
||||||
on:
|
|
||||||
push:
|
|
||||||
tags:
|
|
||||||
- "v*"
|
|
||||||
|
|
||||||
jobs:
|
|
||||||
deploy:
|
|
||||||
runs-on: ubuntu-latest
|
|
||||||
permissions:
|
|
||||||
# IMPORTANT: this permission is mandatory for trusted publishing
|
|
||||||
id-token: write
|
|
||||||
contents: write
|
|
||||||
|
|
||||||
steps:
|
|
||||||
- uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4
|
|
||||||
with:
|
|
||||||
token: "${{ secrets.GITHUB_TOKEN }}"
|
|
||||||
fetch-depth: 0
|
|
||||||
ref: main
|
|
||||||
- uses: olegtarasov/get-tag@v2.1.4
|
|
||||||
id: get_tag_name
|
|
||||||
with:
|
|
||||||
tagRegex: "v(?<version>.*)"
|
|
||||||
- name: Set up Python
|
|
||||||
uses: actions/setup-python@v5
|
|
||||||
with:
|
|
||||||
python-version: '3.x'
|
|
||||||
- name: Install dependencies
|
|
||||||
run: |
|
|
||||||
python -m pip install -r requirements-dev.txt
|
|
||||||
poetry --version
|
|
||||||
poetry build
|
|
||||||
# Note that we don't need credentials.
|
|
||||||
# We rely on https://docs.pypi.org/trusted-publishers/.
|
|
||||||
- name: Upload to PyPI
|
|
||||||
uses: pypa/gh-action-pypi-publish@release/v1
|
|
||||||
with:
|
|
||||||
packages-dir: dist
|
|
||||||
- name: Generate changelog
|
|
||||||
id: changelog
|
|
||||||
uses: jaywcjlove/changelog-generator@main
|
|
||||||
with:
|
|
||||||
token: ${{ secrets.PERSONAL_ACCESS_TOKEN }}
|
|
||||||
filter-author: (|dependabot|renovate\\[bot\\]|dependabot\\[bot\\]|Renovate Bot)
|
|
||||||
filter: '[R|r]elease[d]\s+[v|V]\d(\.\d+){0,2}'
|
|
||||||
template: |
|
|
||||||
## Bugs
|
|
||||||
{{fix}}
|
|
||||||
## Feature
|
|
||||||
{{feat}}
|
|
||||||
## Improve
|
|
||||||
{{refactor,perf,clean}}
|
|
||||||
## Misc
|
|
||||||
{{chore,style,ci||🔶 Nothing change}}
|
|
||||||
## Unknown
|
|
||||||
{{__unknown__}}
|
|
||||||
- uses: ncipollo/release-action@v1
|
|
||||||
with:
|
|
||||||
artifacts: "dist/*"
|
|
||||||
token: ${{ secrets.PERSONAL_ACCESS_TOKEN }}
|
|
||||||
body: |
|
|
||||||
Comparing Changes: ${{ steps.changelog.outputs.compareurl }}
|
|
||||||
|
|
||||||
${{ steps.changelog.outputs.changelog }}
|
|
102
.github/workflows/release.yml
vendored
Normal file
102
.github/workflows/release.yml
vendored
Normal file
@ -0,0 +1,102 @@
|
|||||||
|
name: Release
|
||||||
|
|
||||||
|
on:
|
||||||
|
push:
|
||||||
|
tags:
|
||||||
|
- 'v*'
|
||||||
|
|
||||||
|
permissions:
|
||||||
|
contents: read
|
||||||
|
|
||||||
|
jobs:
|
||||||
|
test:
|
||||||
|
uses: ./.github/workflows/test.yml
|
||||||
|
|
||||||
|
build:
|
||||||
|
needs: test
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
steps:
|
||||||
|
- name: Checkout code
|
||||||
|
uses: actions/checkout@v4
|
||||||
|
with:
|
||||||
|
fetch-depth: 0
|
||||||
|
|
||||||
|
- name: Set up Python
|
||||||
|
uses: actions/setup-python@v5
|
||||||
|
with:
|
||||||
|
python-version: "3.11"
|
||||||
|
|
||||||
|
- name: Install uv
|
||||||
|
run: |
|
||||||
|
python -m pip install --upgrade pip
|
||||||
|
pip install uv
|
||||||
|
|
||||||
|
- name: Install dependencies
|
||||||
|
run: |
|
||||||
|
uvx poetry install
|
||||||
|
|
||||||
|
- name: Build package
|
||||||
|
run: |
|
||||||
|
uvx poetry build
|
||||||
|
|
||||||
|
- name: Check package
|
||||||
|
run: |
|
||||||
|
uvx poetry run twine check dist/*
|
||||||
|
|
||||||
|
- name: Upload build artifacts
|
||||||
|
uses: actions/upload-artifact@v4
|
||||||
|
with:
|
||||||
|
name: dist
|
||||||
|
path: dist/
|
||||||
|
|
||||||
|
publish:
|
||||||
|
needs: build
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
environment: release
|
||||||
|
permissions:
|
||||||
|
id-token: write # IMPORTANT: this permission is mandatory for trusted publishing
|
||||||
|
steps:
|
||||||
|
- name: Download build artifacts
|
||||||
|
uses: actions/download-artifact@v4
|
||||||
|
with:
|
||||||
|
name: dist
|
||||||
|
path: dist/
|
||||||
|
|
||||||
|
- name: Publish to PyPI
|
||||||
|
uses: pypa/gh-action-pypi-publish@release/v1
|
||||||
|
with:
|
||||||
|
skip-existing: true
|
||||||
|
|
||||||
|
github-release:
|
||||||
|
needs: publish
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
permissions:
|
||||||
|
contents: write
|
||||||
|
steps:
|
||||||
|
- name: Checkout code
|
||||||
|
uses: actions/checkout@v4
|
||||||
|
with:
|
||||||
|
fetch-depth: 0
|
||||||
|
|
||||||
|
- name: Generate changelog
|
||||||
|
id: changelog
|
||||||
|
run: |
|
||||||
|
# Extract version from tag
|
||||||
|
VERSION=${GITHUB_REF#refs/tags/v}
|
||||||
|
echo "version=$VERSION" >> $GITHUB_OUTPUT
|
||||||
|
|
||||||
|
# Generate changelog (basic implementation)
|
||||||
|
echo "## Changes in v$VERSION" > CHANGELOG.md
|
||||||
|
echo "" >> CHANGELOG.md
|
||||||
|
git log --pretty=format:"- %s" $(git describe --tags --abbrev=0 HEAD^)..HEAD >> CHANGELOG.md || echo "- Initial release" >> CHANGELOG.md
|
||||||
|
|
||||||
|
- name: Create GitHub Release
|
||||||
|
uses: actions/create-release@v1
|
||||||
|
env:
|
||||||
|
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
with:
|
||||||
|
tag_name: ${{ github.ref }}
|
||||||
|
release_name: Release v${{ steps.changelog.outputs.version }}
|
||||||
|
body_path: CHANGELOG.md
|
||||||
|
draft: false
|
||||||
|
prerelease: false
|
104
.github/workflows/test.yml
vendored
Normal file
104
.github/workflows/test.yml
vendored
Normal file
@ -0,0 +1,104 @@
|
|||||||
|
name: Tests
|
||||||
|
|
||||||
|
on:
|
||||||
|
push:
|
||||||
|
branches: [ main, develop ]
|
||||||
|
pull_request:
|
||||||
|
branches: [ main, develop ]
|
||||||
|
|
||||||
|
jobs:
|
||||||
|
test:
|
||||||
|
strategy:
|
||||||
|
max-parallel: 6
|
||||||
|
matrix:
|
||||||
|
os: [ubuntu-latest, windows-latest, macos-latest]
|
||||||
|
python-version: ["3.10", "3.11", "3.12"]
|
||||||
|
fail-fast: false
|
||||||
|
|
||||||
|
runs-on: ${{ matrix.os }}
|
||||||
|
|
||||||
|
steps:
|
||||||
|
- name: Checkout code
|
||||||
|
uses: actions/checkout@v4
|
||||||
|
with:
|
||||||
|
fetch-depth: 0
|
||||||
|
|
||||||
|
- name: Set up Python ${{ matrix.python-version }}
|
||||||
|
uses: actions/setup-python@v5
|
||||||
|
with:
|
||||||
|
python-version: ${{ matrix.python-version }}
|
||||||
|
cache: 'pip'
|
||||||
|
cache-dependency-path: '**/pyproject.toml'
|
||||||
|
|
||||||
|
# Cache Poetry dependencies
|
||||||
|
- name: Cache Poetry dependencies
|
||||||
|
uses: actions/cache@v4
|
||||||
|
with:
|
||||||
|
path: ~/.cache/pypoetry
|
||||||
|
key: ${{ runner.os }}-poetry-${{ matrix.python-version }}-${{ hashFiles('**/poetry.lock') }}
|
||||||
|
restore-keys: |
|
||||||
|
${{ runner.os }}-poetry-${{ matrix.python-version }}-
|
||||||
|
|
||||||
|
# Cache nox environments
|
||||||
|
- name: Cache nox environments
|
||||||
|
uses: actions/cache@v4
|
||||||
|
with:
|
||||||
|
path: .nox
|
||||||
|
key: ${{ runner.os }}-nox-${{ matrix.python-version }}-${{ hashFiles('**/noxfile.py') }}
|
||||||
|
restore-keys: |
|
||||||
|
${{ runner.os }}-nox-${{ matrix.python-version }}-
|
||||||
|
|
||||||
|
- name: Install uv
|
||||||
|
run: |
|
||||||
|
python -m pip install --upgrade pip
|
||||||
|
pip install uv
|
||||||
|
uv --version
|
||||||
|
|
||||||
|
- name: Install dependencies
|
||||||
|
run: |
|
||||||
|
uvx poetry install
|
||||||
|
|
||||||
|
- name: Lint with ruff
|
||||||
|
run: |
|
||||||
|
uvx nox -s lint
|
||||||
|
|
||||||
|
- name: Type check with mypy
|
||||||
|
run: |
|
||||||
|
uvx nox -s mypy
|
||||||
|
|
||||||
|
- name: Run tests
|
||||||
|
run: |
|
||||||
|
uvx nox -s pytest
|
||||||
|
|
||||||
|
- name: Upload coverage to Codecov
|
||||||
|
if: matrix.os == 'ubuntu-latest' && matrix.python-version == '3.11'
|
||||||
|
uses: codecov/codecov-action@v4
|
||||||
|
with:
|
||||||
|
file: ./coverage.xml
|
||||||
|
flags: unittests
|
||||||
|
name: codecov-umbrella
|
||||||
|
fail_ci_if_error: false
|
||||||
|
|
||||||
|
security:
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
steps:
|
||||||
|
- name: Checkout code
|
||||||
|
uses: actions/checkout@v4
|
||||||
|
|
||||||
|
- name: Set up Python
|
||||||
|
uses: actions/setup-python@v5
|
||||||
|
with:
|
||||||
|
python-version: "3.11"
|
||||||
|
|
||||||
|
- name: Install uv
|
||||||
|
run: |
|
||||||
|
python -m pip install --upgrade pip
|
||||||
|
pip install uv
|
||||||
|
|
||||||
|
- name: Install dependencies
|
||||||
|
run: |
|
||||||
|
uvx poetry install
|
||||||
|
|
||||||
|
- name: Run security checks
|
||||||
|
run: |
|
||||||
|
uvx nox -s safety
|
2
.gitignore
vendored
2
.gitignore
vendored
@ -20,3 +20,5 @@ target/*
|
|||||||
/build/
|
/build/
|
||||||
/coverage.xml
|
/coverage.xml
|
||||||
/.zip/
|
/.zip/
|
||||||
|
.env
|
||||||
|
|
||||||
|
@ -2,7 +2,7 @@ default_language_version:
|
|||||||
python: python3.10
|
python: python3.10
|
||||||
repos:
|
repos:
|
||||||
- repo: https://github.com/pre-commit/pre-commit-hooks
|
- repo: https://github.com/pre-commit/pre-commit-hooks
|
||||||
rev: v4.3.0
|
rev: v4.5.0
|
||||||
hooks:
|
hooks:
|
||||||
- id: no-commit-to-branch # prevent direct commits to main branch
|
- id: no-commit-to-branch # prevent direct commits to main branch
|
||||||
- id: check-yaml
|
- id: check-yaml
|
||||||
@ -10,3 +10,26 @@ repos:
|
|||||||
- id: check-toml
|
- id: check-toml
|
||||||
- id: end-of-file-fixer
|
- id: end-of-file-fixer
|
||||||
- id: trailing-whitespace
|
- id: trailing-whitespace
|
||||||
|
- id: check-added-large-files
|
||||||
|
- id: check-merge-conflict
|
||||||
|
- id: debug-statements
|
||||||
|
|
||||||
|
- repo: https://github.com/astral-sh/ruff-pre-commit
|
||||||
|
rev: v0.1.8
|
||||||
|
hooks:
|
||||||
|
- id: ruff
|
||||||
|
args: [--fix, --exit-non-zero-on-fix]
|
||||||
|
- id: ruff-format
|
||||||
|
|
||||||
|
- repo: https://github.com/pycqa/isort
|
||||||
|
rev: 5.13.2
|
||||||
|
hooks:
|
||||||
|
- id: isort
|
||||||
|
args: ["--profile", "black"]
|
||||||
|
|
||||||
|
- repo: https://github.com/pre-commit/mirrors-mypy
|
||||||
|
rev: v1.8.0
|
||||||
|
hooks:
|
||||||
|
- id: mypy
|
||||||
|
additional_dependencies: [types-requests]
|
||||||
|
args: [--ignore-missing-imports]
|
||||||
|
84
README.md
84
README.md
@ -1,2 +1,82 @@
|
|||||||
# repo-template
|
# PyPI Query MCP Server
|
||||||
This is a boilerplate git repository for creating new project
|
|
||||||
|
A Model Context Protocol (MCP) server for querying PyPI package information, dependencies, and compatibility checking.
|
||||||
|
|
||||||
|
## Features
|
||||||
|
|
||||||
|
- 📦 Query PyPI package information (name, version, description, dependencies)
|
||||||
|
- 🐍 Python version compatibility checking
|
||||||
|
- 🔍 Dependency analysis and resolution
|
||||||
|
- 🏢 Private PyPI repository support
|
||||||
|
- ⚡ Fast async operations with caching
|
||||||
|
- 🛠️ Easy integration with MCP clients
|
||||||
|
|
||||||
|
## Quick Start
|
||||||
|
|
||||||
|
### Installation
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# Install from PyPI (coming soon)
|
||||||
|
pip install pypi-query-mcp-server
|
||||||
|
|
||||||
|
# Or install from source
|
||||||
|
git clone https://github.com/loonghao/pypi-query-mcp-server.git
|
||||||
|
cd pypi-query-mcp-server
|
||||||
|
poetry install
|
||||||
|
```
|
||||||
|
|
||||||
|
### Usage
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# Start the MCP server
|
||||||
|
pypi-query-mcp
|
||||||
|
|
||||||
|
# Or run directly with Python
|
||||||
|
python -m pypi_query_mcp.server
|
||||||
|
```
|
||||||
|
|
||||||
|
### Available MCP Tools
|
||||||
|
|
||||||
|
The server provides the following MCP tools:
|
||||||
|
|
||||||
|
1. **get_package_info** - Get comprehensive package information
|
||||||
|
2. **get_package_versions** - List all available versions for a package
|
||||||
|
3. **get_package_dependencies** - Analyze package dependencies
|
||||||
|
4. **check_package_python_compatibility** - Check Python version compatibility
|
||||||
|
5. **get_package_compatible_python_versions** - Get all compatible Python versions
|
||||||
|
|
||||||
|
### Example Usage with MCP Client
|
||||||
|
|
||||||
|
```python
|
||||||
|
# Example: Check if Django is compatible with Python 3.9
|
||||||
|
result = await mcp_client.call_tool("check_package_python_compatibility", {
|
||||||
|
"package_name": "django",
|
||||||
|
"target_python_version": "3.9"
|
||||||
|
})
|
||||||
|
|
||||||
|
# Example: Get package information
|
||||||
|
info = await mcp_client.call_tool("get_package_info", {
|
||||||
|
"package_name": "requests"
|
||||||
|
})
|
||||||
|
```
|
||||||
|
|
||||||
|
## Development Status
|
||||||
|
|
||||||
|
🎉 **Core functionality implemented and ready for use!**
|
||||||
|
|
||||||
|
Current implementation status:
|
||||||
|
- ✅ Basic project structure
|
||||||
|
- ✅ PyPI API client with caching
|
||||||
|
- ✅ MCP tools implementation (package info, versions, dependencies)
|
||||||
|
- ✅ Python version compatibility checking
|
||||||
|
- ✅ CI/CD pipeline with multi-platform testing
|
||||||
|
- ⏳ Private repository support (planned)
|
||||||
|
- ⏳ Advanced dependency analysis (planned)
|
||||||
|
|
||||||
|
## Contributing
|
||||||
|
|
||||||
|
Contributions are welcome! Please feel free to submit a Pull Request.
|
||||||
|
|
||||||
|
## License
|
||||||
|
|
||||||
|
This project is licensed under the MIT License - see the [LICENSE](LICENSE) file for details.
|
||||||
|
BIN
dist/pypi_query_mcp_server-0.1.0-py3-none-any.whl
vendored
Normal file
BIN
dist/pypi_query_mcp_server-0.1.0-py3-none-any.whl
vendored
Normal file
Binary file not shown.
BIN
dist/pypi_query_mcp_server-0.1.0.tar.gz
vendored
Normal file
BIN
dist/pypi_query_mcp_server-0.1.0.tar.gz
vendored
Normal file
Binary file not shown.
@ -8,10 +8,26 @@ from nox_actions.utils import THIS_ROOT
|
|||||||
|
|
||||||
|
|
||||||
def pytest(session: nox.Session) -> None:
|
def pytest(session: nox.Session) -> None:
|
||||||
|
"""Run pytest with coverage reporting."""
|
||||||
session.install(".")
|
session.install(".")
|
||||||
session.install("pytest", "pytest_cov", "pytest_mock")
|
session.install("pytest", "pytest-cov", "pytest-mock", "pytest-asyncio")
|
||||||
test_root = os.path.join(THIS_ROOT, "tests")
|
test_root = os.path.join(THIS_ROOT, "tests")
|
||||||
session.run("pytest", f"--cov={PACKAGE_NAME}",
|
session.run("pytest", f"--cov={PACKAGE_NAME}",
|
||||||
"--cov-report=xml:coverage.xml",
|
"--cov-report=xml:coverage.xml",
|
||||||
|
"--cov-report=term-missing",
|
||||||
f"--rootdir={test_root}",
|
f"--rootdir={test_root}",
|
||||||
env={"PYTHONPATH": THIS_ROOT.as_posix()})
|
env={"PYTHONPATH": THIS_ROOT.as_posix()})
|
||||||
|
|
||||||
|
|
||||||
|
def mypy(session: nox.Session) -> None:
|
||||||
|
"""Run mypy type checking."""
|
||||||
|
session.install(".")
|
||||||
|
session.install("mypy", "types-requests")
|
||||||
|
session.run("mypy", PACKAGE_NAME, "--ignore-missing-imports")
|
||||||
|
|
||||||
|
|
||||||
|
def safety(session: nox.Session) -> None:
|
||||||
|
"""Run safety security checks."""
|
||||||
|
session.install(".")
|
||||||
|
session.install("safety")
|
||||||
|
session.run("safety", "check", "--json")
|
||||||
|
@ -10,6 +10,13 @@ from nox_actions.utils import PACKAGE_NAME
|
|||||||
from nox_actions.utils import THIS_ROOT
|
from nox_actions.utils import THIS_ROOT
|
||||||
|
|
||||||
|
|
||||||
|
def build(session: nox.Session) -> None:
|
||||||
|
"""Build Python package distributions."""
|
||||||
|
session.install("build", "twine")
|
||||||
|
session.run("python", "-m", "build")
|
||||||
|
session.run("twine", "check", "dist/*")
|
||||||
|
|
||||||
|
|
||||||
@nox.session(name="build-exe", reuse_venv=True)
|
@nox.session(name="build-exe", reuse_venv=True)
|
||||||
def build_exe(session: nox.Session) -> None:
|
def build_exe(session: nox.Session) -> None:
|
||||||
parser = argparse.ArgumentParser(prog="nox -s build-exe --release")
|
parser = argparse.ArgumentParser(prog="nox -s build-exe --release")
|
||||||
|
@ -2,7 +2,7 @@
|
|||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
|
|
||||||
|
|
||||||
PACKAGE_NAME = ""
|
PACKAGE_NAME = "pypi_query_mcp"
|
||||||
THIS_ROOT = Path(__file__).parent.parent
|
THIS_ROOT = Path(__file__).parent.parent
|
||||||
PROJECT_ROOT = THIS_ROOT.parent
|
PROJECT_ROOT = THIS_ROOT.parent
|
||||||
|
|
||||||
|
@ -8,7 +8,7 @@ import nox
|
|||||||
|
|
||||||
ROOT = os.path.dirname(__file__)
|
ROOT = os.path.dirname(__file__)
|
||||||
|
|
||||||
# Ensure maya_umbrella is importable.
|
# Ensure pypi_query_mcp is importable.
|
||||||
if ROOT not in sys.path:
|
if ROOT not in sys.path:
|
||||||
sys.path.append(ROOT)
|
sys.path.append(ROOT)
|
||||||
|
|
||||||
@ -18,7 +18,10 @@ from nox_actions import lint # noqa: E402
|
|||||||
from nox_actions import release # noqa: E402
|
from nox_actions import release # noqa: E402
|
||||||
|
|
||||||
|
|
||||||
|
# Configure nox sessions
|
||||||
nox.session(lint.lint, name="lint")
|
nox.session(lint.lint, name="lint")
|
||||||
nox.session(lint.lint_fix, name="lint-fix")
|
nox.session(lint.lint_fix, name="lint-fix")
|
||||||
nox.session(codetest.pytest, name="pytest")
|
nox.session(codetest.pytest, name="pytest")
|
||||||
nox.session(release.build_exe, name="build-exe")
|
nox.session(codetest.mypy, name="mypy")
|
||||||
|
nox.session(codetest.safety, name="safety")
|
||||||
|
nox.session(release.build, name="build")
|
||||||
|
1201
poetry.lock
generated
Normal file
1201
poetry.lock
generated
Normal file
File diff suppressed because it is too large
Load Diff
13
pypi_query_mcp/__init__.py
Normal file
13
pypi_query_mcp/__init__.py
Normal file
@ -0,0 +1,13 @@
|
|||||||
|
"""PyPI Query MCP Server.
|
||||||
|
|
||||||
|
A Model Context Protocol (MCP) server for querying PyPI package information,
|
||||||
|
dependencies, and compatibility checking.
|
||||||
|
"""
|
||||||
|
|
||||||
|
__version__ = "0.1.0"
|
||||||
|
__author__ = "Hal"
|
||||||
|
__email__ = "hal.long@outlook.com"
|
||||||
|
|
||||||
|
from pypi_query_mcp.server import app
|
||||||
|
|
||||||
|
__all__ = ["app", "__version__"]
|
8
pypi_query_mcp/config/__init__.py
Normal file
8
pypi_query_mcp/config/__init__.py
Normal file
@ -0,0 +1,8 @@
|
|||||||
|
"""Configuration management for PyPI Query MCP Server.
|
||||||
|
|
||||||
|
This package handles configuration loading, validation, and management
|
||||||
|
for the MCP server, including private registry settings.
|
||||||
|
"""
|
||||||
|
|
||||||
|
# Configuration exports will be added as modules are implemented
|
||||||
|
__all__ = []
|
27
pypi_query_mcp/core/__init__.py
Normal file
27
pypi_query_mcp/core/__init__.py
Normal file
@ -0,0 +1,27 @@
|
|||||||
|
"""Core modules for PyPI Query MCP Server.
|
||||||
|
|
||||||
|
This package contains the core business logic for PyPI package queries,
|
||||||
|
including API clients, data processing, and utility functions.
|
||||||
|
"""
|
||||||
|
|
||||||
|
from .exceptions import (
|
||||||
|
InvalidPackageNameError,
|
||||||
|
NetworkError,
|
||||||
|
PackageNotFoundError,
|
||||||
|
PyPIError,
|
||||||
|
PyPIServerError,
|
||||||
|
RateLimitError,
|
||||||
|
)
|
||||||
|
from .pypi_client import PyPIClient
|
||||||
|
from .version_utils import VersionCompatibility
|
||||||
|
|
||||||
|
__all__ = [
|
||||||
|
"PyPIClient",
|
||||||
|
"VersionCompatibility",
|
||||||
|
"PyPIError",
|
||||||
|
"PackageNotFoundError",
|
||||||
|
"NetworkError",
|
||||||
|
"RateLimitError",
|
||||||
|
"InvalidPackageNameError",
|
||||||
|
"PyPIServerError",
|
||||||
|
]
|
56
pypi_query_mcp/core/exceptions.py
Normal file
56
pypi_query_mcp/core/exceptions.py
Normal file
@ -0,0 +1,56 @@
|
|||||||
|
"""Custom exceptions for PyPI Query MCP Server."""
|
||||||
|
|
||||||
|
|
||||||
|
class PyPIError(Exception):
|
||||||
|
"""Base exception for PyPI-related errors."""
|
||||||
|
|
||||||
|
def __init__(self, message: str, status_code: int = None):
|
||||||
|
super().__init__(message)
|
||||||
|
self.message = message
|
||||||
|
self.status_code = status_code
|
||||||
|
|
||||||
|
|
||||||
|
class PackageNotFoundError(PyPIError):
|
||||||
|
"""Raised when a package is not found on PyPI."""
|
||||||
|
|
||||||
|
def __init__(self, package_name: str):
|
||||||
|
message = f"Package '{package_name}' not found on PyPI"
|
||||||
|
super().__init__(message, status_code=404)
|
||||||
|
self.package_name = package_name
|
||||||
|
|
||||||
|
|
||||||
|
class NetworkError(PyPIError):
|
||||||
|
"""Raised when network-related errors occur."""
|
||||||
|
|
||||||
|
def __init__(self, message: str, original_error: Exception = None):
|
||||||
|
super().__init__(message)
|
||||||
|
self.original_error = original_error
|
||||||
|
|
||||||
|
|
||||||
|
class RateLimitError(PyPIError):
|
||||||
|
"""Raised when API rate limit is exceeded."""
|
||||||
|
|
||||||
|
def __init__(self, retry_after: int = None):
|
||||||
|
message = "PyPI API rate limit exceeded"
|
||||||
|
if retry_after:
|
||||||
|
message += f". Retry after {retry_after} seconds"
|
||||||
|
super().__init__(message, status_code=429)
|
||||||
|
self.retry_after = retry_after
|
||||||
|
|
||||||
|
|
||||||
|
class InvalidPackageNameError(PyPIError):
|
||||||
|
"""Raised when package name is invalid."""
|
||||||
|
|
||||||
|
def __init__(self, package_name: str):
|
||||||
|
message = f"Invalid package name: '{package_name}'"
|
||||||
|
super().__init__(message, status_code=400)
|
||||||
|
self.package_name = package_name
|
||||||
|
|
||||||
|
|
||||||
|
class PyPIServerError(PyPIError):
|
||||||
|
"""Raised when PyPI server returns a server error."""
|
||||||
|
|
||||||
|
def __init__(self, status_code: int, message: str = None):
|
||||||
|
if not message:
|
||||||
|
message = f"PyPI server error (HTTP {status_code})"
|
||||||
|
super().__init__(message, status_code=status_code)
|
238
pypi_query_mcp/core/pypi_client.py
Normal file
238
pypi_query_mcp/core/pypi_client.py
Normal file
@ -0,0 +1,238 @@
|
|||||||
|
"""PyPI API client for package information retrieval."""
|
||||||
|
|
||||||
|
import asyncio
|
||||||
|
import logging
|
||||||
|
import re
|
||||||
|
from typing import Any
|
||||||
|
from urllib.parse import quote
|
||||||
|
|
||||||
|
import httpx
|
||||||
|
|
||||||
|
from .exceptions import (
|
||||||
|
InvalidPackageNameError,
|
||||||
|
NetworkError,
|
||||||
|
PackageNotFoundError,
|
||||||
|
PyPIServerError,
|
||||||
|
RateLimitError,
|
||||||
|
)
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
|
class PyPIClient:
|
||||||
|
"""Async client for PyPI JSON API."""
|
||||||
|
|
||||||
|
def __init__(
|
||||||
|
self,
|
||||||
|
base_url: str = "https://pypi.org/pypi",
|
||||||
|
timeout: float = 30.0,
|
||||||
|
max_retries: int = 3,
|
||||||
|
retry_delay: float = 1.0,
|
||||||
|
):
|
||||||
|
"""Initialize PyPI client.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
base_url: Base URL for PyPI API
|
||||||
|
timeout: Request timeout in seconds
|
||||||
|
max_retries: Maximum number of retry attempts
|
||||||
|
retry_delay: Delay between retries in seconds
|
||||||
|
"""
|
||||||
|
self.base_url = base_url.rstrip("/")
|
||||||
|
self.timeout = timeout
|
||||||
|
self.max_retries = max_retries
|
||||||
|
self.retry_delay = retry_delay
|
||||||
|
|
||||||
|
# Simple in-memory cache
|
||||||
|
self._cache: dict[str, dict[str, Any]] = {}
|
||||||
|
self._cache_ttl = 300 # 5 minutes
|
||||||
|
|
||||||
|
# HTTP client configuration
|
||||||
|
self._client = httpx.AsyncClient(
|
||||||
|
timeout=httpx.Timeout(timeout),
|
||||||
|
headers={
|
||||||
|
"User-Agent": "pypi-query-mcp-server/0.1.0",
|
||||||
|
"Accept": "application/json",
|
||||||
|
},
|
||||||
|
follow_redirects=True,
|
||||||
|
)
|
||||||
|
|
||||||
|
async def __aenter__(self):
|
||||||
|
"""Async context manager entry."""
|
||||||
|
return self
|
||||||
|
|
||||||
|
async def __aexit__(self, exc_type, exc_val, exc_tb):
|
||||||
|
"""Async context manager exit."""
|
||||||
|
await self.close()
|
||||||
|
|
||||||
|
async def close(self):
|
||||||
|
"""Close the HTTP client."""
|
||||||
|
await self._client.aclose()
|
||||||
|
|
||||||
|
def _validate_package_name(self, package_name: str) -> str:
|
||||||
|
"""Validate and normalize package name.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
package_name: Package name to validate
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Normalized package name
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
InvalidPackageNameError: If package name is invalid
|
||||||
|
"""
|
||||||
|
if not package_name or not package_name.strip():
|
||||||
|
raise InvalidPackageNameError(package_name)
|
||||||
|
|
||||||
|
# Normalize package name (convert to lowercase, replace _ with -)
|
||||||
|
normalized = re.sub(r"[-_.]+", "-", package_name.lower())
|
||||||
|
|
||||||
|
# Basic validation - package names should contain only alphanumeric, hyphens, dots, underscores
|
||||||
|
if not re.match(r"^[a-zA-Z0-9]([a-zA-Z0-9._-]*[a-zA-Z0-9])?$", package_name):
|
||||||
|
raise InvalidPackageNameError(package_name)
|
||||||
|
|
||||||
|
return normalized
|
||||||
|
|
||||||
|
def _get_cache_key(self, package_name: str, endpoint: str = "info") -> str:
|
||||||
|
"""Generate cache key for package data."""
|
||||||
|
return f"{endpoint}:{package_name}"
|
||||||
|
|
||||||
|
def _is_cache_valid(self, cache_entry: dict[str, Any]) -> bool:
|
||||||
|
"""Check if cache entry is still valid."""
|
||||||
|
import time
|
||||||
|
return time.time() - cache_entry.get("timestamp", 0) < self._cache_ttl
|
||||||
|
|
||||||
|
async def _make_request(self, url: str) -> dict[str, Any]:
|
||||||
|
"""Make HTTP request with retry logic.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
url: URL to request
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
JSON response data
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
NetworkError: For network-related errors
|
||||||
|
PackageNotFoundError: When package is not found
|
||||||
|
RateLimitError: When rate limit is exceeded
|
||||||
|
PyPIServerError: For server errors
|
||||||
|
"""
|
||||||
|
last_exception = None
|
||||||
|
|
||||||
|
for attempt in range(self.max_retries + 1):
|
||||||
|
try:
|
||||||
|
logger.debug(f"Making request to {url} (attempt {attempt + 1})")
|
||||||
|
|
||||||
|
response = await self._client.get(url)
|
||||||
|
|
||||||
|
# Handle different HTTP status codes
|
||||||
|
if response.status_code == 200:
|
||||||
|
return response.json()
|
||||||
|
elif response.status_code == 404:
|
||||||
|
# Extract package name from URL for better error message
|
||||||
|
package_name = url.split("/")[-2] if "/" in url else "unknown"
|
||||||
|
raise PackageNotFoundError(package_name)
|
||||||
|
elif response.status_code == 429:
|
||||||
|
retry_after = response.headers.get("Retry-After")
|
||||||
|
retry_after_int = int(retry_after) if retry_after else None
|
||||||
|
raise RateLimitError(retry_after_int)
|
||||||
|
elif response.status_code >= 500:
|
||||||
|
raise PyPIServerError(response.status_code)
|
||||||
|
else:
|
||||||
|
raise PyPIServerError(
|
||||||
|
response.status_code,
|
||||||
|
f"Unexpected status code: {response.status_code}"
|
||||||
|
)
|
||||||
|
|
||||||
|
except httpx.TimeoutException as e:
|
||||||
|
last_exception = NetworkError(f"Request timeout: {e}", e)
|
||||||
|
except httpx.NetworkError as e:
|
||||||
|
last_exception = NetworkError(f"Network error: {e}", e)
|
||||||
|
except (PackageNotFoundError, RateLimitError, PyPIServerError):
|
||||||
|
# Don't retry these errors
|
||||||
|
raise
|
||||||
|
except Exception as e:
|
||||||
|
last_exception = NetworkError(f"Unexpected error: {e}", e)
|
||||||
|
|
||||||
|
# Wait before retry (except on last attempt)
|
||||||
|
if attempt < self.max_retries:
|
||||||
|
await asyncio.sleep(self.retry_delay * (2 ** attempt)) # Exponential backoff
|
||||||
|
|
||||||
|
# If we get here, all retries failed
|
||||||
|
raise last_exception
|
||||||
|
|
||||||
|
async def get_package_info(self, package_name: str, use_cache: bool = True) -> dict[str, Any]:
|
||||||
|
"""Get comprehensive package information from PyPI.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
package_name: Name of the package to query
|
||||||
|
use_cache: Whether to use cached data if available
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Dictionary containing package information
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
InvalidPackageNameError: If package name is invalid
|
||||||
|
PackageNotFoundError: If package is not found
|
||||||
|
NetworkError: For network-related errors
|
||||||
|
"""
|
||||||
|
normalized_name = self._validate_package_name(package_name)
|
||||||
|
cache_key = self._get_cache_key(normalized_name, "info")
|
||||||
|
|
||||||
|
# Check cache first
|
||||||
|
if use_cache and cache_key in self._cache:
|
||||||
|
cache_entry = self._cache[cache_key]
|
||||||
|
if self._is_cache_valid(cache_entry):
|
||||||
|
logger.debug(f"Using cached data for package: {normalized_name}")
|
||||||
|
return cache_entry["data"]
|
||||||
|
|
||||||
|
# Make API request
|
||||||
|
url = f"{self.base_url}/{quote(normalized_name)}/json"
|
||||||
|
logger.info(f"Fetching package info for: {normalized_name}")
|
||||||
|
|
||||||
|
try:
|
||||||
|
data = await self._make_request(url)
|
||||||
|
|
||||||
|
# Cache the result
|
||||||
|
import time
|
||||||
|
self._cache[cache_key] = {
|
||||||
|
"data": data,
|
||||||
|
"timestamp": time.time()
|
||||||
|
}
|
||||||
|
|
||||||
|
return data
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"Failed to fetch package info for {normalized_name}: {e}")
|
||||||
|
raise
|
||||||
|
|
||||||
|
async def get_package_versions(self, package_name: str, use_cache: bool = True) -> list[str]:
|
||||||
|
"""Get list of available versions for a package.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
package_name: Name of the package to query
|
||||||
|
use_cache: Whether to use cached data if available
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
List of version strings
|
||||||
|
"""
|
||||||
|
package_info = await self.get_package_info(package_name, use_cache)
|
||||||
|
releases = package_info.get("releases", {})
|
||||||
|
return list(releases.keys())
|
||||||
|
|
||||||
|
async def get_latest_version(self, package_name: str, use_cache: bool = True) -> str:
|
||||||
|
"""Get the latest version of a package.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
package_name: Name of the package to query
|
||||||
|
use_cache: Whether to use cached data if available
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Latest version string
|
||||||
|
"""
|
||||||
|
package_info = await self.get_package_info(package_name, use_cache)
|
||||||
|
return package_info.get("info", {}).get("version", "")
|
||||||
|
|
||||||
|
def clear_cache(self):
|
||||||
|
"""Clear the internal cache."""
|
||||||
|
self._cache.clear()
|
||||||
|
logger.debug("Cache cleared")
|
274
pypi_query_mcp/core/version_utils.py
Normal file
274
pypi_query_mcp/core/version_utils.py
Normal file
@ -0,0 +1,274 @@
|
|||||||
|
"""Version parsing and compatibility checking utilities."""
|
||||||
|
|
||||||
|
import logging
|
||||||
|
import re
|
||||||
|
from typing import Any
|
||||||
|
|
||||||
|
from packaging.specifiers import SpecifierSet
|
||||||
|
from packaging.version import InvalidVersion, Version
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
|
class VersionCompatibility:
|
||||||
|
"""Utility class for Python version compatibility checking."""
|
||||||
|
|
||||||
|
def __init__(self):
|
||||||
|
"""Initialize version compatibility checker."""
|
||||||
|
# Common Python version patterns in classifiers
|
||||||
|
self.python_classifier_pattern = re.compile(
|
||||||
|
r"Programming Language :: Python :: (\d+(?:\.\d+)*)"
|
||||||
|
)
|
||||||
|
|
||||||
|
# Implementation-specific classifiers
|
||||||
|
self.implementation_pattern = re.compile(
|
||||||
|
r"Programming Language :: Python :: Implementation :: (\w+)"
|
||||||
|
)
|
||||||
|
|
||||||
|
def parse_requires_python(self, requires_python: str) -> SpecifierSet | None:
|
||||||
|
"""Parse requires_python field into a SpecifierSet.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
requires_python: The requires_python string from package metadata
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
SpecifierSet object or None if parsing fails
|
||||||
|
"""
|
||||||
|
if not requires_python or not requires_python.strip():
|
||||||
|
return None
|
||||||
|
|
||||||
|
try:
|
||||||
|
# Clean up the version specification
|
||||||
|
cleaned = requires_python.strip()
|
||||||
|
return SpecifierSet(cleaned)
|
||||||
|
except Exception as e:
|
||||||
|
logger.warning(f"Failed to parse requires_python '{requires_python}': {e}")
|
||||||
|
return None
|
||||||
|
|
||||||
|
def extract_python_versions_from_classifiers(self, classifiers: list[str]) -> set[str]:
|
||||||
|
"""Extract Python version information from classifiers.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
classifiers: List of classifier strings
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Set of Python version strings
|
||||||
|
"""
|
||||||
|
versions = set()
|
||||||
|
|
||||||
|
for classifier in classifiers:
|
||||||
|
match = self.python_classifier_pattern.search(classifier)
|
||||||
|
if match:
|
||||||
|
version = match.group(1)
|
||||||
|
versions.add(version)
|
||||||
|
|
||||||
|
return versions
|
||||||
|
|
||||||
|
def extract_python_implementations(self, classifiers: list[str]) -> set[str]:
|
||||||
|
"""Extract Python implementation information from classifiers.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
classifiers: List of classifier strings
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Set of Python implementation names (CPython, PyPy, etc.)
|
||||||
|
"""
|
||||||
|
implementations = set()
|
||||||
|
|
||||||
|
for classifier in classifiers:
|
||||||
|
match = self.implementation_pattern.search(classifier)
|
||||||
|
if match:
|
||||||
|
implementation = match.group(1)
|
||||||
|
implementations.add(implementation)
|
||||||
|
|
||||||
|
return implementations
|
||||||
|
|
||||||
|
def check_version_compatibility(
|
||||||
|
self,
|
||||||
|
target_version: str,
|
||||||
|
requires_python: str | None = None,
|
||||||
|
classifiers: list[str] | None = None
|
||||||
|
) -> dict[str, Any]:
|
||||||
|
"""Check if a target Python version is compatible with package requirements.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
target_version: Target Python version (e.g., "3.9", "3.10.5")
|
||||||
|
requires_python: The requires_python specification
|
||||||
|
classifiers: List of package classifiers
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Dictionary containing compatibility information
|
||||||
|
"""
|
||||||
|
result = {
|
||||||
|
"target_version": target_version,
|
||||||
|
"is_compatible": False,
|
||||||
|
"compatibility_source": None,
|
||||||
|
"details": {},
|
||||||
|
"warnings": [],
|
||||||
|
"suggestions": []
|
||||||
|
}
|
||||||
|
|
||||||
|
try:
|
||||||
|
target_ver = Version(target_version)
|
||||||
|
except InvalidVersion as e:
|
||||||
|
result["warnings"].append(f"Invalid target version format: {e}")
|
||||||
|
return result
|
||||||
|
|
||||||
|
# Check requires_python first (more authoritative)
|
||||||
|
if requires_python:
|
||||||
|
spec_set = self.parse_requires_python(requires_python)
|
||||||
|
if spec_set:
|
||||||
|
is_compatible = target_ver in spec_set
|
||||||
|
result.update({
|
||||||
|
"is_compatible": is_compatible,
|
||||||
|
"compatibility_source": "requires_python",
|
||||||
|
"details": {
|
||||||
|
"requires_python": requires_python,
|
||||||
|
"parsed_spec": str(spec_set),
|
||||||
|
"check_result": is_compatible
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
if not is_compatible:
|
||||||
|
result["suggestions"].append(
|
||||||
|
f"Package requires Python {requires_python}, "
|
||||||
|
f"but target is {target_version}"
|
||||||
|
)
|
||||||
|
|
||||||
|
return result
|
||||||
|
|
||||||
|
# Fall back to classifiers if no requires_python
|
||||||
|
if classifiers:
|
||||||
|
supported_versions = self.extract_python_versions_from_classifiers(classifiers)
|
||||||
|
implementations = self.extract_python_implementations(classifiers)
|
||||||
|
|
||||||
|
if supported_versions:
|
||||||
|
# Check if target version matches any supported version
|
||||||
|
target_major_minor = f"{target_ver.major}.{target_ver.minor}"
|
||||||
|
target_major = str(target_ver.major)
|
||||||
|
|
||||||
|
is_compatible = (
|
||||||
|
target_version in supported_versions or
|
||||||
|
target_major_minor in supported_versions or
|
||||||
|
target_major in supported_versions
|
||||||
|
)
|
||||||
|
|
||||||
|
result.update({
|
||||||
|
"is_compatible": is_compatible,
|
||||||
|
"compatibility_source": "classifiers",
|
||||||
|
"details": {
|
||||||
|
"supported_versions": sorted(supported_versions),
|
||||||
|
"implementations": sorted(implementations),
|
||||||
|
"target_major_minor": target_major_minor,
|
||||||
|
"check_result": is_compatible
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
if not is_compatible:
|
||||||
|
result["suggestions"].append(
|
||||||
|
f"Package supports Python versions: {', '.join(sorted(supported_versions))}, "
|
||||||
|
f"but target is {target_version}"
|
||||||
|
)
|
||||||
|
|
||||||
|
return result
|
||||||
|
|
||||||
|
# No version information available
|
||||||
|
result["warnings"].append(
|
||||||
|
"No Python version requirements found in package metadata"
|
||||||
|
)
|
||||||
|
result["suggestions"].append(
|
||||||
|
"Consider checking package documentation for Python version compatibility"
|
||||||
|
)
|
||||||
|
|
||||||
|
return result
|
||||||
|
|
||||||
|
def get_compatible_versions(
|
||||||
|
self,
|
||||||
|
requires_python: str | None = None,
|
||||||
|
classifiers: list[str] | None = None,
|
||||||
|
available_pythons: list[str] | None = None
|
||||||
|
) -> dict[str, Any]:
|
||||||
|
"""Get list of compatible Python versions for a package.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
requires_python: The requires_python specification
|
||||||
|
classifiers: List of package classifiers
|
||||||
|
available_pythons: List of Python versions to check against
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Dictionary containing compatible versions and recommendations
|
||||||
|
"""
|
||||||
|
if available_pythons is None:
|
||||||
|
# Default Python versions to check
|
||||||
|
available_pythons = [
|
||||||
|
"3.7", "3.8", "3.9", "3.10", "3.11", "3.12", "3.13"
|
||||||
|
]
|
||||||
|
|
||||||
|
compatible = []
|
||||||
|
incompatible = []
|
||||||
|
|
||||||
|
for python_version in available_pythons:
|
||||||
|
result = self.check_version_compatibility(
|
||||||
|
python_version, requires_python, classifiers
|
||||||
|
)
|
||||||
|
|
||||||
|
if result["is_compatible"]:
|
||||||
|
compatible.append({
|
||||||
|
"version": python_version,
|
||||||
|
"source": result["compatibility_source"]
|
||||||
|
})
|
||||||
|
else:
|
||||||
|
incompatible.append({
|
||||||
|
"version": python_version,
|
||||||
|
"reason": result["suggestions"][0] if result["suggestions"] else "Unknown"
|
||||||
|
})
|
||||||
|
|
||||||
|
return {
|
||||||
|
"compatible_versions": compatible,
|
||||||
|
"incompatible_versions": incompatible,
|
||||||
|
"total_checked": len(available_pythons),
|
||||||
|
"compatibility_rate": len(compatible) / len(available_pythons) if available_pythons else 0,
|
||||||
|
"recommendations": self._generate_recommendations(compatible, incompatible)
|
||||||
|
}
|
||||||
|
|
||||||
|
def _generate_recommendations(
|
||||||
|
self,
|
||||||
|
compatible: list[dict[str, Any]],
|
||||||
|
incompatible: list[dict[str, Any]]
|
||||||
|
) -> list[str]:
|
||||||
|
"""Generate recommendations based on compatibility results.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
compatible: List of compatible versions
|
||||||
|
incompatible: List of incompatible versions
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
List of recommendation strings
|
||||||
|
"""
|
||||||
|
recommendations = []
|
||||||
|
|
||||||
|
if not compatible:
|
||||||
|
recommendations.append(
|
||||||
|
"⚠️ No compatible Python versions found. "
|
||||||
|
"Check package documentation for requirements."
|
||||||
|
)
|
||||||
|
elif len(compatible) == 1:
|
||||||
|
version = compatible[0]["version"]
|
||||||
|
recommendations.append(
|
||||||
|
f"📌 Only Python {version} is compatible with this package."
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
versions = [v["version"] for v in compatible]
|
||||||
|
latest = max(versions, key=lambda x: tuple(map(int, x.split("."))))
|
||||||
|
recommendations.append(
|
||||||
|
f"✅ Compatible with Python {', '.join(versions)}. "
|
||||||
|
f"Recommended: Python {latest}"
|
||||||
|
)
|
||||||
|
|
||||||
|
if len(incompatible) > len(compatible):
|
||||||
|
recommendations.append(
|
||||||
|
"⚠️ This package has limited Python version support. "
|
||||||
|
"Consider using a more recent version of the package if available."
|
||||||
|
)
|
||||||
|
|
||||||
|
return recommendations
|
295
pypi_query_mcp/server.py
Normal file
295
pypi_query_mcp/server.py
Normal file
@ -0,0 +1,295 @@
|
|||||||
|
"""FastMCP server for PyPI package queries."""
|
||||||
|
|
||||||
|
import logging
|
||||||
|
from typing import Any
|
||||||
|
|
||||||
|
import click
|
||||||
|
from fastmcp import FastMCP
|
||||||
|
|
||||||
|
from .core.exceptions import InvalidPackageNameError, NetworkError, PackageNotFoundError
|
||||||
|
from .tools import (
|
||||||
|
check_python_compatibility,
|
||||||
|
get_compatible_python_versions,
|
||||||
|
query_package_dependencies,
|
||||||
|
query_package_info,
|
||||||
|
query_package_versions,
|
||||||
|
)
|
||||||
|
|
||||||
|
# Configure logging
|
||||||
|
logging.basicConfig(
|
||||||
|
level=logging.INFO,
|
||||||
|
format="%(asctime)s - %(name)s - %(levelname)s - %(message)s"
|
||||||
|
)
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
# Create FastMCP application
|
||||||
|
app = FastMCP("PyPI Query MCP Server")
|
||||||
|
|
||||||
|
|
||||||
|
@app.tool()
|
||||||
|
async def get_package_info(package_name: str) -> dict[str, Any]:
|
||||||
|
"""Query comprehensive information about a PyPI package.
|
||||||
|
|
||||||
|
This tool retrieves detailed information about a Python package from PyPI,
|
||||||
|
including metadata, description, author information, dependencies, and more.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
package_name: The name of the PyPI package to query (e.g., 'requests', 'django')
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Dictionary containing comprehensive package information including:
|
||||||
|
- Basic metadata (name, version, summary, description)
|
||||||
|
- Author and maintainer information
|
||||||
|
- License and project URLs
|
||||||
|
- Python version requirements
|
||||||
|
- Dependencies and classifiers
|
||||||
|
- Version history summary
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
InvalidPackageNameError: If package name is empty or invalid
|
||||||
|
PackageNotFoundError: If package is not found on PyPI
|
||||||
|
NetworkError: For network-related errors
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
logger.info(f"MCP tool: Querying package info for {package_name}")
|
||||||
|
result = await query_package_info(package_name)
|
||||||
|
logger.info(f"Successfully retrieved info for package: {package_name}")
|
||||||
|
return result
|
||||||
|
except (InvalidPackageNameError, PackageNotFoundError, NetworkError) as e:
|
||||||
|
logger.error(f"Error querying package {package_name}: {e}")
|
||||||
|
return {
|
||||||
|
"error": str(e),
|
||||||
|
"error_type": type(e).__name__,
|
||||||
|
"package_name": package_name
|
||||||
|
}
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"Unexpected error querying package {package_name}: {e}")
|
||||||
|
return {
|
||||||
|
"error": f"Unexpected error: {e}",
|
||||||
|
"error_type": "UnexpectedError",
|
||||||
|
"package_name": package_name
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
@app.tool()
|
||||||
|
async def get_package_versions(package_name: str) -> dict[str, Any]:
|
||||||
|
"""Get version information for a PyPI package.
|
||||||
|
|
||||||
|
This tool retrieves comprehensive version information for a Python package,
|
||||||
|
including all available versions, release details, and distribution formats.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
package_name: The name of the PyPI package to query (e.g., 'requests', 'numpy')
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Dictionary containing version information including:
|
||||||
|
- Latest version and total version count
|
||||||
|
- List of all available versions (sorted)
|
||||||
|
- Recent versions with release details
|
||||||
|
- Distribution format information (wheel, source)
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
InvalidPackageNameError: If package name is empty or invalid
|
||||||
|
PackageNotFoundError: If package is not found on PyPI
|
||||||
|
NetworkError: For network-related errors
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
logger.info(f"MCP tool: Querying versions for {package_name}")
|
||||||
|
result = await query_package_versions(package_name)
|
||||||
|
logger.info(f"Successfully retrieved versions for package: {package_name}")
|
||||||
|
return result
|
||||||
|
except (InvalidPackageNameError, PackageNotFoundError, NetworkError) as e:
|
||||||
|
logger.error(f"Error querying versions for {package_name}: {e}")
|
||||||
|
return {
|
||||||
|
"error": str(e),
|
||||||
|
"error_type": type(e).__name__,
|
||||||
|
"package_name": package_name
|
||||||
|
}
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"Unexpected error querying versions for {package_name}: {e}")
|
||||||
|
return {
|
||||||
|
"error": f"Unexpected error: {e}",
|
||||||
|
"error_type": "UnexpectedError",
|
||||||
|
"package_name": package_name
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
@app.tool()
|
||||||
|
async def get_package_dependencies(package_name: str, version: str | None = None) -> dict[str, Any]:
|
||||||
|
"""Get dependency information for a PyPI package.
|
||||||
|
|
||||||
|
This tool retrieves comprehensive dependency information for a Python package,
|
||||||
|
including runtime dependencies, development dependencies, and optional dependencies.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
package_name: The name of the PyPI package to query (e.g., 'django', 'flask')
|
||||||
|
version: Specific version to query (optional, defaults to latest version)
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Dictionary containing dependency information including:
|
||||||
|
- Runtime dependencies and development dependencies
|
||||||
|
- Optional dependency groups
|
||||||
|
- Python version requirements
|
||||||
|
- Dependency counts and summary statistics
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
InvalidPackageNameError: If package name is empty or invalid
|
||||||
|
PackageNotFoundError: If package is not found on PyPI
|
||||||
|
NetworkError: For network-related errors
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
logger.info(f"MCP tool: Querying dependencies for {package_name}" +
|
||||||
|
(f" version {version}" if version else " (latest)"))
|
||||||
|
result = await query_package_dependencies(package_name, version)
|
||||||
|
logger.info(f"Successfully retrieved dependencies for package: {package_name}")
|
||||||
|
return result
|
||||||
|
except (InvalidPackageNameError, PackageNotFoundError, NetworkError) as e:
|
||||||
|
logger.error(f"Error querying dependencies for {package_name}: {e}")
|
||||||
|
return {
|
||||||
|
"error": str(e),
|
||||||
|
"error_type": type(e).__name__,
|
||||||
|
"package_name": package_name,
|
||||||
|
"version": version
|
||||||
|
}
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"Unexpected error querying dependencies for {package_name}: {e}")
|
||||||
|
return {
|
||||||
|
"error": f"Unexpected error: {e}",
|
||||||
|
"error_type": "UnexpectedError",
|
||||||
|
"package_name": package_name,
|
||||||
|
"version": version
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
@app.tool()
|
||||||
|
async def check_package_python_compatibility(
|
||||||
|
package_name: str,
|
||||||
|
target_python_version: str,
|
||||||
|
use_cache: bool = True
|
||||||
|
) -> dict[str, Any]:
|
||||||
|
"""Check if a package is compatible with a specific Python version.
|
||||||
|
|
||||||
|
This tool analyzes a package's Python version requirements and determines
|
||||||
|
if it's compatible with your target Python version.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
package_name: The name of the PyPI package to check (e.g., 'django', 'requests')
|
||||||
|
target_python_version: Target Python version to check (e.g., '3.9', '3.10.5', '3.11')
|
||||||
|
use_cache: Whether to use cached package data (default: True)
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Dictionary containing detailed compatibility information including:
|
||||||
|
- Compatibility status (True/False)
|
||||||
|
- Source of compatibility information (requires_python or classifiers)
|
||||||
|
- Detailed analysis and suggestions
|
||||||
|
- Package version requirements
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
InvalidPackageNameError: If package name is empty or invalid
|
||||||
|
PackageNotFoundError: If package is not found on PyPI
|
||||||
|
NetworkError: For network-related errors
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
logger.info(f"MCP tool: Checking Python {target_python_version} compatibility for {package_name}")
|
||||||
|
result = await check_python_compatibility(package_name, target_python_version, use_cache)
|
||||||
|
logger.info(f"Compatibility check completed for {package_name}")
|
||||||
|
return result
|
||||||
|
except (InvalidPackageNameError, PackageNotFoundError, NetworkError) as e:
|
||||||
|
logger.error(f"Error checking compatibility for {package_name}: {e}")
|
||||||
|
return {
|
||||||
|
"error": str(e),
|
||||||
|
"error_type": type(e).__name__,
|
||||||
|
"package_name": package_name,
|
||||||
|
"target_python_version": target_python_version
|
||||||
|
}
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"Unexpected error checking compatibility for {package_name}: {e}")
|
||||||
|
return {
|
||||||
|
"error": f"Unexpected error: {e}",
|
||||||
|
"error_type": "UnexpectedError",
|
||||||
|
"package_name": package_name,
|
||||||
|
"target_python_version": target_python_version
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
@app.tool()
|
||||||
|
async def get_package_compatible_python_versions(
|
||||||
|
package_name: str,
|
||||||
|
python_versions: list[str] | None = None,
|
||||||
|
use_cache: bool = True
|
||||||
|
) -> dict[str, Any]:
|
||||||
|
"""Get all Python versions compatible with a package.
|
||||||
|
|
||||||
|
This tool analyzes a package and returns which Python versions are
|
||||||
|
compatible with it, along with recommendations.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
package_name: The name of the PyPI package to analyze (e.g., 'numpy', 'pandas')
|
||||||
|
python_versions: List of Python versions to check (optional, defaults to common versions)
|
||||||
|
use_cache: Whether to use cached package data (default: True)
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Dictionary containing compatibility information including:
|
||||||
|
- List of compatible Python versions
|
||||||
|
- List of incompatible versions with reasons
|
||||||
|
- Compatibility rate and recommendations
|
||||||
|
- Package version requirements
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
InvalidPackageNameError: If package name is empty or invalid
|
||||||
|
PackageNotFoundError: If package is not found on PyPI
|
||||||
|
NetworkError: For network-related errors
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
logger.info(f"MCP tool: Getting compatible Python versions for {package_name}")
|
||||||
|
result = await get_compatible_python_versions(package_name, python_versions, use_cache)
|
||||||
|
logger.info(f"Compatible versions analysis completed for {package_name}")
|
||||||
|
return result
|
||||||
|
except (InvalidPackageNameError, PackageNotFoundError, NetworkError) as e:
|
||||||
|
logger.error(f"Error getting compatible versions for {package_name}: {e}")
|
||||||
|
return {
|
||||||
|
"error": str(e),
|
||||||
|
"error_type": type(e).__name__,
|
||||||
|
"package_name": package_name
|
||||||
|
}
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"Unexpected error getting compatible versions for {package_name}: {e}")
|
||||||
|
return {
|
||||||
|
"error": f"Unexpected error: {e}",
|
||||||
|
"error_type": "UnexpectedError",
|
||||||
|
"package_name": package_name
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
@click.command()
|
||||||
|
@click.option(
|
||||||
|
"--host",
|
||||||
|
default="localhost",
|
||||||
|
help="Host to bind the server to"
|
||||||
|
)
|
||||||
|
@click.option(
|
||||||
|
"--port",
|
||||||
|
default=8000,
|
||||||
|
type=int,
|
||||||
|
help="Port to bind the server to"
|
||||||
|
)
|
||||||
|
@click.option(
|
||||||
|
"--log-level",
|
||||||
|
default="INFO",
|
||||||
|
type=click.Choice(["DEBUG", "INFO", "WARNING", "ERROR"]),
|
||||||
|
help="Logging level"
|
||||||
|
)
|
||||||
|
def main(host: str, port: int, log_level: str) -> None:
|
||||||
|
"""Start the PyPI Query MCP Server."""
|
||||||
|
# Set logging level
|
||||||
|
logging.getLogger().setLevel(getattr(logging, log_level))
|
||||||
|
|
||||||
|
logger.info(f"Starting PyPI Query MCP Server on {host}:{port}")
|
||||||
|
logger.info(f"Log level set to: {log_level}")
|
||||||
|
|
||||||
|
# Run the FastMCP server
|
||||||
|
app.run(host=host, port=port)
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
main()
|
25
pypi_query_mcp/tools/__init__.py
Normal file
25
pypi_query_mcp/tools/__init__.py
Normal file
@ -0,0 +1,25 @@
|
|||||||
|
"""MCP tools for PyPI package queries.
|
||||||
|
|
||||||
|
This package contains the FastMCP tool implementations that provide
|
||||||
|
the user-facing interface for PyPI package operations.
|
||||||
|
"""
|
||||||
|
|
||||||
|
from .compatibility_check import (
|
||||||
|
check_python_compatibility,
|
||||||
|
get_compatible_python_versions,
|
||||||
|
suggest_python_version_for_packages,
|
||||||
|
)
|
||||||
|
from .package_query import (
|
||||||
|
query_package_dependencies,
|
||||||
|
query_package_info,
|
||||||
|
query_package_versions,
|
||||||
|
)
|
||||||
|
|
||||||
|
__all__ = [
|
||||||
|
"query_package_info",
|
||||||
|
"query_package_versions",
|
||||||
|
"query_package_dependencies",
|
||||||
|
"check_python_compatibility",
|
||||||
|
"get_compatible_python_versions",
|
||||||
|
"suggest_python_version_for_packages",
|
||||||
|
]
|
260
pypi_query_mcp/tools/compatibility_check.py
Normal file
260
pypi_query_mcp/tools/compatibility_check.py
Normal file
@ -0,0 +1,260 @@
|
|||||||
|
"""Python version compatibility checking tools for PyPI MCP server."""
|
||||||
|
|
||||||
|
import logging
|
||||||
|
from typing import Any
|
||||||
|
|
||||||
|
from ..core import (
|
||||||
|
InvalidPackageNameError,
|
||||||
|
NetworkError,
|
||||||
|
PyPIClient,
|
||||||
|
PyPIError,
|
||||||
|
)
|
||||||
|
from ..core.version_utils import VersionCompatibility
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
|
async def check_python_compatibility(
|
||||||
|
package_name: str,
|
||||||
|
target_python_version: str,
|
||||||
|
use_cache: bool = True
|
||||||
|
) -> dict[str, Any]:
|
||||||
|
"""Check if a package is compatible with a specific Python version.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
package_name: Name of the package to check
|
||||||
|
target_python_version: Target Python version (e.g., "3.9", "3.10.5")
|
||||||
|
use_cache: Whether to use cached package data
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Dictionary containing compatibility information
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
InvalidPackageNameError: If package name is invalid
|
||||||
|
PackageNotFoundError: If package is not found
|
||||||
|
NetworkError: For network-related errors
|
||||||
|
"""
|
||||||
|
if not package_name or not package_name.strip():
|
||||||
|
raise InvalidPackageNameError(package_name)
|
||||||
|
|
||||||
|
if not target_python_version or not target_python_version.strip():
|
||||||
|
raise ValueError("Target Python version cannot be empty")
|
||||||
|
|
||||||
|
logger.info(f"Checking Python {target_python_version} compatibility for package: {package_name}")
|
||||||
|
|
||||||
|
try:
|
||||||
|
async with PyPIClient() as client:
|
||||||
|
package_data = await client.get_package_info(package_name, use_cache)
|
||||||
|
|
||||||
|
info = package_data.get("info", {})
|
||||||
|
requires_python = info.get("requires_python")
|
||||||
|
classifiers = info.get("classifiers", [])
|
||||||
|
|
||||||
|
# Perform compatibility check
|
||||||
|
compat_checker = VersionCompatibility()
|
||||||
|
result = compat_checker.check_version_compatibility(
|
||||||
|
target_python_version,
|
||||||
|
requires_python,
|
||||||
|
classifiers
|
||||||
|
)
|
||||||
|
|
||||||
|
# Add package information to result
|
||||||
|
result.update({
|
||||||
|
"package_name": info.get("name", package_name),
|
||||||
|
"package_version": info.get("version", ""),
|
||||||
|
"requires_python": requires_python,
|
||||||
|
"supported_implementations": compat_checker.extract_python_implementations(classifiers),
|
||||||
|
"classifier_versions": sorted(compat_checker.extract_python_versions_from_classifiers(classifiers))
|
||||||
|
})
|
||||||
|
|
||||||
|
return result
|
||||||
|
|
||||||
|
except PyPIError:
|
||||||
|
# Re-raise PyPI-specific errors
|
||||||
|
raise
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"Unexpected error checking compatibility for {package_name}: {e}")
|
||||||
|
raise NetworkError(f"Failed to check Python compatibility: {e}", e) from e
|
||||||
|
|
||||||
|
|
||||||
|
async def get_compatible_python_versions(
|
||||||
|
package_name: str,
|
||||||
|
python_versions: list[str] | None = None,
|
||||||
|
use_cache: bool = True
|
||||||
|
) -> dict[str, Any]:
|
||||||
|
"""Get list of Python versions compatible with a package.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
package_name: Name of the package to check
|
||||||
|
python_versions: List of Python versions to check (optional)
|
||||||
|
use_cache: Whether to use cached package data
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Dictionary containing compatibility information for multiple versions
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
InvalidPackageNameError: If package name is invalid
|
||||||
|
PackageNotFoundError: If package is not found
|
||||||
|
NetworkError: For network-related errors
|
||||||
|
"""
|
||||||
|
if not package_name or not package_name.strip():
|
||||||
|
raise InvalidPackageNameError(package_name)
|
||||||
|
|
||||||
|
logger.info(f"Getting compatible Python versions for package: {package_name}")
|
||||||
|
|
||||||
|
try:
|
||||||
|
async with PyPIClient() as client:
|
||||||
|
package_data = await client.get_package_info(package_name, use_cache)
|
||||||
|
|
||||||
|
info = package_data.get("info", {})
|
||||||
|
requires_python = info.get("requires_python")
|
||||||
|
classifiers = info.get("classifiers", [])
|
||||||
|
|
||||||
|
# Get compatibility information
|
||||||
|
compat_checker = VersionCompatibility()
|
||||||
|
result = compat_checker.get_compatible_versions(
|
||||||
|
requires_python,
|
||||||
|
classifiers,
|
||||||
|
python_versions
|
||||||
|
)
|
||||||
|
|
||||||
|
# Add package information to result
|
||||||
|
result.update({
|
||||||
|
"package_name": info.get("name", package_name),
|
||||||
|
"package_version": info.get("version", ""),
|
||||||
|
"requires_python": requires_python,
|
||||||
|
"supported_implementations": sorted(compat_checker.extract_python_implementations(classifiers)),
|
||||||
|
"classifier_versions": sorted(compat_checker.extract_python_versions_from_classifiers(classifiers))
|
||||||
|
})
|
||||||
|
|
||||||
|
return result
|
||||||
|
|
||||||
|
except PyPIError:
|
||||||
|
# Re-raise PyPI-specific errors
|
||||||
|
raise
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"Unexpected error getting compatible versions for {package_name}: {e}")
|
||||||
|
raise NetworkError(f"Failed to get compatible Python versions: {e}", e) from e
|
||||||
|
|
||||||
|
|
||||||
|
async def suggest_python_version_for_packages(
|
||||||
|
package_names: list[str],
|
||||||
|
use_cache: bool = True
|
||||||
|
) -> dict[str, Any]:
|
||||||
|
"""Suggest optimal Python version for a list of packages.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
package_names: List of package names to analyze
|
||||||
|
use_cache: Whether to use cached package data
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Dictionary containing version suggestions and compatibility matrix
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
ValueError: If package_names is empty
|
||||||
|
NetworkError: For network-related errors
|
||||||
|
"""
|
||||||
|
if not package_names:
|
||||||
|
raise ValueError("Package names list cannot be empty")
|
||||||
|
|
||||||
|
logger.info(f"Analyzing Python version compatibility for {len(package_names)} packages")
|
||||||
|
|
||||||
|
# Default Python versions to analyze
|
||||||
|
python_versions = ["3.7", "3.8", "3.9", "3.10", "3.11", "3.12"]
|
||||||
|
|
||||||
|
compatibility_matrix = {}
|
||||||
|
package_details = {}
|
||||||
|
errors = {}
|
||||||
|
|
||||||
|
async with PyPIClient() as client:
|
||||||
|
for package_name in package_names:
|
||||||
|
try:
|
||||||
|
package_data = await client.get_package_info(package_name, use_cache)
|
||||||
|
info = package_data.get("info", {})
|
||||||
|
|
||||||
|
requires_python = info.get("requires_python")
|
||||||
|
classifiers = info.get("classifiers", [])
|
||||||
|
|
||||||
|
compat_checker = VersionCompatibility()
|
||||||
|
compat_result = compat_checker.get_compatible_versions(
|
||||||
|
requires_python,
|
||||||
|
classifiers,
|
||||||
|
python_versions
|
||||||
|
)
|
||||||
|
|
||||||
|
# Store compatibility for this package
|
||||||
|
compatible_versions = [v["version"] for v in compat_result["compatible_versions"]]
|
||||||
|
compatibility_matrix[package_name] = compatible_versions
|
||||||
|
|
||||||
|
package_details[package_name] = {
|
||||||
|
"version": info.get("version", ""),
|
||||||
|
"requires_python": requires_python,
|
||||||
|
"compatible_versions": compatible_versions,
|
||||||
|
"compatibility_rate": compat_result["compatibility_rate"]
|
||||||
|
}
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
logger.warning(f"Failed to analyze package {package_name}: {e}")
|
||||||
|
errors[package_name] = str(e)
|
||||||
|
compatibility_matrix[package_name] = []
|
||||||
|
|
||||||
|
# Find common compatible versions
|
||||||
|
if compatibility_matrix:
|
||||||
|
all_versions = set(python_versions)
|
||||||
|
common_versions = all_versions.copy()
|
||||||
|
|
||||||
|
for _package_name, compatible in compatibility_matrix.items():
|
||||||
|
if compatible: # Only consider packages with known compatibility
|
||||||
|
common_versions &= set(compatible)
|
||||||
|
else:
|
||||||
|
common_versions = set()
|
||||||
|
|
||||||
|
# Generate recommendations
|
||||||
|
recommendations = []
|
||||||
|
if common_versions:
|
||||||
|
latest_common = max(common_versions, key=lambda x: tuple(map(int, x.split("."))))
|
||||||
|
recommendations.append(
|
||||||
|
f"✅ Recommended Python version: {latest_common} "
|
||||||
|
f"(compatible with all {len([p for p in compatibility_matrix if compatibility_matrix[p]])} packages)"
|
||||||
|
)
|
||||||
|
|
||||||
|
if len(common_versions) > 1:
|
||||||
|
all_common = sorted(common_versions, key=lambda x: tuple(map(int, x.split("."))))
|
||||||
|
recommendations.append(
|
||||||
|
f"📋 All compatible versions: {', '.join(all_common)}"
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
recommendations.append(
|
||||||
|
"⚠️ No Python version is compatible with all packages. "
|
||||||
|
"Consider updating packages or using different versions."
|
||||||
|
)
|
||||||
|
|
||||||
|
# Find the version compatible with most packages
|
||||||
|
version_scores = {}
|
||||||
|
for version in python_versions:
|
||||||
|
score = sum(1 for compatible in compatibility_matrix.values() if version in compatible)
|
||||||
|
version_scores[version] = score
|
||||||
|
|
||||||
|
if version_scores:
|
||||||
|
best_version = max(version_scores, key=version_scores.get)
|
||||||
|
best_score = version_scores[best_version]
|
||||||
|
total_packages = len([p for p in compatibility_matrix if compatibility_matrix[p]])
|
||||||
|
|
||||||
|
if best_score > 0:
|
||||||
|
recommendations.append(
|
||||||
|
f"📊 Best compromise: Python {best_version} "
|
||||||
|
f"(compatible with {best_score}/{total_packages} packages)"
|
||||||
|
)
|
||||||
|
|
||||||
|
return {
|
||||||
|
"analyzed_packages": len(package_names),
|
||||||
|
"successful_analyses": len(package_details),
|
||||||
|
"failed_analyses": len(errors),
|
||||||
|
"common_compatible_versions": sorted(common_versions),
|
||||||
|
"recommended_version": max(common_versions, key=lambda x: tuple(map(int, x.split(".")))) if common_versions else None,
|
||||||
|
"compatibility_matrix": compatibility_matrix,
|
||||||
|
"package_details": package_details,
|
||||||
|
"errors": errors,
|
||||||
|
"recommendations": recommendations,
|
||||||
|
"python_versions_analyzed": python_versions
|
||||||
|
}
|
253
pypi_query_mcp/tools/package_query.py
Normal file
253
pypi_query_mcp/tools/package_query.py
Normal file
@ -0,0 +1,253 @@
|
|||||||
|
"""Package query tools for PyPI MCP server."""
|
||||||
|
|
||||||
|
import logging
|
||||||
|
from typing import Any
|
||||||
|
|
||||||
|
from ..core import (
|
||||||
|
InvalidPackageNameError,
|
||||||
|
NetworkError,
|
||||||
|
PyPIClient,
|
||||||
|
PyPIError,
|
||||||
|
)
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
|
def format_package_info(package_data: dict[str, Any]) -> dict[str, Any]:
|
||||||
|
"""Format package information for MCP response.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
package_data: Raw package data from PyPI API
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Formatted package information
|
||||||
|
"""
|
||||||
|
info = package_data.get("info", {})
|
||||||
|
|
||||||
|
# Extract basic information
|
||||||
|
formatted = {
|
||||||
|
"name": info.get("name", ""),
|
||||||
|
"version": info.get("version", ""),
|
||||||
|
"summary": info.get("summary", ""),
|
||||||
|
"description": info.get("description", "")[:500] + "..." if len(info.get("description", "")) > 500 else info.get("description", ""),
|
||||||
|
"author": info.get("author", ""),
|
||||||
|
"author_email": info.get("author_email", ""),
|
||||||
|
"maintainer": info.get("maintainer", ""),
|
||||||
|
"maintainer_email": info.get("maintainer_email", ""),
|
||||||
|
"license": info.get("license", ""),
|
||||||
|
"home_page": info.get("home_page", ""),
|
||||||
|
"project_url": info.get("project_url", ""),
|
||||||
|
"download_url": info.get("download_url", ""),
|
||||||
|
"requires_python": info.get("requires_python", ""),
|
||||||
|
"platform": info.get("platform", ""),
|
||||||
|
"keywords": info.get("keywords", ""),
|
||||||
|
"classifiers": info.get("classifiers", []),
|
||||||
|
"requires_dist": info.get("requires_dist", []),
|
||||||
|
"project_urls": info.get("project_urls", {}),
|
||||||
|
}
|
||||||
|
|
||||||
|
# Add release information
|
||||||
|
releases = package_data.get("releases", {})
|
||||||
|
formatted["total_versions"] = len(releases)
|
||||||
|
formatted["available_versions"] = list(releases.keys())[-10:] # Last 10 versions
|
||||||
|
|
||||||
|
# Add download statistics if available
|
||||||
|
if "urls" in package_data:
|
||||||
|
urls = package_data["urls"]
|
||||||
|
if urls:
|
||||||
|
formatted["download_info"] = {
|
||||||
|
"files_count": len(urls),
|
||||||
|
"file_types": list({url.get("packagetype", "") for url in urls}),
|
||||||
|
"python_versions": list({url.get("python_version", "") for url in urls if url.get("python_version")}),
|
||||||
|
}
|
||||||
|
|
||||||
|
return formatted
|
||||||
|
|
||||||
|
|
||||||
|
def format_version_info(package_data: dict[str, Any]) -> dict[str, Any]:
|
||||||
|
"""Format version information for MCP response.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
package_data: Raw package data from PyPI API
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Formatted version information
|
||||||
|
"""
|
||||||
|
info = package_data.get("info", {})
|
||||||
|
releases = package_data.get("releases", {})
|
||||||
|
|
||||||
|
# Sort versions (basic sorting, could be improved with proper version parsing)
|
||||||
|
sorted_versions = sorted(releases.keys(), reverse=True)
|
||||||
|
|
||||||
|
return {
|
||||||
|
"package_name": info.get("name", ""),
|
||||||
|
"latest_version": info.get("version", ""),
|
||||||
|
"total_versions": len(releases),
|
||||||
|
"versions": sorted_versions,
|
||||||
|
"recent_versions": sorted_versions[:20], # Last 20 versions
|
||||||
|
"version_details": {
|
||||||
|
version: {
|
||||||
|
"release_count": len(releases[version]),
|
||||||
|
"has_wheel": any(file.get("packagetype") == "bdist_wheel" for file in releases[version]),
|
||||||
|
"has_source": any(file.get("packagetype") == "sdist" for file in releases[version]),
|
||||||
|
}
|
||||||
|
for version in sorted_versions[:10] # Details for last 10 versions
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
def format_dependency_info(package_data: dict[str, Any]) -> dict[str, Any]:
|
||||||
|
"""Format dependency information for MCP response.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
package_data: Raw package data from PyPI API
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Formatted dependency information
|
||||||
|
"""
|
||||||
|
info = package_data.get("info", {})
|
||||||
|
requires_dist = info.get("requires_dist", []) or []
|
||||||
|
|
||||||
|
# Parse dependencies
|
||||||
|
runtime_deps = []
|
||||||
|
dev_deps = []
|
||||||
|
optional_deps = {}
|
||||||
|
|
||||||
|
for dep in requires_dist:
|
||||||
|
if not dep:
|
||||||
|
continue
|
||||||
|
|
||||||
|
# Basic parsing - could be improved with proper dependency parsing
|
||||||
|
if "extra ==" in dep:
|
||||||
|
# Optional dependency
|
||||||
|
parts = dep.split(";")
|
||||||
|
dep_name = parts[0].strip()
|
||||||
|
extra_part = parts[1] if len(parts) > 1 else ""
|
||||||
|
|
||||||
|
if "extra ==" in extra_part:
|
||||||
|
extra_name = extra_part.split("extra ==")[1].strip().strip('"\'')
|
||||||
|
if extra_name not in optional_deps:
|
||||||
|
optional_deps[extra_name] = []
|
||||||
|
optional_deps[extra_name].append(dep_name)
|
||||||
|
elif "dev" in dep.lower() or "test" in dep.lower():
|
||||||
|
dev_deps.append(dep)
|
||||||
|
else:
|
||||||
|
runtime_deps.append(dep)
|
||||||
|
|
||||||
|
return {
|
||||||
|
"package_name": info.get("name", ""),
|
||||||
|
"version": info.get("version", ""),
|
||||||
|
"requires_python": info.get("requires_python", ""),
|
||||||
|
"runtime_dependencies": runtime_deps,
|
||||||
|
"development_dependencies": dev_deps,
|
||||||
|
"optional_dependencies": optional_deps,
|
||||||
|
"total_dependencies": len(requires_dist),
|
||||||
|
"dependency_summary": {
|
||||||
|
"runtime_count": len(runtime_deps),
|
||||||
|
"dev_count": len(dev_deps),
|
||||||
|
"optional_groups": len(optional_deps),
|
||||||
|
"total_optional": sum(len(deps) for deps in optional_deps.values()),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
async def query_package_info(package_name: str) -> dict[str, Any]:
|
||||||
|
"""Query comprehensive package information from PyPI.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
package_name: Name of the package to query
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Formatted package information
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
InvalidPackageNameError: If package name is invalid
|
||||||
|
PackageNotFoundError: If package is not found
|
||||||
|
NetworkError: For network-related errors
|
||||||
|
"""
|
||||||
|
if not package_name or not package_name.strip():
|
||||||
|
raise InvalidPackageNameError(package_name)
|
||||||
|
|
||||||
|
logger.info(f"Querying package info for: {package_name}")
|
||||||
|
|
||||||
|
try:
|
||||||
|
async with PyPIClient() as client:
|
||||||
|
package_data = await client.get_package_info(package_name)
|
||||||
|
return format_package_info(package_data)
|
||||||
|
except PyPIError:
|
||||||
|
# Re-raise PyPI-specific errors
|
||||||
|
raise
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"Unexpected error querying package {package_name}: {e}")
|
||||||
|
raise NetworkError(f"Failed to query package information: {e}", e) from e
|
||||||
|
|
||||||
|
|
||||||
|
async def query_package_versions(package_name: str) -> dict[str, Any]:
|
||||||
|
"""Query package version information from PyPI.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
package_name: Name of the package to query
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Formatted version information
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
InvalidPackageNameError: If package name is invalid
|
||||||
|
PackageNotFoundError: If package is not found
|
||||||
|
NetworkError: For network-related errors
|
||||||
|
"""
|
||||||
|
if not package_name or not package_name.strip():
|
||||||
|
raise InvalidPackageNameError(package_name)
|
||||||
|
|
||||||
|
logger.info(f"Querying versions for package: {package_name}")
|
||||||
|
|
||||||
|
try:
|
||||||
|
async with PyPIClient() as client:
|
||||||
|
package_data = await client.get_package_info(package_name)
|
||||||
|
return format_version_info(package_data)
|
||||||
|
except PyPIError:
|
||||||
|
# Re-raise PyPI-specific errors
|
||||||
|
raise
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"Unexpected error querying versions for {package_name}: {e}")
|
||||||
|
raise NetworkError(f"Failed to query package versions: {e}", e) from e
|
||||||
|
|
||||||
|
|
||||||
|
async def query_package_dependencies(package_name: str, version: str | None = None) -> dict[str, Any]:
|
||||||
|
"""Query package dependency information from PyPI.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
package_name: Name of the package to query
|
||||||
|
version: Specific version to query (optional, defaults to latest)
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Formatted dependency information
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
InvalidPackageNameError: If package name is invalid
|
||||||
|
PackageNotFoundError: If package is not found
|
||||||
|
NetworkError: For network-related errors
|
||||||
|
"""
|
||||||
|
if not package_name or not package_name.strip():
|
||||||
|
raise InvalidPackageNameError(package_name)
|
||||||
|
|
||||||
|
logger.info(f"Querying dependencies for package: {package_name}" +
|
||||||
|
(f" version {version}" if version else " (latest)"))
|
||||||
|
|
||||||
|
try:
|
||||||
|
async with PyPIClient() as client:
|
||||||
|
package_data = await client.get_package_info(package_name)
|
||||||
|
|
||||||
|
# TODO: In future, support querying specific version dependencies
|
||||||
|
# For now, we return dependencies for the latest version
|
||||||
|
if version and version != package_data.get("info", {}).get("version"):
|
||||||
|
logger.warning(f"Specific version {version} requested but not implemented yet. "
|
||||||
|
f"Returning dependencies for latest version.")
|
||||||
|
|
||||||
|
return format_dependency_info(package_data)
|
||||||
|
except PyPIError:
|
||||||
|
# Re-raise PyPI-specific errors
|
||||||
|
raise
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"Unexpected error querying dependencies for {package_name}: {e}")
|
||||||
|
raise NetworkError(f"Failed to query package dependencies: {e}", e) from e
|
102
pyproject.toml
Normal file
102
pyproject.toml
Normal file
@ -0,0 +1,102 @@
|
|||||||
|
[build-system]
|
||||||
|
requires = ["poetry-core"]
|
||||||
|
build-backend = "poetry.core.masonry.api"
|
||||||
|
|
||||||
|
[tool.poetry]
|
||||||
|
name = "pypi-query-mcp-server"
|
||||||
|
version = "0.1.0"
|
||||||
|
description = "A Model Context Protocol (MCP) server for querying PyPI package information, dependencies, and compatibility"
|
||||||
|
authors = ["Hal <hal.long@outlook.com>"]
|
||||||
|
readme = "README.md"
|
||||||
|
license = "MIT"
|
||||||
|
homepage = "https://github.com/loonghao/pypi-query-mcp-server"
|
||||||
|
repository = "https://github.com/loonghao/pypi-query-mcp-server"
|
||||||
|
documentation = "https://github.com/loonghao/pypi-query-mcp-server"
|
||||||
|
keywords = ["mcp", "pypi", "package", "dependency", "python"]
|
||||||
|
classifiers = [
|
||||||
|
"Development Status :: 3 - Alpha",
|
||||||
|
"Intended Audience :: Developers",
|
||||||
|
"License :: OSI Approved :: MIT License",
|
||||||
|
"Operating System :: OS Independent",
|
||||||
|
"Programming Language :: Python :: 3",
|
||||||
|
"Programming Language :: Python :: 3.10",
|
||||||
|
"Programming Language :: Python :: 3.11",
|
||||||
|
"Programming Language :: Python :: 3.12",
|
||||||
|
"Topic :: Software Development :: Libraries :: Python Modules",
|
||||||
|
"Topic :: System :: Software Distribution",
|
||||||
|
]
|
||||||
|
packages = [{include = "pypi_query_mcp"}]
|
||||||
|
|
||||||
|
[tool.poetry.dependencies]
|
||||||
|
python = "^3.10"
|
||||||
|
fastmcp = "^0.2.0"
|
||||||
|
httpx = "^0.27.0"
|
||||||
|
packaging = "^24.0"
|
||||||
|
pydantic = "^2.0.0"
|
||||||
|
click = "^8.1.0"
|
||||||
|
|
||||||
|
[tool.poetry.group.dev.dependencies]
|
||||||
|
pytest = "^8.0.0"
|
||||||
|
pytest-asyncio = "^0.23.0"
|
||||||
|
pytest-cov = "^4.0.0"
|
||||||
|
pytest-mock = "^3.12.0"
|
||||||
|
ruff = "^0.1.0"
|
||||||
|
mypy = "^1.8.0"
|
||||||
|
pre-commit = "^3.6.0"
|
||||||
|
nox = "^2024.3.2"
|
||||||
|
|
||||||
|
[tool.poetry.scripts]
|
||||||
|
pypi-query-mcp = "pypi_query_mcp.server:main"
|
||||||
|
|
||||||
|
[tool.ruff]
|
||||||
|
target-version = "py310"
|
||||||
|
line-length = 88
|
||||||
|
select = [
|
||||||
|
"E", # pycodestyle errors
|
||||||
|
"W", # pycodestyle warnings
|
||||||
|
"F", # pyflakes
|
||||||
|
"I", # isort
|
||||||
|
"B", # flake8-bugbear
|
||||||
|
"C4", # flake8-comprehensions
|
||||||
|
"UP", # pyupgrade
|
||||||
|
]
|
||||||
|
ignore = [
|
||||||
|
"E501", # line too long, handled by black
|
||||||
|
"B008", # do not perform function calls in argument defaults
|
||||||
|
"C901", # too complex
|
||||||
|
]
|
||||||
|
|
||||||
|
[tool.ruff.per-file-ignores]
|
||||||
|
"__init__.py" = ["F401"]
|
||||||
|
|
||||||
|
[tool.mypy]
|
||||||
|
python_version = "3.10"
|
||||||
|
check_untyped_defs = true
|
||||||
|
disallow_any_generics = true
|
||||||
|
disallow_incomplete_defs = true
|
||||||
|
disallow_untyped_defs = true
|
||||||
|
no_implicit_optional = true
|
||||||
|
warn_redundant_casts = true
|
||||||
|
warn_unused_ignores = true
|
||||||
|
|
||||||
|
[tool.pytest.ini_options]
|
||||||
|
minversion = "6.0"
|
||||||
|
addopts = "-ra -q --strict-markers --strict-config"
|
||||||
|
testpaths = ["tests"]
|
||||||
|
asyncio_mode = "auto"
|
||||||
|
|
||||||
|
[tool.coverage.run]
|
||||||
|
source = ["pypi_query_mcp"]
|
||||||
|
omit = ["tests/*"]
|
||||||
|
|
||||||
|
[tool.coverage.report]
|
||||||
|
exclude_lines = [
|
||||||
|
"pragma: no cover",
|
||||||
|
"def __repr__",
|
||||||
|
"if self.debug:",
|
||||||
|
"if settings.DEBUG",
|
||||||
|
"raise AssertionError",
|
||||||
|
"raise NotImplementedError",
|
||||||
|
"if 0:",
|
||||||
|
"if __name__ == .__main__.:",
|
||||||
|
]
|
@ -1,3 +0,0 @@
|
|||||||
poetry
|
|
||||||
nox
|
|
||||||
pytest
|
|
1
tests/__init__.py
Normal file
1
tests/__init__.py
Normal file
@ -0,0 +1 @@
|
|||||||
|
"""Test package for PyPI Query MCP Server."""
|
63
tests/conftest.py
Normal file
63
tests/conftest.py
Normal file
@ -0,0 +1,63 @@
|
|||||||
|
"""Pytest configuration and fixtures."""
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture
|
||||||
|
def sample_package_data():
|
||||||
|
"""Sample package data for testing."""
|
||||||
|
return {
|
||||||
|
"info": {
|
||||||
|
"name": "test-package",
|
||||||
|
"version": "1.0.0",
|
||||||
|
"summary": "A test package",
|
||||||
|
"description": "This is a test package for testing purposes.",
|
||||||
|
"author": "Test Author",
|
||||||
|
"author_email": "test@example.com",
|
||||||
|
"license": "MIT",
|
||||||
|
"requires_python": ">=3.8",
|
||||||
|
"requires_dist": [
|
||||||
|
"requests>=2.25.0",
|
||||||
|
"click>=8.0.0",
|
||||||
|
"pytest>=6.0.0; extra == 'test'"
|
||||||
|
],
|
||||||
|
"classifiers": [
|
||||||
|
"Development Status :: 4 - Beta",
|
||||||
|
"Intended Audience :: Developers",
|
||||||
|
"License :: OSI Approved :: MIT License",
|
||||||
|
"Programming Language :: Python :: 3",
|
||||||
|
"Programming Language :: Python :: 3.8",
|
||||||
|
"Programming Language :: Python :: 3.9",
|
||||||
|
"Programming Language :: Python :: 3.10",
|
||||||
|
"Programming Language :: Python :: 3.11",
|
||||||
|
"Programming Language :: Python :: Implementation :: CPython"
|
||||||
|
]
|
||||||
|
},
|
||||||
|
"releases": {
|
||||||
|
"1.0.0": [
|
||||||
|
{
|
||||||
|
"filename": "test_package-1.0.0-py3-none-any.whl",
|
||||||
|
"packagetype": "bdist_wheel",
|
||||||
|
"python_version": "py3"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"filename": "test-package-1.0.0.tar.gz",
|
||||||
|
"packagetype": "sdist",
|
||||||
|
"python_version": "source"
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"0.9.0": [
|
||||||
|
{
|
||||||
|
"filename": "test-package-0.9.0.tar.gz",
|
||||||
|
"packagetype": "sdist",
|
||||||
|
"python_version": "source"
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture
|
||||||
|
def mock_pypi_response(sample_package_data):
|
||||||
|
"""Mock PyPI API response."""
|
||||||
|
return sample_package_data
|
78
tests/test_basic.py
Normal file
78
tests/test_basic.py
Normal file
@ -0,0 +1,78 @@
|
|||||||
|
"""Basic tests for PyPI Query MCP Server."""
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
|
||||||
|
from pypi_query_mcp import __version__
|
||||||
|
|
||||||
|
|
||||||
|
def test_version():
|
||||||
|
"""Test that version is defined."""
|
||||||
|
assert __version__ == "0.1.0"
|
||||||
|
|
||||||
|
|
||||||
|
def test_import():
|
||||||
|
"""Test that main modules can be imported."""
|
||||||
|
from pypi_query_mcp.core import PyPIClient, VersionCompatibility
|
||||||
|
from pypi_query_mcp.server import app
|
||||||
|
|
||||||
|
assert PyPIClient is not None
|
||||||
|
assert VersionCompatibility is not None
|
||||||
|
assert app is not None
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_pypi_client_basic():
|
||||||
|
"""Test basic PyPI client functionality."""
|
||||||
|
from pypi_query_mcp.core import PyPIClient
|
||||||
|
|
||||||
|
async with PyPIClient() as client:
|
||||||
|
# Test that client can be created and closed
|
||||||
|
assert client is not None
|
||||||
|
|
||||||
|
# Test cache clearing
|
||||||
|
client.clear_cache()
|
||||||
|
|
||||||
|
|
||||||
|
def test_version_compatibility():
|
||||||
|
"""Test version compatibility utility."""
|
||||||
|
from pypi_query_mcp.core import VersionCompatibility
|
||||||
|
|
||||||
|
compat = VersionCompatibility()
|
||||||
|
|
||||||
|
# Test requires_python parsing
|
||||||
|
spec = compat.parse_requires_python(">=3.8")
|
||||||
|
assert spec is not None
|
||||||
|
|
||||||
|
# Test classifier extraction
|
||||||
|
classifiers = [
|
||||||
|
"Programming Language :: Python :: 3.8",
|
||||||
|
"Programming Language :: Python :: 3.9",
|
||||||
|
"Programming Language :: Python :: Implementation :: CPython"
|
||||||
|
]
|
||||||
|
|
||||||
|
versions = compat.extract_python_versions_from_classifiers(classifiers)
|
||||||
|
assert "3.8" in versions
|
||||||
|
assert "3.9" in versions
|
||||||
|
|
||||||
|
implementations = compat.extract_python_implementations(classifiers)
|
||||||
|
assert "CPython" in implementations
|
||||||
|
|
||||||
|
|
||||||
|
def test_mcp_tools_import():
|
||||||
|
"""Test that MCP tools can be imported."""
|
||||||
|
from pypi_query_mcp.tools import (
|
||||||
|
check_python_compatibility,
|
||||||
|
get_compatible_python_versions,
|
||||||
|
query_package_dependencies,
|
||||||
|
query_package_info,
|
||||||
|
query_package_versions,
|
||||||
|
suggest_python_version_for_packages,
|
||||||
|
)
|
||||||
|
|
||||||
|
# Test that all tools are callable
|
||||||
|
assert callable(query_package_info)
|
||||||
|
assert callable(query_package_versions)
|
||||||
|
assert callable(query_package_dependencies)
|
||||||
|
assert callable(check_python_compatibility)
|
||||||
|
assert callable(get_compatible_python_versions)
|
||||||
|
assert callable(suggest_python_version_for_packages)
|
Loading…
x
Reference in New Issue
Block a user