Published on

PEP 582的开发日志

PDM 实现 PEP 582 遇到的坑

Reading Time

6 min read

PEP 582 是 Python 的一个隔离项目环境的提案。PDM 作为现有的唯一一个具有完备 PEP 582 支持的包管理器,在实现的过程中也并非一帆风顺。本文将介绍一些关键 PEP 582 特性的实现方法和历程。

加载项目包目录

这是 PEP 582 的核心,也是事实上提案唯一阐明的事情,就是项目的包都会安装在__pypackages__/X.Y/lib下面。在 Python 中如何挂载一个额外路径到包搜索路径中呢?再简单不过,就是利用环境变量PYTHONPATH。这也是现在所有 PEP 582 的实现使用的方法,包括:

  1. pythonloc,一个 PEP 582 的试验项目
  2. pyflow,一个用 Rust 做的 Python 包管理器
  3. pdm 的实现

但仅仅做到这里是不够的,在 pythonloc 的 README 中提到:

This PEP first looks to __pypackages__ but will fall back to looking in site-packages. This is not entirely hermetic and could lead to some confusion around which packages are being used. I would prefer the default search path be only __pypackages__ and nothing else.

那么如何让 Python 启动时不要加载 site-packages 呢?这个特性也是我最近才实现的。乍一看 site-packages 好像是 Python 的机制,不好做手脚,但经过一番搜索我发现了 Python 的内置模块site就是控制这个事情的:

This module is automatically imported during initialization. The automatic import can be suppressed using the interpreter’s -S option.

Importing this module will append site-specific paths to the module search path and add a few builtins, unless -S was used. In that case, this module can be safely imported with no automatic modifications to the module search path or additions to the builtins. To explicitly trigger the usual site-specific additions, call the site.main() function.

我觉得找到了解决方法:在用户pdm run python的时候,自动给他添加-S参数不就行了?看看效果:

$ pdm run python -S -c "import sys;print(sys.path)"
[
    '',
    '/Users/fming/wkspace/github/pdm-test/__pypackages__/3.8/lib',
    '/Users/fming/Library/PythonUp/versions/3.8/lib/python38.zip',
    '/Users/fming/Library/PythonUp/versions/3.8/lib/python3.8',
    '/Users/fming/Library/PythonUp/versions/3.8/lib/python3.8/lib-dynload'
]

Perfect like a shit! 不是吗?sys.path里面有本地的包目录,但没有site-packages了。问题解决了吗?没有!先喘口气,看看这个sys.path,不知有没人发现问题在哪。

好了不卖关子了,这里面缺少了.pth文件1包含的搜索路径。通常setuptools在安装可编辑(editable)包的时候会在__pypackages__/X.Y/lib下面塞一个easy-install.pth文件,用来把可编辑包的真正路径给包含进sys.path中来,而这个过程恰好是由site.py完成的,现在把它禁掉了就都没了。

那么只能走另外的路了,其实除了easy-install.pthsetuptools还会添加一个site.py来完成这个.pth 文件的加载。这个文件会在 Python 启动时执行,那就可以在这里操作sys.path去掉 site-packages 的路径了。具体改动可以看这个 PR,改完之后再看看效果:

$ pdm run python -c "import sys;print(sys.path)"
[
    '',
    '/Users/fming/wkspace/github/pdm-test/__pypackages__/3.8/lib',
    '/Users/fming/wkspace/github/pdm-test',
    '/Users/fming/Library/PythonUp/versions/3.8/lib/python38.zip',
    '/Users/fming/Library/PythonUp/versions/3.8/lib/python3.8',
    '/Users/fming/Library/PythonUp/versions/3.8/lib/python3.8/lib-dynload'
]

这其中第二个路径,就是通过easy-install.pth加载的路径。

执行可执行文件时自动加载项目包目录

除了通过pdm run加载项目包目录,我还希望在(外部)直接执行可执行文件时自动加载项目包目录。因为包都是通过 PDM 安装的,我们自然可以在可执行文件里做修改来达成这个效果。

先简单说下 PDM 安装包的过程,无论是什么形式的依赖定义,最终都会构造出一个 wheel 包的格式,再安装这个 wheel 包。如果你打开__pypackages__/X.Y/bin下面的任意一个可执行文件看,会发现它们的内容都是差不多的:

#!/Users/fming/Library/PythonUp/bin/python3
# -*- coding: utf-8 -*-
import re
import sys

from wheel.cli import main
if __name__ == '__main__':
    sys.argv[0] = re.sub(r'(-script\.pyw|\.exe)?$', '', sys.argv[0])
    sys.exit(main())

第六行是实际的入口,只有这一行会变动,于是我猜测一定是有模板填充。所以我全局搜索特征字符串,果然在distlib/scripts.py里面找到了这个模板,它是ScriptMaker类的一属性,而ScriptMaker刚好是作为wheel.install()的参数传进去的。那么解决方法就比较明显了——自己构造ScriptMaker实例,然后修改script_template属性:

maker.script_template = maker.script_template.replace(
    "import sys",
    "import sys\nsys.path.insert(0, {!r})".format(paths["platlib"]),
)

import sys后面直接加了一句,插入包的路径。

问题还没有完全解决,对于可编辑的包,并不是由 wheel 格式安装的,查看可编辑包的可执行文件,可以发现内容稍有不同:

#!/Users/fming/Library/PythonUp/bin/python3
# EASY-INSTALL-ENTRY-SCRIPT: 'pdm-test','console_scripts','pdm-test'
__requires__ = 'pdm-test'
import re
import sys
from pkg_resources import load_entry_point

if __name__ == '__main__':
    sys.argv[0] = re.sub(r'(-script\.pyw?|\.exe)?$', '', sys.argv[0])
    sys.exit(
        load_entry_point('pdm-test', 'console_scripts', 'pdm-test')()
    )

类似的,这个文件内容也是有一个模板,只是在setuptools中,解决方法也是一样,就不赘述了。

Footnotes

  1. .pth 文件中包含的路径可以被加载到sys.path中,参考官方文档

Share: