本文介绍如何让 argparse 的可选参数（如 -m）在命令行未提供时，自动从对应 JSON 配置文件中读取默认值，并结合多值参数 -x 实现按输入 shop 名称动态定位配置文件，同时确保参数校验与错误处理的健壮性。
本文介绍如何让 argparse 的可选参数（如 `-m`）在命令行未提供时，自动从对应 json 配置文件中读取默认值，并结合多值参数 `-x` 实现按输入 shop 名称动态定位配置文件，同时确保参数校验与错误处理的健壮性。
在实际运维或数据处理脚本中，常需兼顾命令行灵活性与配置文件的可维护性。例如，当用户通过 -x abc_m 指定目标 shop 时，系统应自动查找 /opt/abc_m.json 并从中提取 memory 等默认配置；若用户显式传入 -m 16000，则优先使用命令行值。argparse 本身不支持“延迟解析的动态默认值”，但可通过两阶段处理优雅实现：先解析基础参数，再按需补全缺失值并验证。
以下为完整、生产就绪的实现方案：
import argparse
import json
import os
import sys
def load_memory_from_shop(shop_name: str) -> int:
"""根据 shop 名称加载 /opt/{shop}.json 中的 memory 字段（单位 MB），失败时退出"""
config_path = f"/opt/{shop_name}.json"
if not os.path.exists(config_path):
print(f"Error: Configuration file not found: {config_path}")
sys.exit(1)
try:
with open(config_path, "r", encoding="utf-8") as f:
data = json.load(f)
# 注意：示例 JSON 中 "memory": "'12000" 含多余单引号，需兼容处理
raw_mem = data.get("memory", "")
if isinstance(raw_mem, str):
raw_mem = raw_mem.strip("'\"") # 去除可能的引号包裹
return int(raw_mem)
except (json.JSONDecodeError, KeyError, ValueError, TypeError) as e:
print(f"Error parsing memory from {config_path}: {e}")
sys.exit(1)
def main():
parser = argparse.ArgumentParser(description="Process shops with configurable memory")
parser.add_argument('-x', dest='shops', metavar='SHOP', type=str, nargs='+',
required=True, # 强制 -x 必填
help='Shop name(s) (e.g., "abc_m") — each triggers loading of /opt/{SHOP}.json')
parser.add_argument('-s', '--start', dest='start', metavar='START', type=str,
required=True, # 强制 -s 必填
help='Start date as YYYYMMDD (integer string)')
parser.add_argument('-e', '--end', dest='end', metavar='END', type=str,
required=True, # 强制 -e 必填
help='End date as YYYYMMDD (integer string)')
parser.add_argument('-m', '--memory', dest='memory', metavar='MEMORY', type=int,
required=False, default=None,
help='Memory allocation in MB (defaults to value from /opt/<SHOP>.json)')
args = parser.parse_args()
# ✅ 校验 -x 输入：必须非空，且每个 shop 对应配置文件存在
if not args.shops:
print("Error: At least one -x shop must be specified.")
sys.exit(1)
# ✅ 若 -m 未提供，则从第一个 -x 对应的 JSON 加载（或按需扩展为 per-shop 逻辑）
if args.memory is None:
if args.shops:
args.memory = load_memory_from_shop(args.shops[0])
print(f"Using memory={args.memory} MB from /opt/{args.shops[0]}.json")
else:
print("Error: Cannot infer memory source — no -x shop provided.")
sys.exit(1)
# ✅ 可选：对其他字段做类型/范围校验（如日期格式、内存合理性）
try:
int(args.start)
int(args.end)
if len(args.start) != 8 or len(args.end) != 8:
raise ValueError("Date must be exactly 8 digits (YYYYMMDD)")
except ValueError as e:
print(f"Invalid date format: {e}")
sys.exit(1)
# ✅ 此处开始业务逻辑...
print(f"Processing shops: {args.shops}")
print(f"Date range: {args.start} to {args.end}")
print(f"Allocated memory: {args.memory} MB")
if __name__ == "__main__":
main()关键设计说明：
- 强制参数校验：将 -x、-s、-e 显式设为 required=True，避免运行时空值风险；
- 动态默认值时机：parse_args() 后立即检查 args.memory is None，再调用 load_memory_from_shop()，确保路径和内容校验可控；
- 健壮的 JSON 解析：自动剥离字符串型 "memory" 值两侧可能存在的引号（适配原始数据中的 '12000）；
- 清晰的错误反馈：文件缺失、JSON 格式错误、字段不存在等均输出明确提示并 sys.exit(1)；
- 可扩展性：当前以首个 -x 值为准加载 memory，如需 per-shop 内存配置，可将 args.memory 改为列表，并遍历 args.shops 分别加载。
⚠️ 注意事项：
立即学习“Python免费学习笔记（深入）”；
- 避免在 default= 参数中直接写文件读取逻辑（如 default=json.load(open(...))），这会导致模块导入时即执行，且无法捕获运行时错误；
- 生产环境建议使用 pathlib.Path 替代字符串拼接路径，并添加更细粒度的权限检查；
- 若需支持多个 -x 对应不同 memory，应在后续逻辑中构建 {shop: memory} 映射，而非全局单一 memory 值。
