From 86f1b44117580a66c639ebb327137b50dc414fe4 Mon Sep 17 00:00:00 2001
From: Lucas Cimon <925560+Lucas-C@users.noreply.github.com>
Date: Wed, 23 Sep 2020 11:00:13 +0200
Subject: [PATCH] fix(pip_setup): Allow setup.py to import stuff from their own
package (#6545)
---
data/extract.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/data/extract.py b/data/extract.py
index eca591cf16..d384f54483 100644
--- a/data/extract.py
+++ b/data/extract.py
@@ -2,6 +2,7 @@ import sys
import json
import os
import distutils.core
+from os.path import dirname, realpath
if sys.version_info[:2] >= (3, 3):
from importlib.machinery import SourceFileLoader
@@ -32,6 +33,8 @@ except ImportError:
@mock.patch.object(setuptools, 'setup')
@mock.patch.object(distutils.core, 'setup')
def invoke(mock1, mock2):
+ # Inserting the parent directory of the target setup.py in Python import path:
+ sys.path.append(dirname(realpath(sys.argv[-1])))
# This is setup.py which calls setuptools.setup
load_source('_target_setup_', sys.argv[-1])
# called arguments are in `mock_setup.call_args`
--
GitLab