Added --remove-dist-egg option

This commit is contained in:
Andrew Murray 2020-10-18 22:32:44 +11:00
parent 9158037592
commit c4414287dd

View File

@ -857,8 +857,12 @@ ext_modules = [
with open("README.md") as f: with open("README.md") as f:
long_description = f.read() long_description = f.read()
remove_dist_egg = "--remove-dist-egg" in sys.argv
if remove_dist_egg:
while "--remove-dist-egg" in sys.argv:
sys.argv.remove("--remove-dist-egg")
try: try:
setup( distribution = setup(
name=NAME, name=NAME,
version=PILLOW_VERSION, version=PILLOW_VERSION,
description="Python Imaging Library (Fork)", description="Python Imaging Library (Fork)",
@ -921,3 +925,12 @@ which was requested by the option flag --enable-{str(err)}
""" """
sys.stderr.write(msg) sys.stderr.write(msg)
raise DependencyException(msg) raise DependencyException(msg)
if remove_dist_egg:
egg_path = distribution.get_command_obj('bdist_egg').egg_output
if egg_path:
os.remove(egg_path)
try:
os.rmdir(os.path.dirname(egg_path))
except OSError:
pass