diff -r 901d7ba146ad roundup/cgi/PageTemplates/PythonExpr.py --- a/roundup/cgi/PageTemplates/PythonExpr.py Thu Sep 27 11:38:05 2018 +0000 +++ b/roundup/cgi/PageTemplates/PythonExpr.py Mon Mar 04 14:46:47 2019 -0800 @@ -17,6 +17,8 @@ """Generic Python Expression Handler """ +import symtable + from .TALES import CompilerError from sys import exc_info @@ -31,7 +33,8 @@ self.expr = expr = expr.strip().replace('\n', ' ') try: d = {} - exec('def f():\n return %s\n' % expr.strip(), d) + self.f_code = 'def f():\n return %s\n' % expr.strip() + exec(self.f_code, d) self._f = d['f'] except: raise CompilerError(('Python expression error:\n' @@ -40,10 +43,31 @@ def _get_used_names(self): self._f_varnames = vnames = [] - for vname in self._f.__code__.co_names: - if vname[0] not in '$_': + for vname in self._get_from_symtab(): + if vname[0] not in '$_.': vnames.append(vname) + def _get_from_symtab(self): + """ + Get the variables used in the 'f' function. + """ + vars = set() + table = symtable.symtable(self.f_code, "", "exec") + if table.has_children(): + vars.update(self._walk_children(table)) + return vars + + def _walk_children(self, sym): + """ + Get the variables at this level. Recurse to get them all. + """ + vars = set() + for child in sym.get_children(): + vars.update(set(child.get_identifiers())) + if child.has_children(): + vars.update(self._walk_children(child)) + return vars + def _bind_used_names(self, econtext, _marker=[]): # Bind template variables names = {'CONTEXTS': econtext.contexts}