diff options
Diffstat (limited to 'setuptools/pep517.py')
| -rw-r--r-- | setuptools/pep517.py | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/setuptools/pep517.py b/setuptools/pep517.py index b3535790..4ba23552 100644 --- a/setuptools/pep517.py +++ b/setuptools/pep517.py @@ -68,6 +68,11 @@ def build_wheel(wheel_directory, config_settings=None, shutil.rmtree(wheel_directory) shutil.copytree('dist', wheel_directory) + wheels = [f for f in os.listdir(wheel_directory) + if f.endswith('.whl')] + + assert len(wheels) == 1 + return wheels[0] def build_sdist(sdist_directory, config_settings=None): config_settings = fix_config(config_settings) @@ -78,3 +83,9 @@ def build_sdist(sdist_directory, config_settings=None): if sdist_directory != 'dist': shutil.rmtree(sdist_directory) shutil.copytree('dist', sdist_directory) + + sdists = [f for f in os.listdir(sdist_directory) + if f.endswith('.tar.gz')] + + assert len(sdists) == 1 + return sdists[0]
\ No newline at end of file |