codeparser: simplify how we compare the called node names

With the previous method, using the compare_name methods, we split the
requested match name by '.', reversed it, then compared them piecemeal
during the node traversal. The new method walks the nodes and hands back
the name of what's being called, and then we check that. This also
consolidates the two different implementations of traversal of the
attribute/name nodes (one in compare_name, one for the execs).

(Bitbake rev: 84e535b5165c7e936c5b1486bdf4626ed3649f5f)

Signed-off-by: Christopher Larson <kergoth@gmail.com>
Signed-off-by: Richard Purdie <richard.purdie@linuxfoundation.org>
This commit is contained in:
Christopher Larson 2011-10-27 22:37:11 -07:00 committed by Richard Purdie
parent ada59bde67
commit 28ca6cc34b
1 changed files with 20 additions and 52 deletions

View File

@ -155,40 +155,6 @@ class PythonParser():
expands = ("d.expand", "bb.data.expand", "data.expand") expands = ("d.expand", "bb.data.expand", "data.expand")
execfuncs = ("bb.build.exec_func", "bb.build.exec_task") execfuncs = ("bb.build.exec_func", "bb.build.exec_task")
@classmethod
def _compare_name(cls, strparts, node):
"""Given a sequence of strings representing a python name,
where the last component is the actual Name and the prior
elements are Attribute nodes, determine if the supplied node
matches.
"""
if not strparts:
return True
current, rest = strparts[0], strparts[1:]
if isinstance(node, ast.Attribute):
if current == node.attr:
return cls._compare_name(rest, node.value)
elif isinstance(node, ast.Name):
if current == node.id:
return True
return False
@classmethod
def compare_name(cls, value, node):
"""Convenience function for the _compare_node method, which
can accept a string (which is split by '.' for you), or an
iterable of strings, in which case it checks to see if any of
them match, similar to isinstance.
"""
if isinstance(value, basestring):
return cls._compare_name(tuple(reversed(value.split("."))),
node)
else:
return any(cls.compare_name(item, node) for item in value)
@classmethod @classmethod
def warn(cls, func, arg): def warn(cls, func, arg):
"""Warn about calls of bitbake APIs which pass a non-literal """Warn about calls of bitbake APIs which pass a non-literal
@ -206,39 +172,41 @@ class PythonParser():
"not a literal", funcstr, argstr) "not a literal", funcstr, argstr)
def visit_Call(self, node): def visit_Call(self, node):
if self.compare_name(self.getvars, node.func): name = self.called_node_name(node.func)
if name in self.getvars:
if isinstance(node.args[0], ast.Str): if isinstance(node.args[0], ast.Str):
self.var_references.add(node.args[0].s) self.var_references.add(node.args[0].s)
else: else:
self.warn(node.func, node.args[0]) self.warn(node.func, node.args[0])
elif self.compare_name(self.expands, node.func): elif name in self.expands:
if isinstance(node.args[0], ast.Str): if isinstance(node.args[0], ast.Str):
self.warn(node.func, node.args[0]) self.warn(node.func, node.args[0])
self.var_expands.add(node.args[0].s) self.var_expands.add(node.args[0].s)
elif isinstance(node.args[0], ast.Call) and \ elif isinstance(node.args[0], ast.Call) and \
self.compare_name(self.getvars, node.args[0].func): self.called_node_name(node.args[0].func) in self.getvars:
pass pass
else: else:
self.warn(node.func, node.args[0]) self.warn(node.func, node.args[0])
elif self.compare_name(self.execfuncs, node.func): elif name in self.execfuncs:
if isinstance(node.args[0], ast.Str): if isinstance(node.args[0], ast.Str):
self.var_execs.add(node.args[0].s) self.var_execs.add(node.args[0].s)
else: else:
self.warn(node.func, node.args[0]) self.warn(node.func, node.args[0])
elif isinstance(node.func, ast.Name): elif name and isinstance(node.func, (ast.Name, ast.Attribute)):
self.execs.add(node.func.id) self.execs.add(name)
elif isinstance(node.func, ast.Attribute):
# We must have a qualified name. Therefore we need def called_node_name(self, node):
# to walk the chain of 'Attribute' nodes to determine """Given a called node, return its original string form"""
# the qualification. components = []
attr_node = node.func.value while node:
identifier = node.func.attr if isinstance(node, ast.Attribute):
while isinstance(attr_node, ast.Attribute): components.append(node.attr)
identifier = attr_node.attr + "." + identifier node = node.value
attr_node = attr_node.value elif isinstance(node, ast.Name):
if isinstance(attr_node, ast.Name): components.append(node.id)
identifier = attr_node.id + "." + identifier return '.'.join(reversed(components))
self.execs.add(identifier) else:
break
def __init__(self): def __init__(self):
self.var_references = set() self.var_references = set()