python脚本编辑oss文件

1、安装oss2库

root@ubuntu:~# pip3 install oss2
Collecting oss2
  Downloading oss2-2.18.5.tar.gz (283 kB)
     |████████████████████████████████| 283 kB 6.9 MB/s 
Collecting aliyun-python-sdk-core>=2.13.12
  Downloading aliyun-python-sdk-core-2.15.1.tar.gz (443 kB)
     |████████████████████████████████| 443 kB 67.8 MB/s 
Collecting aliyun-python-sdk-kms>=2.4.1
  Downloading aliyun_python_sdk_kms-2.16.3-py2.py3-none-any.whl (98 kB)
     |████████████████████████████████| 98 kB 8.3 MB/s 
Collecting crcmod>=1.7
  Downloading crcmod-1.7.tar.gz (89 kB)
     |████████████████████████████████| 89 kB 14.3 MB/s 
Collecting pycryptodome>=3.4.7
  Downloading pycryptodome-3.20.0-cp35-abi3-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (2.1 MB)
     |████████████████████████████████| 2.1 MB 7.0 MB/s 
Requirement already satisfied: requests!=2.9.0 in /usr/lib/python3/dist-packages (from oss2) (2.22.0)
Requirement already satisfied: six in /usr/lib/python3/dist-packages (from oss2) (1.14.0)
Requirement already satisfied: cryptography>=2.6.0 in /usr/lib/python3/dist-packages (from aliyun-python-sdk-core>=2.13.12->oss2) (2.8)
Collecting jmespath<1.0.0,>=0.9.3
  Downloading jmespath-0.10.0-py2.py3-none-any.whl (24 kB)
Building wheels for collected packages: oss2, aliyun-python-sdk-core, crcmod
  Building wheel for oss2 (setup.py) ... done
  Created wheel for oss2: filename=oss2-2.18.5-py3-none-any.whl size=118170 sha256=cd8f24bb98e8449af56d9df9e826cce691d0527e53420f9c2f14a5b85b059d0c
  Stored in directory: /root/.cache/pip/wheels/b6/af/6a/36f940ed11f11b5e1002f94160a82c3dc35e8a357cedc02bb7
  Building wheel for aliyun-python-sdk-core (setup.py) ... done
  Created wheel for aliyun-python-sdk-core: filename=aliyun_python_sdk_core-2.15.1-py3-none-any.whl size=535318 sha256=83c40a77e365ad83d4b74b43e757c6cea78628e8054706e48ae424ed8ddf7b69
  Stored in directory: /root/.cache/pip/wheels/64/af/7e/b3ec025852e53d69463404a2b281d4587d45ee710cd45d0f38
  Building wheel for crcmod (setup.py) ... done
  Created wheel for crcmod: filename=crcmod-1.7-cp38-cp38-linux_x86_64.whl size=35994 sha256=9fc029c468fe548c4862be3ab4783e1d17b2ba52b72aef537d520b42f1c0a0ad
  Stored in directory: /root/.cache/pip/wheels/ca/5a/02/f3acf982a026f3319fb3e798a8dca2d48fafee7761788562e9
Successfully built oss2 aliyun-python-sdk-core crcmod
Installing collected packages: jmespath, aliyun-python-sdk-core, aliyun-python-sdk-kms, crcmod, pycryptodome, oss2
Successfully installed aliyun-python-sdk-core-2.15.1 aliyun-python-sdk-kms-2.16.3 crcmod-1.7 jmespath-0.10.0 oss2-2.18.5 pycryptodome-3.20.0

2、编写脚本
1）更改后将文件更美观的输出

脚本json格式化输出

import json
import oss2

def update_oss_json_value(access_key_id, access_key_secret, endpoint, bucket_name, file_path, key_value_map):
    # 创建 OSS 客户端
    auth = oss2.Auth(access_key_id, access_key_secret)
    bucket = oss2.Bucket(auth, endpoint, bucket_name)

    # 读取文件内容
    content = bucket.get_object(file_path).read()

    # 解析 JSON
    data = json.loads(content)

    # 修改指定 key 的值
    for key, value in key_value_map.items():
        if key in data:
            data[key] = value

    # 将修改后的 JSON 格式化为美观的字符串
    new_content = json.dumps(data, indent=4, ensure_ascii=False)

    # 上传修改后的文件内容
    bucket.put_object(object_key, new_content.encode('utf-8'))

    print("File content has been updated.")

# 要修改的 OSS 文件路径
file_path = 'test/domain_bak.json'

# 指定正确的 endpoint
endpoint = 'http://oss-cn-hongkong.aliyuncs.com'

# 要修改的 key-value 映射
key_value_map = {
    'superSign1': 'https://uykkr.abc.com/c/gulorex6',
    'superSign2': 'https://dspcs.abc.com/c/xqjqujc2'
}

# 调用函数更新 OSS 文件内容
update_oss_json_value('XXXXXXXXXXX', 'XXXXXXXXXXXXXXXXXXXXXX', endpoint, 'test', file_path, key_value_map)

执行后会将json文件格式化输出，如有不是很规范的都会更加美观的显示
2）保持每个键值对都占据一行，并在开头空四格

import json
import oss2

def update_oss_json_value(access_key_id, access_key_secret, endpoint, bucket_name, file_path, key_value_map):
    # 创建 OSS 客户端
    auth = oss2.Auth(access_key_id, access_key_secret)
    bucket = oss2.Bucket(auth, endpoint, bucket_name)

    # 读取文件内容
    content = bucket.get_object(file_path).read()

    # 解析 JSON
    data = json.loads(content)

    # 修改指定 key 的值
    for key, value in key_value_map.items():
        if key in data:
            data[key] = value

    # 将修改后的 JSON 转换为字符串，保持每个键值对都占据一行，并在开头空四格
    new_content = ',\n'.join([f'    "{k}": "{v}"' for k, v in data.items()])

    # 添加大括号，保持 JSON 格式
    new_content = '{\n' + new_content + '\n}'

    # 上传修改后的文件内容
    bucket.put_object(object_key, new_content.encode('utf-8'))

    print("File content has been updated.")

# 要修改的 OSS 文件路径
file_path = 'test/domain_bak.json'

# 指定正确的 endpoint
endpoint = 'http://oss-cn-hongkong.aliyuncs.com'

# 要修改的 key-value 映射
key_value_map = {
    'superSign1': 'https://uykkr.abc.com/c/gulorex6',
    'superSign2': 'https://dspcs.abc.com/c/xqjqujc2'
}

# 调用函数更新 OSS 文件内容
update_oss_json_value('XXXXXXXXXXX', 'XXXXXXXXXXXXXXXXXXXXXX', endpoint, 'test', file_path, key_value_map)

3、通过位置参数进行新增键值对

import json
import oss2
import argparse

def update_oss_json_value(access_key_id, access_key_secret, endpoint, bucket_name, file_path, *superSigns):
    # 创建 OSS 客户端
    auth = oss2.Auth(access_key_id, access_key_secret)
    bucket = oss2.Bucket(auth, endpoint, bucket_name)

    # 读取文件内容
    content = bucket.get_object(file_path).read()

    # 解析 JSON
    data = json.loads(content)

    # 生成 key-value 映射
    key_value_map = {}
    for i, superSign in enumerate(superSigns, start=1):
        key = f'superSign{i}'
        key_value_map[key] = superSign

    # 修改指定 key 的值
    for key, value in key_value_map.items():
        #if key in data:
        data[key] = value
        print(key,value)

    # 将修改后的 JSON 转换为字符串，保持每个键值对都占据一行，并在开头空四格
    new_content = ',\n'.join([f'    "{k}": "{v}"' for k, v in data.items()])

    # 添加大括号，保持 JSON 格式
    new_content = '{\n' + new_content + '\n}'

    # 打印更新后的键值对
    for key, value in data.items():
        print(f"{key}: {value}")
             
    # 上传修改后的文件内容
    bucket.put_object(file_path, new_content.encode('utf-8'))

    print("File content has been updated.")

if __name__ == "__main__":
    # 创建参数解析器
    parser = argparse.ArgumentParser(description="Update OSS JSON file")

    # 添加命令行参数
    #parser.add_argument("access_key_id", type=str, help="Access Key ID")
    #parser.add_argument("access_key_secret", type=str, help="Access Key Secret")
    #parser.add_argument("endpoint", type=str, help="OSS Endpoint")
    parser.add_argument("bucket_name", type=str, help="Bucket Name")
    parser.add_argument("file_path", type=str, help="Object Key")
    parser.add_argument("superSigns", nargs='+', type=str, help="Values for superSigns")

    # 解析命令行参数
    args = parser.parse_args()

    # 调用函数更新 OSS 文件内容
    update_oss_json_value('XXXXXXXXXXX', 'XXXXXXXXXXXXXXXXXXXXXX', 'http://oss-cn-hongkong.aliyuncs.com', args.bucket_name, args.file_path, *args.superSigns)

执行

root@ubuntu:~# python3 kong.py 'test-prod' 'test/domain_bak.json' 'https://uykkr.abc.com/api/c/gulorex6' 'https://dspcs.abc.com/api/c/xqjqujc2'
superSign1 https://uykkr.abc.com/api/c/gulorex6
superSign2 https://dspcs.abc.com/api/c/xqjqujc2
。。。。。文本json。。。。。
File content has been updated. 
root@ubuntu:~# python3 kong.py 'test-prod' 'test/domain_bak.json' 'https://uykkr.abc.com/api/c/gulorex6' 'https://dspcs.abc.com/api/c/xqjqujc2' 'https://dspcs.abc.com/api/c/acrik5'
superSign1 https://uykkr.abc.com/api/c/gulorex6
superSign2 https://dspcs.abc.com/api/c/xqjqujc2
superSign3 https://dspcs.abc.com/api/c/acrik5
。。。。。文本json。。。。。
File content has been updated.
root@ubuntu:~# python3 kong.py 'test-prod' 'test/domain_bak.json' 'https://uykkr.abc.com/api/c/gulorex6' 'https://dspcs.abc.com/api/c/xqjqujc2' 'https://dspcs.abc.com/api/c/acrik5' 'https://dspcs.abc.com/api/c/acrik5'
superSign1 https://uykkr.abc.com/api/c/gulorex6
superSign2 https://dspcs.abc.com/api/c/xqjqujc2
superSign3 https://dspcs.abc.com/api/c/acrik5
superSign4 https://dspcs.abc.com/api/c/acrik5
。。。。。文本json。。。。。
File content has been updated.