Скомпилировать python AST к методу - PullRequest
Новая
       3

Скомпилировать python AST к методу

10 голосов
/ 15 августа 2011

Я экспериментировал с AST в python. Я хочу изменить методы путем преобразования AST во время выполнения.

Я могу получить источник предварительно скомпилированного метода, используя inspect.getsource(), и Я могу изменить AST по мере необходимости, используя посетителей AST.

Это, вероятно, довольно наивно, но я бы хотел иметь возможность скомпилировать AST и сделать что-то похожее на: 

myClass.method.__func__.__code__ = compile(newAST, '<string>', 'exec')

Но компиляция будет принимать только AST с модулем ast.Module в качестве пользователя root. Есть ли способ либо скомпилировать просто ast.FunctionDef, либо получить объект кода функции из скомпилированного (и в противном случае пустого) кода модуля?

Буду признателен за любые указатели на информацию, охватывающую подобные вещи. Примеры AST, которые я видел, имеют дело только с простыми выражениями.

Я понял, что мне нужно просто выполнить модуль в пространстве имен, и тогда я получу доступ к нормальной структуре. Итак, картина такова: 

src = inspect.getsource(myClass.myMethod)
astFromSrc = ast.parse(unindent(src))         # need to remove extra indent from method
transform(astFromSrc.body)                    # transform the AST as you need 
ast.fix_missing_locations(astFromSrc)         # fix up line numbers etc
compiled = compile(astFromSrc, '<string>', 'exec') # compile the module AST

#######  From here is the part I was missing
myScope = {}                                  # make an empty namespace
exec compiled in myScope  # now myScope contains a proper compiled function

# Now replace the original with the modified version
myClass.myMethod.__func__.__code__ = myScope['myMethod'].__code__

Но теперь у меня есть еще один вопрос: Как вставить это в обычный процесс сборки Python, чтобы изменения выполнялись только один раз, а затем загружаться из файла .pyc?

1 Ответ

1 голос
/ 15 августа 2011

Вам не следует задавать новый вопрос в своем собственном ответе, а создавать для этого отдельный вопрос.

Обратите внимание, что ваш первый и второй вопрос также как бы противоречат друг другу. Сначала вы хотите что-то делать во время выполнения, а затем вы хотите записать их в файл, чтобы вам не приходилось делать это снова и снова. Пожалуйста, укажите, что ваша цель такова, чтобы мы четко знали, чего вы хотите.

Похоже, вы могли бы с таким же успехом создать новый файл .py после изменения ast файла, который вы проанализировали, как объяснено в Разобрать файл .py, прочитать AST, изменить его, а затем записать обратно модифицированный исходный код

Затем вы получите свой pyc-файл, скомпилировав недавно созданный py-файл.

...