如何将argparse命令行参数传入独立配置文件？

问题根源 #

1. 变量未定义错误：sales_report_config.py中的sql_query直接引用start_date/end_date，但这两个变量仅存在于sales_report.py的命令行解析作用域中，配置文件无法直接访问。
2. 循环导入：若在配置文件中导入sales_report.py的参数，会触发循环依赖——sales_report.py需要导入配置文件的SQL，配置文件又需要导入主文件的参数，Python无法解析这种循环引用。

可行解决方案 #

方案1：将SQL定义为带参数的生成函数（简单直接） #

把配置文件中的固定SQL字符串改成一个接收日期参数的函数，主文件调用时传入命令行参数即可，完全避免循环依赖。

sales_report_config.py：

def generate_sales_query(start_date, end_date):
    # 注意：若日期格式需要转义，需额外处理，或优先用方案2的参数化查询
    return f"""
    SELECT * FROM sales
    WHERE sale_date BETWEEN '{start_date}' AND '{end_date}'
    ORDER BY sale_date DESC
    """

sales_report.py：

import argparse
import psycopg2
import pandas as pd
from sales_report_config import generate_sales_query

# 解析命令行参数
parser = argparse.ArgumentParser(description='Generate sales report')
parser.add_argument('--start_date', required=True, help='Start date (YYYY-MM-DD)')
parser.add_argument('--end_date', required=True, help='End date (YYYY-MM-DD)')
args = parser.parse_args()

# 生成带参数的SQL查询语句
sql_query = generate_sales_query(args.start_date, args.end_date)

# 连接数据库并执行查询
conn = psycopg2.connect("dbname=your_db user=your_user password=your_pass host=your_host")
sales_df = pd.read_sql(sql_query, conn)
conn.close()

# 后续处理逻辑
print(f"生成了{len(sales_df)}条销售记录的报表")

方案2：使用参数化查询（推荐，防SQL注入） #

直接拼接SQL字符串存在SQL注入风险，更安全的做法是用数据库驱动支持的参数化查询，配置文件仅存放SQL模板，主文件传递参数。

sales_report_config.py：

# 用%s作为psycopg2的参数占位符（不同数据库占位符可能不同，比如SQLite用?）
SALES_QUERY_TEMPLATE = """
SELECT * FROM sales
WHERE sale_date BETWEEN %s AND %s
ORDER BY sale_date DESC
"""

sales_report.py：

import argparse
import psycopg2
import pandas as pd
from sales_report_config import SALES_QUERY_TEMPLATE

parser = argparse.ArgumentParser(description='Generate sales report')
parser.add_argument('--start_date', required=True, help='Start date (YYYY-MM-DD)')
parser.add_argument('--end_date', required=True, help='End date (YYYY-MM-DD)')
args = parser.parse_args()

conn = psycopg2.connect("dbname=your_db user=your_user password=your_pass host=your_host")
# 用pandas的params参数传递，自动处理参数转义和注入防护
sales_df = pd.read_sql(SALES_QUERY_TEMPLATE, conn, params=(args.start_date, args.end_date))
conn.close()

print(f"生成了{len(sales_df)}条销售记录的报表")

额外提示 #

• 若需要更复杂的SQL模板（比如多条件可选），可以结合string.Template或第三方库（如Jinja2）来管理，但对于简单场景，上述两种方案足够。
• 永远优先使用参数化查询，避免直接拼接用户输入（包括命令行参数）到SQL语句中，防止恶意注入攻击。

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