Extract function for restoring install lib to encapsulate behavior.

5642e413 · Jason R. Coombs · 268ef5f5 · 5642e413
Commit 5642e413 authored Jul 26, 2020 by Jason R. Coombs
Hide whitespace changes
Inline Side-by-side

Showing with 7 additions and 6 deletions

setup.py setup.py +7 -6

No files found.
--- a/setup.py
+++ b/setup.py
@@ -104,14 +104,15 @@ class install_with_pth(install):
    def finalize_options(self):
        install.finalize_options(self)
+        self._restore_install_lib()
-        # undo secondary effect of `extra_path` adding to `install_lib`
+    def _restore_install_lib(self):
-        install_suffix = os.path.relpath(
+        """
-            self.install_lib,
+        Undo secondary effect of `extra_path` adding to `install_lib`
-            self.install_libbase,
+        """
-        )
+        suffix = os.path.relpath(self.install_lib, self.install_libbase)
-        if install_suffix == self._pth_contents:
+        if suffix == self._pth_contents:
            self.install_lib = self.install_libbase