Whoa now, don't *always* do this--only when we're actually running se…

…tup.py
astropy · Sep 23, 2015 · 7c427d8 · 7c427d8
1 parent 2f58cbe
commit 7c427d8
Showing 1 changed file with 8 additions and 5 deletions.
diff --git 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