feat: add upload and download function

This commit is contained in:
2024-09-23 04:32:57 +08:00
parent 68f83b611c
commit d0471bd9cd
+30
View File
@@ -0,0 +1,30 @@
from flask import Flask, request, send_from_directory, abort
import os
app = Flask(__name__)
UPLOAD_FOLDER = './uploads'
os.makedirs(UPLOAD_FOLDER, exist_ok=True)
@app.route('/upload', methods=['POST'])
def upload_file():
if 'file' not in request.files:
return 'No file part'
file = request.files['file']
if file.filename == '':
return 'No selected file'
# 保存文件到指定目录
file.save(os.path.join(UPLOAD_FOLDER, file.filename))
return 'File uploaded successfully'
@app.route('/download/<filename>', methods=['GET'])
def download_file(filename):
try:
return send_from_directory(UPLOAD_FOLDER, filename, as_attachment=True)
except FileNotFoundError:
abort(404)
if __name__ == '__main__':
app.run(host='0.0.0.0', port=880)