summaryrefslogtreecommitdiff
path: root/lookup.py
diff options
context:
space:
mode:
Diffstat (limited to 'lookup.py')
-rw-r--r--lookup.py8
1 files changed, 6 insertions, 2 deletions
diff --git a/lookup.py b/lookup.py
index 278072db..6da0ecbf 100644
--- a/lookup.py
+++ b/lookup.py
@@ -222,6 +222,10 @@ def _decorate(astmodule):
astmodule.Module.scope_lookup = scope_lookup
astmodule.GenExpr.scope_lookup = scope_lookup
for name in ('Class', 'Function', 'Lambda',
- 'For', 'ListCompFor', 'GenExprFor',
- 'AssName', 'Name', 'Const'):
+ 'For', 'Name', 'Const'):
globals()[name] = getattr(astmodule, name)
+ if hasattr(astmodule, 'ListCompFor'):
+ for name in ('ListCompFor', 'GenExprFor', 'AssName',):
+ globals()[name] = getattr(astmodule, name)
+ #else: XXX
+