如何在Jupyter Notebook中模拟命令行参数以测试Python脚本

# 单元格 1: 准备测试数据
# 创建临时的csv文件和输出目录，用于演示
!mkdir -p temp_data
!echo "col1,col2\n1,a\n2,b" > temp_data/file1.csv
!echo "Campaign Key\nCK1\nCK2" > temp_data/file2.csv
print("测试文件已创建：temp_data/file1.csv, temp_data/file2.csv")
print("测试输出目录已准备：temp_data/output_dir")

# 单元格 2: 使用 %%python 魔术命令运行带参数的脚本
# 请确保此行是单元格的第一行，并且参数值根据您的实际情况修改
%%python - --SP_File temp_data/file1.csv --File2 temp_data/file2.csv temp_data/output_dir

import argparse
from pathlib import Path
import pandas as pd
import sys

# 1. 定义 ArgumentParser
parser = argparse.ArgumentParser(description='Jupyter Notebook中的参数传递演示')
parser.add_argument("--SP_File",
                    help="SP文件路径", type=Path)
parser.add_argument("--File2",
                    help="Datorama文件路径", type=Path)
parser.add_argument("Missing_Campaign_File",
                    help="输出文件保存目录", type=Path)

# 2. 解析参数
# 当使用 %%python 运行此单元格时，sys.argv 会被自动修改以包含传递的参数
args = parser.parse_args()

print(f"--- 接收到的参数 ---")
print(f"SP文件路径: {args.SP_File}")
print(f"Datorama文件路径: {args.File2}")
print(f"输出目录路径: {args.Missing_Campaign_File}")
print(f"--------------------")

# 3. 文件路径验证与读取
sp_file_path = args.SP_File
if not sp_file_path.exists():
    # 注意：在实际脚本中，这会是一个致命错误
    print(f"错误: 未找到SP文件: {sp_file_path}")
    sys.exit(1) # 模拟脚本退出
try:
    #