argparse在Jupyter Notebook中传递文件路径参数的教程

import argparse
from pathlib import Path
import pandas as pd
import numpy as np
import sys
import os
from datetime import datetime, timedelta, date, time

parser = argparse.ArgumentParser(description='Tractive_missing_campaign')
parser.add_argument("--SP_File",
                    help="Add path of your SP file", type=Path)
parser.add_argument("--File2",
                    help="Add path of your Datorama file", type=Path)
parser.add_argument("Missing_Campaign_File", # 这是一个位置参数
                    help="Add path where you want to save your file", type=Path)

# 在Jupyter Notebook中直接运行 parser.parse_args() 将会出错或无法获取参数
# args = parser.parse_args()

# 后续代码将使用 args.SP_File, args.File2, args.Missing_Campaign_File
# 例如：
# Sp_Report = args.SP_File
# if not Sp_Report.exists():
#     raise ValueError(f"I did not find the SP file at {Sp_Report}")
# ...

python your_script.py --SP_File 'path/to/sp_report.csv' --File2 'path/to/datorama.csv' 'path/to/output_directory'

import argparse
from pathlib import Path
import pandas as pd
import numpy as np
import sys
import os
from datetime import datetime, timedelta, date, time

# --- argparse 定义部分 (与上文相同) ---
parser = argparse.ArgumentParser(description='Tractive_missing_campaign')
parser.add_argument("--SP_File",
                    help="Add path of your SP file", type=Path)
parser.add_argument("--File2",
                    help="Add path of your Datorama file", type=Path)
parser.add_argument("Missing_Campaign_File",
                    help="Add path where you want to save your file", type=Path)
# --- argparse 定义部分结束 ---

# 1. 保存原始 sys.argv
original_sys_argv = sys.argv

try:
    # 2. 构建模拟参数列表
    # 注意：第一个元素通常是脚本名，后续是实际的命令行参数
    simulated_args = [
        'notebook_test.py',
        '--SP_File', 'data/sp_report_sample.csv',  # 模拟SP文件路径
        '--File2', 'data/datorama_sample.csv',    # 模拟Datorama文件路径
        'output/missing_campaign_results'         # 模拟输出目录 (位置参数)
    ]

    # 3. 替换 sys.argv
    sys.argv = simulated_args

    # 4. 调用 parser.parse_args()
    args = parser.parse_args()

    # --- 使用解析后的参数执行业务逻辑 ---
    print(f"解析到的SP文件路径: {args.SP_File}")
    print(f"解析到的Datorama文件路径: {args.File2}")
    print(f"解析到的输出文件目录: {args.Missing_Campaign_File}")

    # 示例：文件读取逻辑 (已纠正原代码中的 os.chdir 错误)
    sp_file_path = args.SP_File
    if not sp_file_path.exists():
        raise ValueError(f"错误：未找到SP文件于 {sp_file_path}")
    try:
        # 使用 pathlib.Path 对象直接打开文件
        with open(sp_file_path, "r+") as csvfile:
            sp_df = pd.read_csv(csvfile)
        print(f"SP文件 '{sp_file_path}' 读取成功，形状：{sp_df.shape}")
    except Exception as ex:
        template = "读取SP文件时发生异常: {0}. 参数:\n{1!r}"
        message = template.format(type(ex).__name__, ex.args)
        print(message)

    datorama_file_path = args.File2
    if not datorama_file_path.exists():
        raise ValueError(f"错误：未找到Datorama文件于 {datorama_file_path}")
    try:
        with open(datorama_file_path, "r+") as csvfile:
            datorama_df = pd.read_csv(csvfile, usecols = ['Campaign Key'])
        print(f"Datorama文件 '{datorama_file_path}' 读取成功，形状：{datorama_df.shape}")
    except Exception as ex:
        template = "读取Datorama文件时发生异常: {0}. 参数:\n{1!r}"
        message = template.format(type(ex).__name__, ex.args)
        print(message)

    # 模拟输出文件创建
    output_dir = args.Missing_Campaign_File
    os.makedirs(output_dir, exist_ok=True) # 确保输出目录存在
    # 假设有一些数据和逻辑来生成文件
    dummy_data = pd.DataFrame({'Campaign Key': ['A1', 'B2', 'C3'], 'Value': [1, 2, 3]})
    # 假设 slicer_column_values, start_date, end_date 已定义
    slicer_column_values = ['USD', 'EUR']
    start_date = '2023-01-01'
    end_date = '2023-01-31'

    for slicer_column_value in slicer_column_values:
        # 这里的 left_merged, Match_Values, currency, Match 变量未定义，仅为示例
        # 实际代码需要根据业务逻辑定义这些变量
        # temporary_df = left_merged[(left_merged['currency'] == slicer_column_value) & (left_merged['Match'] == Match_Values)]
        # 为了示例运行，我们使用 dummy_data
        temporary_df = dummy_data.copy() 
        PartnerWiseFileName = os.path.join(output_dir, 'Missing_camp_{}_{}_{}_data.xlsx'.format(slicer_column_value, start_date, end_date))
        print(f"正在保存文件到: {PartnerWiseFileName}")
        temporary_df.to_excel(PartnerWiseFileName, index = False)
    print("任务完成")

except SystemExit as e:
    # argparse 在解析失败时会调用 sys.exit()，导致 SystemExit 异常
    print(f"参数解析错误: {e}")
finally:
    # 5. 恢复原始 sys.argv
    sys.argv = original_sys_argv
    print("\nsys.argv 已恢复到原始状态。")