正規表現を用いて文字列を置換する場合に、一致した部分を別の関数に渡し、返り値で置換したい

Question

前提・実現したいこと

pythonで正規表現に一致した部分の一部を別の関数に渡して、処理をしたい

発生している問題

下記コードで正規表現に一致した部分を\1として他の関数(replacement_word)に渡すと、
replcement_word内ではstr型の\1になってしまいます。
正規表現に一致した部分を別の関数に渡した際に、一致した部分を文字列として関数の処理をすることはできないでしょうか。

def replacement(text):
    replacement_text = re.sub(r'[command((.+))]', replacement_word(r'\1'), text
    return replacement_text

def replacement_word(command_name):
    type(command_name)#<class 'str'>
    print(command_name)#\1
    #本来は一致した文字列で下のif文を処理したい
    if str(command_name) == 'str1':
        return 'commandは' + command_name + 'です' 
    else:
        return 'そのほかのコマンドです'

Answer 1

re.subの第二引数は関数オブジェクトを取れるので、それを使えば良いです。

6.2. re — 正規表現操作 — Python 3.5.3 ドキュメント　#re.sub

引数はマッチオブジェクトとして渡されるので、ちょっと注意が必要。

import re
def replacement(text):
    replacement_text = re.sub(r'[command((.+))]', replacement_word, text)
    return replacement_text

def replacement_word(command):
    command_name = command.group(1)
    print(command_name)
    if command_name == 'str1':
        return 'commandは' + command_name + 'です' 
    else:
        return 'そのほかのコマンドです'

print(replacement("[command(str1)]"))
print(replacement("[command(hoge)]"))
""" =>
str1
commandはstr1です
hoge
そのほかのコマンドです
"""

マッチオブジェクトの細かい仕様などはドキュメントで確認してください。

Comments

[wawawatataru]

hayataka2049様
ご回答ありがとうございます。
あげていただいたリンクとサンプルコードで問題を解決することができました。
ありがとうございました。