python - 如何在 virtualenv 中修改 pip？

Question

我正在尝试修改 pip，以便在每次安装或卸载后，我的 requirements.txt 都会得到更新。

为此，我修改了虚拟环境的 bin/ 中的 pip 文件。

pip :

#!/Users/username/ProjectEnv/bin/python3

# -*- coding: utf-8 -*-
import re
import sys

from pip import main

if __name__ == '__main__':
    sys.argv[0] = re.sub(r'(-script\.pyw?|\.exe)?$', '', sys.argv[0])
    status = main()
    if status == 0 or status is None:
        sys.argv = ['pip', 'freeze', '>', '../Project/requirements.txt']
        sys.exit(main())

这不起作用。我尝试过打印命令行参数并设置断 pip ，但效果不佳。

pip :

#!/Users/username/ProjectEnv/bin/python3

# -*- coding: utf-8 -*-
import re
import sys
import pdb
pdb.set_trace()

from pip import main

if __name__ == '__main__':
    for arg in sys.argv:
        print(arg)
    sys.argv[0] = re.sub(r'(-script\.pyw?|\.exe)?$', '', sys.argv[0])
    status = main()
    if status == 0 or status is None:
        sys.argv = ['pip', 'freeze', '>', '../Project/requirements.txt']
        sys.exit(main())

我在这里做错了什么？

Answer 1

您可以自己在 pip 库上调用 freeze 方法。它将返回一个生成器，该生成器为您提供在 pip --freeze 中打印到屏幕的逐行输出。

from pip.operations import freeze

packages = freeze.freeze()
with open('../Project/requirements.txt', 'w') as f:
    for x in packages:
        f.write(x)
        f.write('\n')