From d0471bd9cda8db8c889bd6ac7380f0a21f2b2313 Mon Sep 17 00:00:00 2001
From: LufsX <lufsx@outlook.com>
Date: Mon, 23 Sep 2024 04:32:57 +0800
Subject: [PATCH] feat: add upload and download function

---
 app.py | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)
 create mode 100644 app.py

diff --git a/app.py b/app.py
new file mode 100644
index 0000000..848a267
--- /dev/null
+++ b/app.py
@@ -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)