处理文件上传和下载在Web应用中是一个常见的需求。在Python中，我们通常使用Flask或Django这样的Web框架来构建这样的功能。下面，我将使用Flask框架来演示如何实现文件的上传和下载。
1. 安装Flask
首先，确保你已经安装了Flask。如果未安装，可以通过pip安装：
bash复制代码 pip install Flask
2. 文件上传
为了处理文件上传，我们需要一个HTML表单，该表单包含enctype="multipart/form-data"属性，以便可以发送文件数据。同时，在Flask应用中，我们将使用request.files来访问上传的文件。
HTML (upload.html):
html复制代码 <!DOCTYPE html> <html lang="en"> <head> <meta charset="UTF-8"> <title>File Upload</title> </head> <body> <h1>Upload File</h1> <form action="/upload" method="post" enctype="multipart/form-data"> <input type="file" name="file"> <input type="submit" value="Upload"> </form> </body> </html>
在Flask应用中，你需要定义一个路由来处理POST请求，并在视图函数中处理文件上传。
- 验证文件存在：检查
request.files中是否包含文件。 - 验证文件名：确保文件名不为空，并且可能需要进行进一步的验证（如文件类型、大小等）。
- 保存文件：使用
file.save(path)将文件保存到服务器上。 - 返回响应：通常，上传成功后重定向到另一个页面或显示上传成功的消息。
Flask App (app.py):
python复制代码 from flask import Flask, request, render_template, redirect, url_for, flash import os app = Flask(__name__) app.secret_key = 'supersecretkey' # 用于闪现消息 UPLOAD_FOLDER = 'uploads/' if not os.path.exists(UPLOAD_FOLDER): os.makedirs(UPLOAD_FOLDER) app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER @app.route('/') def index(): return render_template('upload.html') @app.route('/upload', methods=['POST']) def upload_file(): if 'file' not in request.files: flash('No file part') return redirect(request.url) file = request.files['file'] if file.filename == '': flash('No selected file') return redirect(request.url) if file: filename = file.filename filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename) file.save(filepath) return redirect(url_for('uploaded_file', filename=filename)) @app.route('/uploads/<filename>') def uploaded_file(filename): return redirect(url_for('static', filename='uploads/' + filename)) if __name__ == '__main__': app.run(debug=True)
注意：这里的uploaded_file函数实际上是一个重定向到静态文件的尝试，但在实际部署中，你可能需要更精细地处理文件访问权限和URL。
3. 文件下载
定义一个路由来处理文件下载请求，并在视图函数中使用send_from_directory发送文件。
- 验证文件存在：在发送文件之前，检查文件是否存在于指定的上传目录中。
- 设置HTTP头部：
send_from_directory函数允许你设置as_attachment=True来触发浏览器下载文件，而不是尝试在浏览器中打开它。
Flask App (app.py) 更新下载部分:
python复制代码 from flask import send_from_directory @app.route('/download/<filename>') def download_file(filename): return send_from_directory(directory=app.config['UPLOAD_FOLDER'], filename=filename, as_attachment=True)
这个download_file函数使用send_from_directory来发送文件作为附件，即触发浏览器下载文件。
4. 注意事项
- 验证文件类型：确保上传的文件类型是你所期望的，以避免上传恶意文件（如可执行文件、脚本等）。
- 限制文件大小：设置最大上传文件大小以防止过大的文件耗尽服务器资源。
- 使用安全的文件名：在保存文件之前，使用
werkzeug.utils.secure_filename或其他类似函数来清理文件名，以防止路径遍历等安全漏洞。 - 使用高效的文件存储：对于大量文件或大型文件，考虑使用数据库来存储文件元数据，并将文件本身存储在更高效的文件系统中（如NFS、分布式文件系统等）。
- 缓存和压缩：对于静态文件（包括上传的文件），可以使用Web服务器的缓存和压缩功能来减少带宽使用和提高加载速度。
- 异步处理：对于大型文件或需要复杂处理的上传请求，考虑使用异步处理来避免阻塞主线程。
通过以上细节和注意事项，你可以构建一个既安全又高效的Flask应用来处理文件上传和下载。
