如何用Python脚本检查pip包是否存在于指定认证自定义索引

方法一：使用pip内部API（推荐，贴近pip原生逻辑） #

pip内部提供了索引查询的相关模块，可以直接复用pip的包查找逻辑，包括认证处理。以下是示例代码：

from pip._internal.index.package_finder import PackageFinder
from pip._internal.network.session import PipSession
from pip._internal.locations import USER_CACHE_DIR

def check_package_exists(package_name, index_url, username=None, password=None):
    # 配置会话，处理认证
    session = PipSession()
    if username and password:
        session.auth = (username, password)
    
    # 创建包查找器
    finder = PackageFinder.create(
        session=session,
        index_urls=[index_url],
        user_cache_dir=USER_CACHE_DIR,
    )
    
    # 尝试查找包
    try:
        # 获取包的所有可用版本（如果存在则返回非空列表）
        candidates = finder.find_all_candidates(package_name)
        return len(candidates) > 0
    except Exception as e:
        # 处理索引不可达、认证失败等异常
        print(f"检查出错: {str(e)}")
        return False

# 示例调用
package = "package-1"
index = "https://mytestdomain.com/pypi/pypi/simple/"
# 如果索引需要Basic认证，传入用户名和密码
exists = check_package_exists(package, index, username="your_user", password="your_pass")
print(f"包 {package} 是否存在: {exists}")

注意：pip的内部API（带_internal前缀的模块）没有公开的兼容性承诺，不同pip版本可能有接口变化，使用前建议在目标环境测试。

方法二：直接请求Simple索引页面（轻量，无需依赖pip内部模块） #

pip的Simple索引是基于HTML的静态页面，每个包对应一个子页面链接。可以通过requests库请求索引首页，检查是否包含目标包的链接（注意包名在索引中通常是小写形式）。

import requests
from requests.auth import HTTPBasicAuth

def check_package_via_requests(package_name, index_url, username=None, password=None):
    # 处理包名（索引页面通常用小写）
    target_package = package_name.lower()
    auth = HTTPBasicAuth(username, password) if (username and password) else None
    
    try:
        # 请求索引首页
        response = requests.get(index_url, auth=auth, timeout=10)
        response.raise_for_status()  # 抛出HTTP错误
        
        # 检查页面中是否包含目标包的链接（格式通常是 <a href="package-name/">）
        return f'<a href="{target_package}/"' in response.text
    except requests.exceptions.RequestException as e:
        print(f"请求出错: {str(e)}")
        return False

# 示例调用
package = "package-1"
index = "https://mytestdomain.com/pypi/pypi/simple/"
exists = check_package_via_requests(package, index, username="your_user", password="your_pass")
print(f"包 {package} 是否存在: {exists}")

注意：部分索引可能会对包名做特殊处理（比如替换下划线为连字符），需要根据实际索引的规则调整包名的转换逻辑。

方法三：调用pip CLI命令（你已尝试方式的脚本化） #

如果更倾向于用熟悉的CLI逻辑，可以在Python中通过subprocess调用pip download命令，通过返回码判断是否存在：

import subprocess

def check_package_via_cli(package_name, index_url, username=None, password=None):
    # 构建pip命令
    cmd = [
        "pip", "download", "--no-deps", "--dry-run", package_name,
        "--index-url", index_url
    ]
    # 添加认证信息（如果需要）
    if username and password:
        cmd.extend(["--username", username, "--password", password])
    
    try:
        # 执行命令，捕获输出
        result = subprocess.run(cmd, capture_output=True, text=True, check=True)
        return True
    except subprocess.CalledProcessError as e:
        # pip download失败通常意味着包不存在或认证错误
        if "Could not find a version" in e.stderr:
            return False
        else:
            print(f"命令执行出错: {e.stderr}")
            return False

# 示例调用
package = "package-1"
index = "https://mytestdomain.com/pypi/pypi/simple/"
exists = check_package_via_cli(package, index, username="your_user", password="your_pass")
print(f"包 {package} 是否存在: {exists}")

注意：这种方法依赖系统中的pip环境，需要确保脚本运行环境的pip配置正确，且命令权限足够。

内容的提问来源于stack exchange，提问作者DilLip_Chowdary