повторно преобразовать шаблон в функцию

Question

Я использую re.sub для замены имен в стиле ARB на имена в стиле GLSL в строке.Теперь я хочу дополнительно сохранить все преобразованные совпадения в набор строк.Могу ли я сделать это, все еще используя синтаксис «шаблона» re.sub?

Вот код:

# set of replacement rules
expl_table = [
    (r'program.env\[(\d+)\]'  , r'program_env_\1'),
    (r'program.local\[(\d+)\]', r'program_local_\1'),
]
for props in expl_table:
    (re_from, re_to) = props
    # arg = re.sub(re_from, re_to, arg)       # simple and good
    def replace_func(m):
        result = ??repl_template??(m, re_to)  # where can I find it?
        declarations.append(result)           # want to save all replacements
        return result
    arg = re.sub(re_from, replace_func, arg)

Я нашел что-то вроде _subx в источникекод , но, похоже, закрыт.Кажется, я должен реализовать это сам, как бы глупо это не звучало.

Answer 1

Вы можете изменить свою строку, перебирая ее с помощью re.finditer():

# set of replacement rules
expl_table = [
    (r'program.env\[(\d+)\]'  , r'program_env_dsdsds\1'),
    (r'program.local\[(\d+)\]', r'program_local_\1'),
]

declarations = []
for props in expl_table:
    (re_from, re_to) = props

    offset = 0
    for m in re.finditer(re_from, string):
        sub = m.expand(re_to)

        string = string[:m.start()+offset] + sub + string[m.end()+offset:]
        offset = max(map(len, [sub, m.group(0)])) - min(map(len, [sub, m.group(0)]))
        declarations.append(sub)

print(string)

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

for props in expl_table:
    (re_from, re_to) = props
    string = re.sub(re_from,
                lambda m: [
                           (result, declarations.append(result)) 
                           for result in [m.expand(re_to)]
                          ][0][0],
                 string)

print(string)
print(declarations)