本文详解如何使用 flask 接收用户输入、调用 matplotlib 的 pdfpages 生成多页 pdf，并通过 send_from_directory 安全响应下载请求，避免线程阻塞与路径错误。
本文详解如何使用 flask 接收用户输入、调用 matplotlib 的 pdfpages 生成多页 pdf，并通过 send_from_directory 安全响应下载请求，避免线程阻塞与路径错误。
在 Flask 中动态生成并提供 PDF 文件（如由 matplotlib.backends.backend_pdf.PdfPages 创建的报告），关键在于分离用户交互、文件生成与静态资源分发。原始代码中直接使用 input() 会导致主线程阻塞（尤其在 macOS 上触发 NSWindow drag regions 异常），且硬编码路径、未处理 POST 请求、依赖本地终端输入，完全违背 Web 应用无状态、异步响应的设计原则。
以下是推荐的生产就绪型实现方案：
✅ 正确架构：表单提交 → 后端生成 → 临时存储 → 安全下载
from flask import Flask, request, render_template_string, send_from_directory
from matplotlib.figure import Figure
from matplotlib.backends.backend_pdf import PdfPages
import os
import tempfile
import datetime
app = Flask(__name__)
# 配置 PDF 存储目录（使用 instance_path 更符合 Flask 最佳实践）
PLOTS_DIR = os.path.join(app.instance_path, 'plots')
os.makedirs(PLOTS_DIR, exist_ok=True)
def plotting_function(output_path: str, param1: str, param2: str):
"""模拟你的绘图逻辑：生成多页 PDF 并写入 metadata"""
with PdfPages(output_path) as pdf:
# 示例：生成 3 个子图页
for i in range(1, 4):
fig = Figure(figsize=(8, 6))
ax = fig.subplots()
ax.plot([0, 1, 2, 3], [i, i+1, i-1, i*2], label=f'Plot {i}')
ax.set_title(f'Parameter: {param1}, Variant: {param2} — Page {i}')
ax.legend()
pdf.savefig(fig)
fig.clear() # 释放内存
# 设置 PDF 元数据（可选但推荐）
info = pdf.infodict()
info['Title'] = f'Report for {param1}'
info['Author'] = 'Flask-PDF Generator'
info['Subject'] = 'Automated Plot Report'
info['Keywords'] = f'{param1} {param2} matplotlib pdf'
info['CreationDate'] = datetime.datetime.now()
info['ModDate'] = datetime.datetime.now()
@app.route('/', methods=['GET', 'POST'])
def index():
if request.method == 'POST':
# ✅ 安全获取表单字段（非 input()！）
param1 = request.form.get('param1', '').strip()
param2 = request.form.get('param2', '').strip()
if not (param1 and param2):
return "⚠️ 参数不能为空，请返回重试", 400
# ✅ 使用临时目录确保线程安全 & 自动清理
with tempfile.TemporaryDirectory(dir=PLOTS_DIR) as tmpdir:
filename = f"plots_for_{param1.lower()}.pdf"
filepath = os.path.join(tmpdir, filename)
try:
plotting_function(filepath, param1, param2)
# ✅ 直接发送临时文件（Flask 自动处理流式响应）
return send_from_directory(
tmpdir,
filename,
as_attachment=True,
download_name=filename # Flask 2.1+ 推荐用 download_name 替代 deprecated as_attachment=True 的歧义
)
except Exception as e:
app.logger.error(f"PDF generation failed: {e}")
return f"❌ 生成 PDF 失败：{e}", 500
# GET 请求：渲染 HTML 表单
return render_template_string('''
<!DOCTYPE html>
<html>
<head><title>PDF 报告生成器</title></head>
<body>
<h2>配置参数并生成 PDF 报告</h2>
<form method="post">
<div style="margin: 1em 0;">
<label for="param1">参数 1（例如：实验组名）：</label>
<input type="text" id="param1" name="param1" required />
</div>
<div style="margin: 1em 0;">
<label for="param2">参数 2（例如：时间范围）：</label>
<input type="text" id="param2" name="param2" required />
</div>
<button type="submit" style="padding: 0.5em 1em;">生成 PDF</button>
</form>
</body>
</html>
''')
# 可选：添加健康检查或静态文件服务路由
@app.route('/health')
def health():
return {"status": "ok", "plots_dir": PLOTS_DIR}? 关键要点说明
- 禁止 input()：Web 应用无法访问终端 stdin，input() 会挂起主线程并引发平台级异常（如 macOS 的 NSInternalInconsistencyException）。必须改用 HTML 表单 + HTTP POST。
- 路径安全：send_from_directory 是唯一安全的文件响应方式，它自动校验路径防止目录遍历攻击（如 ../../../etc/passwd）。绝不可用 send_file(open(...)) 或拼接字符串路径。
- 临时文件管理：tempfile.TemporaryDirectory() 确保每次请求独立隔离，退出 with 块后自动删除，避免磁盘泄漏和并发冲突。
- 错误处理：捕获 plotting_function 异常并返回 HTTP 错误码，便于前端调试；日志记录便于运维排查。
- 元数据增强：为 PDF 添加 Title/Author 等信息，提升专业性与可追溯性。
? 运行前准备
- 创建 Flask 实例目录结构：
mkdir -p instance/plots
- 安装依赖：
pip install flask matplotlib
- 启动应用（开发环境）：
export FLASK_APP=app.py export FLASK_ENV=development flask run --host=127.0.0.1 --port=5000
访问 http://127.0.0.1:5000 即可测试。
此方案兼顾安全性、可维护性与用户体验，是 Flask 动态 PDF 生成的标准实践。如需支持大文件流式传输或异步生成（Celery），可在本基础之上扩展。
