如何动态设置复杂json内容的值，接收key (或key path)和value作为参数

import json

def set_json_value(json_obj, key_path, value):
    if not key_path:
        return value

    keys = key_path.split('.')
    current_key = keys.pop(0)

    if isinstance(json_obj, list):
        current_key = int(current_key)

    if len(keys) > 0:
        json_obj[current_key] = set_json_value(json_obj[current_key], '.'.join(keys), value)
    else:
        json_obj[current_key] = value

    return json_obj

# 示例 JSON 对象
json_str = '''
{
    "name": "John",
    "age": 30,
    "skills": {
        "programming": ["Python", "JavaScript"],
        "design": ["Photoshop", "Illustrator"]
    },
    "address": {
        "street": "123 ABC Street",
        "city": "XYZ City"
    }
}
'''

# 将 JSON 字符串解析为对象
json_obj = json.loads(json_str)

# 设置 JSON 对象的值
json_obj = set_json_value(json_obj, 'name', 'Alice')
json_obj = set_json_value(json_obj, 'skills.programming.1', 'Java')
json_obj = set_json_value(json_obj, 'address.city', 'ABC City')

# 将 JSON 对象转换为字符串
updated_json_str = json.dumps(json_obj, indent=4)
print(updated_json_str)