Разбор и изменение структуры кода, безусловно, возможны с помощью astмодуля, и я покажу это на примере ниже. Однако запись измененного исходного кода невозможна только с astодним модулем. Для этой работы доступны и другие модули, например, один здесь .
ПРИМЕЧАНИЕ. Пример, приведенный ниже, можно рассматривать как вводное руководство по использованию astмодуля, но более полное руководство по использованию astмодуля доступно здесь, в руководстве Green Tree snakes и официальной документации по astмодулю .
Введение в ast:
>>> import ast
>>> tree = ast.parse("print 'Hello Python!!'")
>>> exec(compile(tree, filename="<ast>", mode="exec"))
Hello Python!!
Вы можете проанализировать код Python (представленный в виде строки), просто вызвав API ast.parse(). Это возвращает дескриптор в структуру абстрактного синтаксического дерева (AST). Интересно, что вы можете скомпилировать эту структуру и выполнить ее, как показано выше.
Другой очень полезный API - ast.dump()это дамп всего AST в виде строки. Он может использоваться для проверки древовидной структуры и очень полезен при отладке. Например,
На Python 2.7:
>>> import ast
>>> tree = ast.parse("print 'Hello Python!!'")
>>> ast.dump(tree)
"Module(body=[Print(dest=None, values=[Str(s='Hello Python!!')], nl=True)])"
На Python 3.5:
>>> import ast
>>> tree = ast.parse("print ('Hello Python!!')")
>>> ast.dump(tree)
"Module(body=[Expr(value=Call(func=Name(id='print', ctx=Load()), args=[Str(s='Hello Python!!')], keywords=[]))])"
Обратите внимание на разницу в синтаксисе для оператора печати в Python 2.7 по сравнению с Python 3.5 и разницу в типе узла AST в соответствующих деревьях.
Как изменить код, используя ast:
Теперь давайте рассмотрим пример модификации кода Python astмодулем. Основным инструментом для изменения структуры AST является ast.NodeTransformerкласс. Всякий раз, когда нужно модифицировать AST, ему / ей нужно подклассы из него и написать Node Transformation соответственно.
Для нашего примера давайте попробуем написать простую утилиту, которая преобразует Python 2, операторы print в вызовы функций Python 3.
Вывести оператор в утилиту конвертирования вызовов Fun: print2to3.py:
#!/usr/bin/env python
'''
This utility converts the python (2.7) statements to Python 3 alike function calls before running the code.
USAGE:
python print2to3.py <filename>
'''
import ast
import sys
class P2to3(ast.NodeTransformer):
def visit_Print(self, node):
new_node = ast.Expr(value=ast.Call(func=ast.Name(id='print', ctx=ast.Load()),
args=node.values,
keywords=[], starargs=None, kwargs=None))
ast.copy_location(new_node, node)
return new_node
def main(filename=None):
if not filename:
return
with open(filename, 'r') as fp:
data = fp.readlines()
data = ''.join(data)
tree = ast.parse(data)
print "Converting python 2 print statements to Python 3 function calls"
print "-" * 35
P2to3().visit(tree)
ast.fix_missing_locations(tree)
# print ast.dump(tree)
exec(compile(tree, filename="p23", mode="exec"))
if __name__ == '__main__':
if len(sys.argv) <=1:
print ("\nUSAGE:\n\t print2to3.py <filename>")
sys.exit(1)
else:
main(sys.argv[1])
Эту утилиту можно попробовать на небольшом примере файла, например, приведенном ниже, и она должна работать нормально.
Тестовый входной файл: py2.py
class A(object):
def __init__(self):
pass
def good():
print "I am good"
main = good
if __name__ == '__main__':
print "I am in main"
main()
Обратите внимание, что приведенное выше преобразование предназначено только для astцелей обучения, и в реальном случае нужно будет рассмотреть все различные сценарии, такие как print " x is %s" % ("Hello Python").