如何在遍历json树的根节点到目的节点时插入key？

def insert_key(json_obj, target_key, insert_key):
    if isinstance(json_obj, dict):
        for key, value in json_obj.items():
            if key == target_key:
                json_obj[insert_key] = None  # 在目标节点的父节点中插入新的key
            insert_key(value, target_key, insert_key)  # 递归遍历子节点
    elif isinstance(json_obj, list):
        for item in json_obj:
            insert_key(item, target_key, insert_key)  # 递归遍历列表中的元素

# 示例用法
json_data = {
    "name": "John",
    "age": 30,
    "address": {
        "street": "123 Main St",
        "city": "New York"
    }
}

insert_key(json_data, "city", "country")
print(json_data)

{
    "name": "John",
    "age": 30,
    "address": {
        "street": "123 Main St",
        "city": "New York",
        "country": null
    }
}