Merge pull request #771 from ros/predefined_attributes

set commonly predefines attributes when interrogating setup.py files
ros · Dec 25, 2015 · eaba48d · eaba48d
2 parents e1dc70a + 39a273d
commit eaba48d
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/cmake/interrogate_setup_dot_py.py b/cmake/interrogate_setup_dot_py.py
@@ -241,7 +241,13 @@ def main():
             pass
 
         with open(args.setupfile_path, 'r') as fh:
-            exec(fh.read(), {})
+            local_vars = {
+                '__doc__': None,
+                '__file__': os.path.abspath(args.setupfile_path),
+                '__name__': '__main__',
+                '__package__': None,
+            }
+            exec(fh.read(), {}, local_vars)
     finally:
         distutils.core.setup = distutils_backup
         try: