使用argparse从Azure Blob存储读取TOML文件时遇属性错误与读取异常

问题梳理 #

• 执行命令python src/main --file=something.toml传递Azure Blob上的TOML文件路径
• 用argparse.FileType('r')解析参数后，readlines()仅读取到[mainparams]一行内容
• 调用tomli.loads(parser.file)加载时，触发错误：AttributeError: '_io.TextIOWrapper' object has no attribute 'replace'
• 需要获取TOML文件中[mainparams]节点下的country和activity值

错误原因 #

1. argparse.FileType不适配Azure Blob场景：该类型默认尝试读取本地文件，但你的TOML文件存储在Azure Blob而非本地，导致读取不完整或失败。
2. tomli.loads参数类型不匹配：loads()要求传入字符串格式的TOML内容，你传入的是文件对象（_io.TextIOWrapper），而文件对象没有tomli内部处理所需的replace方法。

解决步骤 #

1. 修改argparse参数配置 #

放弃使用FileType，直接让参数接收Azure Blob上的文件路径（格式为<容器名>/<文件名>）。

2. 通过Azure SDK读取Blob内容 #

使用Azure存储SDK下载Blob中的TOML内容，转换为字符串后再进行解析。

3. 解析TOML并提取目标值 #

用tomli.loads()解析字符串内容，通过字典索引直接获取所需参数。

完整示例代码 #

import argparse
import tomli
from azure.storage.blob import BlobServiceClient

def parse_args():
    parser = argparse.ArgumentParser()
    # 接收Azure Blob文件路径，格式示例：'my-container/something.toml'
    parser.add_argument('--file', type=str, required=True, help='Azure Blob中TOML文件的路径（容器名/文件名）')
    return parser.parse_args()

def load_toml_from_blob(blob_path):
    # 替换为你的Azure存储连接字符串
    AZURE_STORAGE_CONN_STR = "你的Azure存储连接字符串"
    
    # 拆分容器名与Blob文件名
    container_name, blob_name = blob_path.split('/', 1)
    
    # 初始化Blob客户端并下载内容
    blob_service_client = BlobServiceClient.from_connection_string(AZURE_STORAGE_CONN_STR)
    blob_client = blob_service_client.get_blob_client(container=container_name, blob=blob_name)
    toml_content = blob_client.download_blob().readall().decode('utf-8')
    
    # 解析TOML内容
    return tomli.loads(toml_content)

if __name__ == "__main__":
    args = parse_args()
    toml_config = load_toml_from_blob(args.file)
    
    # 提取目标参数
    country = toml_config['mainparams']['country']
    activity = toml_config['mainparams']['activity']
    
    print(f"获取到的参数：country={country}, activity={activity}")

本地测试简化方案 #

如果是本地TOML文件测试，可直接使用文件对象解析：

def load_toml_local(file_path):
    with open(file_path, 'r', encoding='utf-8') as f:
        # 本地文件用tomli.load()直接加载文件对象，无需转换字符串
        return tomli.load(f)

内容的提问来源于stack exchange，提问作者alim1990