Как я извлекаю имена из простой функции?

У меня есть эта часть кода:

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 модуль здесь.

10
задан Eddy Pronk 12 July 2010 в 09:08
поделиться

3 ответа

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']
5
ответ дан 4 December 2019 в 03:15
поделиться

Я не уверен, зачем вам нужно восстанавливать имена, очень грубый способ получить все имена и точки в функции -

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

.
1
ответ дан 4 December 2019 в 03:15
поделиться

Я еще не использовал новый модуль 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').

0
ответ дан 4 December 2019 в 03:15
поделиться
Другие вопросы по тегам:

Похожие вопросы: