Как сделать макросы в функции Python?

Question

[Иван Четчасов]

Дарова хабровчане! Я писал библиотеку, где мне понадобилось во всех обрабатываемых функциях заменять элементы кода, например:

def loop(t, spc, ID, num):
    """3n+1_mainloop"""

    steps = 0

    while num not in (4, 2, 1):
        num = hdl.getrun(num)
        steps += 1
        pass
    return steps

превратилось в

def loop(t, spc, ID, num):
    """3n+1_mainloop"""

    steps = 0

    while num not in (4, 2, 1):
        num = hdl.getrun(num)
        steps += 1
        pass
    t.ret = True
    spc.rets[ID] = steps
    return

Как это сделать?

Answer 1

[Vindicar]

В питоне нет макросов. Но вашу задачу вполне можно решить с помощью декораторов.

Но если изменение разовое... это не имеет смысла. Проще переписать один раз и начисто.

Comments

[Иван Четчасов]

нет, про декораторы я знаю. мне нужно понять, как это в декораторе реализовать.

[Vindicar]

Иван Четчасов, ну если на базе примера, то очень просто.

from functools import wraps

def decorate_func(func):
    #Декоратор @wraps позволяет ближе имитировать "внешний вид" декорируемой ф-ии.
    @wraps(func)
    def wrapped_func(t, spc, ID, num):
        steps = func(t, spc, ID, num)
        t.ret = True
        spc.rets[ID] = steps
        return steps
    return wrapped_func #эта функция заменит собой декорируемую

Но это предполагает что добавляемый участок кода один для всех, и что набор параметров тоже одинаковый у всех.

[Иван Четчасов]

Vindicar, это немного не то, что мне нужно. Мне нужно заменить участок кода, а не добавить его в конец.

[Vindicar]

Иван Четчасов, я не вижу, что в примере заменяется. return? Так он нужен.
А если пример не показательный, то ничего не поделаешь.

[Иван Четчасов]

А, все, понял. Я сделал, по сути, так же, только без врапперов. Отмечу ваш ответ верным.