diff options
| author | Anderson Bravalheri <andersonbravalheri@gmail.com> | 2022-02-04 11:16:12 +0000 |
|---|---|---|
| committer | Anderson Bravalheri <andersonbravalheri@gmail.com> | 2022-03-05 09:58:49 +0000 |
| commit | ccd2f073171065ad8fe65215ff837644689c6d85 (patch) | |
| tree | f7ecc0d0132c13f814740a53397d68db064523cf /tools | |
| parent | 73715080cf0fbd5ae65b4f0f5af0e651adda7234 (diff) | |
| download | python-setuptools-git-ccd2f073171065ad8fe65215ff837644689c6d85.tar.gz | |
Ensure relative imports for vendorised tomli
Diffstat (limited to 'tools')
| -rw-r--r-- | tools/vendored.py | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/tools/vendored.py b/tools/vendored.py index 53185437..c1839711 100644 --- a/tools/vendored.py +++ b/tools/vendored.py @@ -94,6 +94,16 @@ def rewrite_more_itertools(pkg_files: Path): more_file.write_text(text) +def rewrite_tomli(pkg_files, new_root): + """ + Rewrite imports in tomli to use the relative form. + """ + for file in pkg_files.glob('*.py'): + text = file.read_text().replace('tomli.', '.') + text = text.replace('tomli', f'{new_root}.tomli') + file.write_text(text) + + def clean(vendor): """ Remove all files out of the vendor directory except the meta @@ -139,6 +149,7 @@ def update_setuptools(): rewrite_importlib_resources(vendor / 'importlib_resources', 'setuptools.extern') rewrite_importlib_metadata(vendor / 'importlib_metadata', 'setuptools.extern') rewrite_more_itertools(vendor / "more_itertools") + rewrite_tomli(vendor / 'tomli', 'setuptools.extern') def install_validate_pyproject(vendor): |