better checking for setup.py in module init

samizdatco · samizdatco · commit fe6e62723eb2 · 2014-06-24T19:21:26.000-04:00
diff --git a/plotdevice/__init__.py b/plotdevice/__init__.py
@@ -11,7 +11,7 @@ class DeviceError(Exception):
 import sys, re
 called_from = getattr(sys.modules['__main__'], '__file__', '<interactive>')
 is_windowed = bool(re.search(r'plotdevice(-app|/run/console)\.py$', called_from))
-in_setup = bool(sys.modules['__main__'].__file__.endswith('setup.py'))
+in_setup = bool(called_from.endswith('setup.py'))
 
 # add the Extras directory to sys.path since every module depends on PyObjC and friends
 try:
