如何仅在函数(Atom)的块代码中搜索/替换？

def my_function():
    # 这里是一些代码
    old_string = "old_value"
    new_string = "new_value"
    # 更多的代码...

# 假设你想替换函数内部的 "old_value" 为 "new_value"

import re

def replace_in_function(code, function_name, old_str, new_str):
    # 定义一个正则表达式来匹配函数体
    pattern = re.compile(rf'def {function_name}\((.*?)\):\s*(.*?)(?=\ndef|\Z)', re.DOTALL)
    match = pattern.search(code)
    
    if match:
        # 获取函数体
        function_body = match.group(2)
        # 在函数体内进行替换
        updated_body = function_body.replace(old_str, new_str)
        # 替换回原代码
        updated_code = pattern.sub(rf'def {function_name}\(\1\):\n    {updated_body}', code, count=1)
        return updated_code
    else:
        return code

# 示例使用
code = """
def my_function():
    old_value = 10
    print(old_value)
"""

updated_code = replace_in_function(code, 'my_function', 'old_value', 'new_value')
print(updated_code)