用Python写一个上传下载接口
时间: 2023-03-19 11:21:38 浏览: 211
python实现上传下载文件功能
好的,以下是一个用 Python 实现的上传下载接口的示例代码:
上传文件:
```python
from flask import Flask, request, jsonify
import os
app = Flask(__name__)
@app.route('/upload', methods=['POST'])
def upload():
if 'file' not in request.files:
return jsonify({'error': 'no file part in request'})
file = request.files['file']
if file.filename == '':
return jsonify({'error': 'no selected file'})
filename = secure_filename(file.filename)
file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))
return jsonify({'success': 'file uploaded successfully'})
if __name__ == '__main__':
app.run()
```
在这个示例代码中,我们使用 Flask 框架创建了一个名为 `upload` 的路由,当请求方法为 POST 时,会执行 `upload` 函数。在这个函数中,我们首先检查请求中是否包含文件,如果没有文件,则返回错误信息。如果有文件,我们使用 Flask 提供的 `secure_filename` 函数生成一个安全的文件名,然后将文件保存到指定的上传文件夹中。最后,返回一个包含成功信息的 JSON 响应。
下载文件:
```python
from flask import Flask, send_file
app = Flask(__name__)
@app.route('/download/<filename>')
def download(filename):
path = os.path.join(app.config['UPLOAD_FOLDER'], filename)
if not os.path.isfile(path):
return jsonify({'error': 'file not found'})
return send_file(path, as_attachment=True)
if __name__ == '__main__':
app.run()
```
在这个示例代码中,我们创建了一个名为 `download` 的路由,当请求方法为 GET 时,会执行 `download` 函数。在这个函数中,我们首先检查指定的文件是否存在,如果不存在,则返回错误信息。如果文件存在,我们使用 Flask 提供的 `send_file` 函数将文件作为附件下载到客户端。
阅读全文