以下は、 astおよびsymtable パッケージを使用した Python コード スニペットです。コードを解析して型をチェックしようとしています。しかし、参照されている実際の変数に到達するためにオブジェクトをトラバースする方法がわかりません。
次のコードは NodeVisitor を実装し、関数がコンパイラに提示され、コンパイラによって解析され、ast が実行されます。分析される関数 (eval_types) には、いくつかのオブジェクトが渡されます。
以下は、この例を構成するコード チャンクです。各チャンクにいくつかのコメントを追加しました。コードを実行するには、「チャンク」を再構築する必要があります。
解析のためにコード ブロックのインデントを解除するインポートと関数。
import inspect
import ast
import symtable
from tokenize import generate_tokens, untokenize, INDENT
from cStringIO import StringIO
# _dedent borrowed from the myhdl package (www.myhdl.org)
def _dedent(s):
"""Dedent python code string."""
result = [t[:2] for t in generate_tokens(StringIO(s).readline)]
# set initial indent to 0 if any
if result[0][0] == INDENT:
result[0] = (INDENT, '')
return untokenize(result)
以下はノード ビジターです。一般的な unhandled および name ビジター オーバーロードがあります。
class NodeVisitor(ast.NodeVisitor):
def __init__(self, SymbolTable):
self.symtable = SymbolTable
for child in SymbolTable.get_children():
self.symtable = child
print(child.get_symbols())
def _visit_children(self, node):
"""Determine if the node has children and visit"""
for _, value in ast.iter_fields(node):
if isinstance(value, list):
for item in value:
if isinstance(item, ast.AST):
print(' visit item %s' % (type(item).__name__))
self.visit(item)
elif isinstance(value, ast.AST):
print(' visit value %s' % (type(value).__name__))
self.visit(value)
def generic_visit(self, node):
print(type(node).__name__)
self._visit_children(node)
def visit_Name(self, node):
print(' variable %s type %s' % (node.id,
self.symtable.lookup(node.id)))
print(dir(self.symtable.lookup(node.id)))
以下は、AST で解析および分析される関数で使用されるいくつかの単純なクラスです。
class MyObj(object):
def __init__(self):
self.val = None
class MyObjFloat(object):
def __init__(self):
self.x = 1.
class MyObjInt(object):
def __init__(self):
self.x = 1
class MyObjObj(object):
def __init__(self):
self.xi = MyObjInt()
self.xf = MyObjFloat()
以下はテスト関数です。eval_types 関数は、AST で分析される関数です。
def testFunc(x,y,xo,z):
def eval_types():
z.val = x + y + xo.xi.x + xo.xf.x
return eval_types
サンプルを実行し、関数をコンパイルして分析するコード。
if __name__ == '__main__':
z = MyObj()
print(z.val)
f = testFunc(1, 2, MyObjObj(), z)
f()
print(z.val)
s = inspect.getsource(f)
s = _dedent(s)
print(type(s))
print(s)
SymbolTable = symtable.symtable(s,'string','exec')
tree = ast.parse(s)
v = NodeVisitor(SymbolTable)
v.visit(tree)
以下は、最初の名前の訪問までの出力例です。
Module
visit item FunctionDef
FunctionDef
visit value arguments
arguments
visit item Assign
Assign
visit item Attribute
Attribute
visit value Name
variable z type <symbol 'z'>
['_Symbol__flags', '_Symbol__name', '_Symbol__namespaces',
'_Symbol__scope', '__class__', '__delattr__', '__dict__',
'__doc__', '__format__', '__getattribute__', '__hash__',
'__init__', '__module__', '__new__', '__reduce__', '__reduce_ex__',
'__repr__', '__setattr__', '__sizeof__', '__str__',
'__subclasshook__', '__weakref__', 'get_name', 'get_namespace',
'get_namespaces', 'is_assigned', 'is_declared_global',
'is_free', 'is_global', 'is_imported', 'is_local',
'is_namespace', 'is_parameter', 'is_referenced']
ノード ビジターの作成は悪くないように思えますが、オブジェクト階層をトラバースする方法がわかりません。一般に、アクセスされる変数はオブジェクトの奥深くに埋め込まれている可能性があります。ast ビジターからアクセスされている実際の変数を取得する方法は? オブジェクトがノードにあることだけがわかりますが、結果変数へのアクセスに関する追加情報はありません。