import re
regex = re.compile(r"def\s+(\w+)\((\w+),\s*(\w+)\):")
test_str = "def wrloc(imm_value, imm_address):"
subst = ".macro \\1_imm \\2 \\3"
result = regex.sub(subst, test_str, count=1)
if result:
print(result)
Please keep in mind that these code samples are automatically generated and are not guaranteed to work. If you find any syntax errors, feel free to submit a bug report. For a full regex reference for Python, please visit: https://docs.python.org/3/library/re.html