#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
自动生成 pip 源配置文件，支持 Python 2 / Python 3，兼容 Windows / Linux / macOS。
用法：
python configure_pip_mirror.py # 默认使用阿里云源
python configure_pip_mirror.py --source douban # 使用豆瓣源
python configure_pip_mirror.py -s tsinghua # 使用清华源
python configure_pip_mirror.py -s aliyun # 使用阿里云源
python configure_pip_mirror.py --source ustc # 使用中科大源
python configure_pip_mirror.py --list # 列出所有可选源
python configure_pip_mirror.py --url https://xxx/simple # 自定义源地址
python configure_pip_mirror.py --url https://xxx/simple --trusted-host xxx # 自定义
"""
from __future__ import print_function, unicode_literals
import os
import platform
import sys
import json
# 兼容 Python 2 的 argparse 和 io.open
if sys.version_info[0] == 2:
from io import open
try:
import argparse
except ImportError:
print("错误: Python 2 需要安装 argparse 模块: pip install argparse")
sys.exit(1)
else:
import argparse
# ============================================================
# 预定义的镜像源
# ============================================================
MIRRORS = {
"aliyun": {
"name": "阿里云",
"index_url": "http://mirrors.aliyun.com/pypi/simple/",
"trusted_host": "mirrors.aliyun.com",
"homepage": "https://mirrors.aliyun.com",
},
"douban": {
"name": "豆瓣",
"index_url": "http://pypi.douban.com/simple",
"trusted_host": "pypi.douban.com",
"homepage": "http://pypi.douban.com",
},
"tsinghua": {
"name": "清华大学",
"index_url": "https://pypi.tuna.tsinghua.edu.cn/simple",
"trusted_host": "pypi.tuna.tsinghua.edu.cn",
"homepage": "https://mirrors.tuna.tsinghua.edu.cn",
},
"ustc": {
"name": "中国科学技术大学",
"index_url": "https://pypi.mirrors.ustc.edu.cn/simple",
"trusted_host": "pypi.mirrors.ustc.edu.cn",
"homepage": "https://mirrors.ustc.edu.cn",
},
"huawei": {
"name": "华为云",
"index_url": "https://repo.huaweicloud.com/repository/pypi/simple",
"trusted_host": "repo.huaweicloud.com",
"homepage": "https://mirrors.huaweicloud.com",
},
"tencent": {
"name": "腾讯云",
"index_url": "https://mirrors.cloud.tencent.com/pypi/simple",
"trusted_host": "mirrors.cloud.tencent.com",
"homepage": "https://mirrors.cloud.tencent.com",
},
"netease": {
"name": "网易",
"index_url": "http://mirrors.163.com/pypi/simple",
"trusted_host": "mirrors.163.com",
"homepage": "http://mirrors.163.com",
},
"pypi": {
"name": "PyPI 官方",
"index_url": "https://pypi.org/simple",
"trusted_host": "pypi.org",
"homepage": "https://pypi.org",
},
}
def expanduser(path):
"""安全的 expanduser，处理 Windows 下 ~/ 开头的路径问题。"""
expanded = os.path.expanduser(path)
if path.startswith('~/') and expanded.startswith('//'):
expanded = expanded[1:]
return expanded
def get_pip_config_path():
"""获取 pip 配置文件的完整路径（跨平台）。"""
system = platform.system()
user_dir = expanduser('~')
if system == 'Windows':
config_basename = 'pip.ini'
legacy_storage_dir = os.path.join(user_dir, 'pip')
else:
config_basename = 'pip.conf'
legacy_storage_dir = os.path.join(user_dir, '.pip')
return os.path.join(legacy_storage_dir, config_basename)
def build_config_content(index_url, trusted_host):
"""生成 pip 配置文件内容。"""
return r'''[global]
index-url = {index_url}
[install]
trusted-host = {trusted_host}
'''.format(index_url=index_url, trusted_host=trusted_host)
def list_mirrors():
"""打印所有预定义源。"""
header = "{:<15} {:<16} {:<36} {}".format("别名", "名称", "源地址", "可信主机")
print(header)
print("-" * 95)
for key, mirror in sorted(MIRRORS.items()):
print("{:<15} {:<16} {:<36} {}".format(
key, mirror["name"], mirror["index_url"], mirror["trusted_host"]))
print()
print("使用方式: python configure_pip_mirror.py -s <别名>")
print("示例: python configure_pip_mirror.py -s tsinghua")
def create_pip_config(index_url, trusted_host, force=False):
"""生成 pip 源配置文件，如有旧文件则备份。"""
file_path = get_pip_config_path()
dirname = os.path.dirname(file_path)
basename = os.path.basename(file_path)
print('当前系统: {}'.format(platform.system()))
print('配置文件路径: {}'.format(file_path))
# 创建目录（如果不存在）
try:
os.makedirs(dirname)
print('创建目录: {}'.format(dirname))
except OSError:
pass # 目录已存在
# 备份已有配置文件
if os.path.isfile(file_path):
if force:
os.remove(file_path)
print('已删除旧配置 (--force): {}'.format(file_path))
else:
backup_path = os.path.join(dirname, basename + '.bak')
if os.path.isfile(backup_path):
os.remove(backup_path)
os.rename(file_path, backup_path)
print('已备份原配置至: {}'.format(backup_path))
# 写入新配置
content = build_config_content(index_url, trusted_host)
with open(file_path, 'w', encoding='utf-8') as f:
f.write(content)
print('✅ pip 源配置文件已生成: {}'.format(file_path))
print(' 源地址: {}'.format(index_url))
print(' 可信主机: {}'.format(trusted_host))
return file_path
def parse_args(argv=None):
"""解析命令行参数。"""
parser = argparse.ArgumentParser(
description='自动生成 pip 源配置文件（支持 Python 2/3，跨平台）',
formatter_class=argparse.RawDescriptionHelpFormatter,
epilog="""
示例:
%(prog)s 使用默认阿里云源
%(prog)s -s tsinghua 使用清华源
%(prog)s --source douban 使用豆瓣源
%(prog)s --list 列出所有可选源
%(prog)s -u https://pypi.org/simple -t pypi.org 自定义源
%(prog)s -s aliyun -f 使用阿里云源，不备份直接覆盖
""",
)
parser.add_argument(
'-s', '--source',
metavar='别名',
help='使用预定义源 (如 aliyun, douban, tsinghua, ustc 等)，'
'使用 --list 查看所有可选源',
)
parser.add_argument(
'-u', '--url',
metavar='地址',
help='自定义 PyPI 源完整地址 (如 https://pypi.tuna.tsinghua.edu.cn/simple)',
)
parser.add_argument(
'-t', '--trusted-host',
metavar='域名',
help='自定义可信主机域名 (如 pypi.tuna.tsinghua.edu.cn)',
)
parser.add_argument(
'-f', '--force',
action='store_true',
help='不备份已有配置文件，直接覆盖',
)
parser.add_argument(
'--list',
action='store_true',
dest='show_list',
help='列出所有预定义源及其别名',
)
parser.add_argument(
'--json',
action='store_true',
dest='show_json',
help='以 JSON 格式输出所有预定义源信息',
)
args = parser.parse_args(argv)
# 处理 --list 和 --json 这两个纯展示参数
if args.show_list:
list_mirrors()
sys.exit(0)
if args.show_json:
print(json.dumps(MIRRORS, indent=2, ensure_ascii=False))
sys.exit(0)
return args
def resolve_source(args):
"""
解析参数，确定最终的 index_url 和 trusted_host。
优先级: --url > --source > 默认 (aliyun)
"""
index_url = None
trusted_host = None
if args.source:
source_key = args.source.lower()
if source_key not in MIRRORS:
print("错误: 未知源 '{}'。使用 --list 查看所有可选源。".format(source_key))
sys.exit(1)
mirror = MIRRORS[source_key]
index_url = mirror["index_url"]
trusted_host = mirror["trusted_host"]
print('使用源: {} ({})'.format(mirror["name"], source_key))
if args.url:
index_url = args.url
if args.trusted_host:
trusted_host = args.trusted_host
elif not trusted_host:
# 尝试从 URL 中提取 host
try:
from urllib.parse import urlparse
except ImportError:
from urlparse import urlparse # Python 2
parsed = urlparse(index_url)
trusted_host = parsed.hostname or index_url
print('自定义源地址: {}'.format(index_url))
if not index_url:
# 默认使用阿里云
mirror = MIRRORS["aliyun"]
index_url = mirror["index_url"]
trusted_host = mirror["trusted_host"]
print('使用默认源: {} ({})'.format(mirror["name"], "aliyun"))
if args.url and not args.trusted_host and not args.source:
print('提示: 未指定 --trusted-host，已自动从 URL 解析为: {}'.format(trusted_host))
return index_url, trusted_host
def main():
args = parse_args()
index_url, trusted_host = resolve_source(args)
create_pip_config(index_url, trusted_host, force=args.force)
if __name__ == '__main__':
main()