From c4414287dde5e664205d0ba8d8aaa938c876411f Mon Sep 17 00:00:00 2001
From: Andrew Murray <radarhere@users.noreply.github.com>
Date: Sun, 18 Oct 2020 22:32:44 +1100
Subject: [PATCH] Added --remove-dist-egg option

---
 setup.py | 15 ++++++++++++++-
 1 file changed, 14 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 15a6ec282..9f406b7c8 100755
--- a/setup.py
+++ b/setup.py
@@ -857,8 +857,12 @@ ext_modules = [
 with open("README.md") as f:
     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:
-    setup(
+    distribution = setup(
         name=NAME,
         version=PILLOW_VERSION,
         description="Python Imaging Library (Fork)",
@@ -921,3 +925,12 @@ which was requested by the option flag --enable-{str(err)}
 """
     sys.stderr.write(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