MCP/pypi-query-mcp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

feat: implement comprehensive configuration management system with multi-mirror support

Browse source 
- Add ServerSettings class with pydantic-settings for type-safe configuration
- Support multiple PyPI mirror sources with priority-based fallback mechanism
- Implement RepositoryConfig and RepositoryManager for multi-repository support
- Add environment variable support for all configuration options
- Include private repository authentication configuration
- Add advanced dependency analysis settings (max depth, concurrency, security)
- Provide secure credential management with sensitive data masking
- Update documentation and configuration examples
- Add comprehensive test suite with 23 test cases covering all features
- Include demo script showcasing multi-mirror configuration capabilities

Configuration features:
- Primary, additional, and fallback index URLs
- Automatic duplicate URL removal with priority preservation
- Runtime configuration reloading
- Integration with repository manager for seamless multi-source queries

Signed-off-by: longhao <hal.long@outlook.com>
This commit is contained in:
longhao 2025-05-27 17:36:25 +08:00 • committed by Hal
parent f27493d8d2
commit a0c507c3ff
20 changed files with 1276 additions and 165 deletions
Download patch file Download diff file Expand all files Collapse all files

25
pypi_query_mcp/core/pypi_client.py
View file

@ -99,6 +99,7 @@ class PyPIClient:
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]:
@ -140,7 +141,7 @@ class PyPIClient:
else:
raise PyPIServerError(
response.status_code,
f"Unexpected status code: {response.status_code}"
f"Unexpected status code: {response.status_code}",
)
except httpx.TimeoutException as e:
@ -155,12 +156,16 @@ class PyPIClient:
# Wait before retry (except on last attempt)
if attempt < self.max_retries:
await asyncio.sleep(self.retry_delay * (2 ** attempt)) # Exponential backoff
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]:
async def get_package_info(
self, package_name: str, use_cache: bool = True
) -> dict[str, Any]:
"""Get comprehensive package information from PyPI.
Args:
@ -194,10 +199,8 @@ class PyPIClient:
# Cache the result
import time
self._cache[cache_key] = {
"data": data,
"timestamp": time.time()
}
self._cache[cache_key] = {"data": data, "timestamp": time.time()}
return data
@ -205,7 +208,9 @@ class PyPIClient:
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]:
async def get_package_versions(
self, package_name: str, use_cache: bool = True
) -> list[str]:
"""Get list of available versions for a package.
Args:
@ -219,7 +224,9 @@ class PyPIClient:
releases = package_info.get("releases", {})
return list(releases.keys())
async def get_latest_version(self, package_name: str, use_cache: bool = True) -> str:
async def get_latest_version(
self, package_name: str, use_cache: bool = True
) -> str:
"""Get the latest version of a package.
Args:

94
pypi_query_mcp/core/version_utils.py
View file

@ -45,7 +45,9 @@ class VersionCompatibility:
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]:
def extract_python_versions_from_classifiers(
self, classifiers: list[str]
) -> set[str]:
"""Extract Python version information from classifiers.
Args:
@ -87,7 +89,7 @@ class VersionCompatibility:
self,
target_version: str,
requires_python: str | None = None,
classifiers: list[str] | None = None
classifiers: list[str] | None = None,
) -> dict[str, Any]:
"""Check if a target Python version is compatible with package requirements.
@ -105,7 +107,7 @@ class VersionCompatibility:
"compatibility_source": None,
"details": {},
"warnings": [],
"suggestions": []
"suggestions": [],
}
try:
@ -119,15 +121,17 @@ class VersionCompatibility:
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
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(
@ -139,7 +143,9 @@ class VersionCompatibility:
# Fall back to classifiers if no requires_python
if classifiers:
supported_versions = self.extract_python_versions_from_classifiers(classifiers)
supported_versions = self.extract_python_versions_from_classifiers(
classifiers
)
implementations = self.extract_python_implementations(classifiers)
if supported_versions:
@ -148,21 +154,23 @@ class VersionCompatibility:
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
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
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(
@ -186,7 +194,7 @@ class VersionCompatibility:
self,
requires_python: str | None = None,
classifiers: list[str] | None = None,
available_pythons: list[str] | None = None
available_pythons: list[str] | None = None,
) -> dict[str, Any]:
"""Get list of compatible Python versions for a package.
@ -200,9 +208,7 @@ class VersionCompatibility:
"""
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"
]
available_pythons = ["3.7", "3.8", "3.9", "3.10", "3.11", "3.12", "3.13"]
compatible = []
incompatible = []
@ -213,28 +219,34 @@ class VersionCompatibility:
)
if result["is_compatible"]:
compatible.append({
"version": python_version,
"source": result["compatibility_source"]
})
compatible.append(
{
"version": python_version,
"source": result["compatibility_source"],
}
)
else:
incompatible.append({
"version": python_version,
"reason": result["suggestions"][0] if result["suggestions"] else "Unknown"
})
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)
"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]]
self, compatible: list[dict[str, Any]], incompatible: list[dict[str, Any]]
) -> list[str]:
"""Generate recommendations based on compatibility results.