У меня есть эта часть кода:
import inspect
import ast
def func(foo):
return foo.bar - foo.baz
s = inspect.getsource(func)
xx = ast.parse(s)
class VisitCalls(ast.NodeVisitor):
def visit_Name(self, what):
if what.id == 'foo':
print ast.dump(what.ctx)
VisitCalls().visit(xx)
От функции 'func' я хотел бы извлечь:
['foo.bar', 'foo.baz']
или что-то как:
(('foo', 'bar'), ('foo', 'baz))
отредактированный
Я хочу преобразовать код тривиальной функции Python к формуле электронной таблицы.
Таким образом, я должен преобразовать:
foo.bar - foo.baz
кому:
=A1-B1
демонстрационная электронная таблица http://img441.imageshack.us/img441/1451/84516405.png
** отредактированный снова*
Программа ниже выводов:
('A1', 5)
('B1', 3)
('C1', '= A1 - B1')
Код:
import ast, inspect
import codegen # by Armin Ronacher
from collections import OrderedDict
class SpreadSheetFormulaTransformer(ast.NodeTransformer):
def __init__(self, sym):
self.sym = sym
def visit_Attribute(self, node):
name = self.sym[id(eval(codegen.to_source(node)))]
return ast.Name(id=name, ctx=ast.Load())
def create(**kwargs):
class Foo(object): pass
x = Foo()
x.__dict__.update(kwargs)
return x
def register(x,y):
cell[y] = x
sym[id(x)] = y
def func(foo):
return foo.bar - foo.baz
foo = create(bar=5, baz=3)
cell = OrderedDict()
sym = {}
register(foo.bar, 'A1')
register(foo.baz, 'B1')
source = inspect.getsource(func)
tree = ast.parse(source)
guts = tree.body[0].body[0].value
SpreadSheetFormulaTransformer(sym).visit(guts)
code = '= ' + codegen.to_source(guts)
cell['C1'] = code
for x in cell.iteritems():
print x
Я нашел некоторые ресурсы здесь: внутренности Python: Работа с Python ASTs я захватил работу codegen модуль здесь.
import ast, inspect
import codegen # by Armin Ronacher
def func(foo):
return foo.bar - foo.baz
names = []
class CollectAttributes(ast.NodeVisitor):
def visit_Attribute(self, node):
names.append(codegen.to_source(node))
source = inspect.getsource(func)
tree = ast.parse(source)
guts = tree.body[0].body[0].value
CollectAttributes().visit(guts)
print names
output:
['foo.bar', 'foo.baz']
Я не уверен, зачем вам нужно восстанавливать имена, очень грубый способ получить все имена и точки в функции -
import inspect
import parser
import symbol
import token
import pprint
def func(foo):
return foo.bar - foo.baz
s = inspect.getsource(func)
st = parser.suite(s)
def search(st):
if not isinstance(st, list):
return
if st[0] in [token.NAME, token.DOT]:
print st[1],
else:
for s in st[1:]:
search(s)
search(parser.ast2list(st))
output:
def func foo return foo . bar foo . baz
Возможно, вы можете улучшить это, читая синтаксическое дерево более элегантно, я использую парсер вместо модуля ast, потому что я на python 2.5
.Я еще не использовал новый модуль ast, но у меня есть рабочий код, который использует старый compiler.ast для достижения чего-то подобного:
def visitGetattr(self, node): full_name = [node.attrname] parent = node.expr while isinstance(parent, compiler.ast.Getattr): full_name.append(parent.attrname) parent = parent.expr if isinstance(parent, compiler.ast.Name): full_name.append(parent.name) full_name = ".".join(reversed(full_name)) # do something with full_name for c in node.getChildNodes(): self.visit(c)
Код немного перефразировал,Возможно, я ввел непреднамеренные ошибки. Я надеюсь, что это даст вам общее представление: вам нужно посетить узлы Name и Getattr и построить пунктирные имена, а также иметь дело с тем фактом, что вы также увидите все промежуточные значения (например, 'foo' и 'foo.bar').