aboutsummaryrefslogtreecommitdiffstats
path: root/lib/mako/ast.py
diff options
context:
space:
mode:
Diffstat (limited to 'lib/mako/ast.py')
-rw-r--r--lib/mako/ast.py2
1 files changed, 2 insertions, 0 deletions
diff --git a/lib/mako/ast.py b/lib/mako/ast.py
index 5556269..db51a88 100644
--- a/lib/mako/ast.py
+++ b/lib/mako/ast.py
@@ -47,6 +47,8 @@ class PythonCode(object):
def _add_declared(s, name):
if not s.in_function:
self.declared_identifiers.add(name)
+ def visitClass(self, node, *args):
+ self._add_declared(node.name)
def visitAssName(self, node, *args):
self._add_declared(node.name)
def visitAssign(self, node, *args):