From 7c427d8892f0bb4c99a9f5659aa3e8964a92d9c6 Mon Sep 17 00:00:00 2001 From: "Erik M. Bray" Date: Wed, 23 Sep 2015 12:21:52 -0400 Subject: [PATCH] Whoa now, don't *always* do this--only when we're actually running setup.py --- astropy_helpers/__init__.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/astropy_helpers/__init__.py b/astropy_helpers/__init__.py index 90ccbf58..95ca366c 100644 --- a/astropy_helpers/__init__.py +++ b/astropy_helpers/__init__.py @@ -37,8 +37,11 @@ # Ensure that all module-level code in astropy or other packages know that # we're in setup mode: -if sys.version_info[0] >= 3: - import builtins -else: - import __builtin__ as builtins -builtins._ASTROPY_SETUP_ = True +if ('__main__' in sys.modules and + hasattr(sys.modules['__main__'], '__file__') and + sys.modules['__main__'].__file__.startswith('setup.py')): + if sys.version_info[0] >= 3: + import builtins + else: + import __builtin__ as builtins + builtins._ASTROPY_SETUP_ = True