Как различать «строку» и «фактический код» в python?

Question

Мои работы относятся к инструментарию фрагментов кода в коде python. Поэтому в моей работе я бы написал скрипт в python, чтобы взять другой файл python в качестве ввода и вставить любой необходимый код в нужное место с моим скриптом.

Следующий код представляет собой пример кода в файл, который я бы инструментирование:

A.py #normal un-instrumented code 

statements 
.... 
.... 

def move(self,a): 
    statements 
    ...... 
    print "My function is defined" 
    ...... 

statements 
...... 

Мой сценарий, что на самом деле делает это, чтобы проверить каждый строк в A.py и если есть «DEF» затем фрагмент коды инструментальный на верхней части коды функции четкости

в следующем примере, как окончательные из пут должны быть:

A.py #instrumented code 

statements 
.... 
.... 

@decorator #<------ inserted code 
def move(self,a): 
    statements 
    ...... 
    print "My function is defined" 
    ...... 

statements 
...... 

Но я привел с различным вывод. Следующий код является конечным результатом, который я получаю:

A.py #instrumented код

statements 
.... 
.... 

@decorator #<------ inserted code 
def move(self,a): 
    statements 
    ...... 
    @decorator #<------ inserted code [this should not occur] 
    print "My function is defined" 
    ...... 

statements 
...... 

Я понимаю, что в инструментованной коде признает «DEF» в слове «определенное» и т.д. он устанавливает над ним код.

В реальности инструментальный код имеет много этих проблем, я не смог правильно измерить данный файл python. Есть ли другой способ отличить фактический «def» от строки?

Спасибо

Answer 1

Используйте ast module разобрать файл должным образом.

Этот код печатает номер строки и столбца смещения каждого def заявления:

import ast 
with open('mymodule.py') as f: 
    tree = ast.parse(f.read()) 
for node in ast.walk(tree): 
    if isinstance(node, ast.FunctionDef): 
     print node.lineno, node.col_offset 

Answer 2

Вы можете использовать регулярные выражения. Чтобы избежать def внутри кавычек, то вы можете использовать отрицательные просмотровых обходные:

import re 

for line in open('A.py'): 
    m = re.search(r"(?!<[\"'])\bdef\b(?![\"'])", line) 
    if m: 
     print r'@decorator #<------ inserted code' 

    print line 

Однако могут быть другие def мест где, что вы или я не могу думать, и если мы не будем осторожны, мы в конечном-вверх пишущий парсер Python снова и снова. Предложение Джанны Карилы об использовании ast.parse, вероятно, будет более безопасным в долгосрочной перспективе.