Update app.py

app.py

@@ -1,6 +1,7 @@
-from flask import Flask, render_template
+from flask import Flask, render_template, request, redirect, url_for, send_from_directory
 from flask_socketio import SocketIO, emit
 from flask_cors import CORS
+from flask_uploads import UploadSet, configure_uploads, ALL
 import pty
 import os
 import subprocess
@@ -9,6 +10,12 @@ import threading
 app = Flask(__name__)
 CORS(app)  # 启用 CORS
 app.config['SECRET_KEY'] = 'your_secret_key'
+
+# 配置文件上传
+app.config['UPLOADED_FILES_DEST'] = 'uploads'  # 上传文件的保存目录
+files = UploadSet('files', ALL)
+configure_uploads(app, files)
+
 socketio = SocketIO(app)
 
 master_fd, slave_fd = pty.openpty()
@@ -26,12 +33,36 @@ def run_shell():
 def index():
     return render_template('index.html')
 
+@app.route('/upload', methods=['POST'])
+def upload_file():
+    if 'file' in request.files:
+        filename = files.save(request.files['file'])
+        return redirect(url_for('index'))
+    return 'No file uploaded', 400
+
+@app.route('/files', defaults={'path': ''})
+@app.route('/files/<path:path>')
+def list_files(path):
+    full_path = os.path.join(app.config['UPLOADED_FILES_DEST'], path)
+    if not os.path.exists(full_path):
+        return 'Directory not found', 404
+
+    files_and_dirs = os.listdir(full_path)
+    files_and_dirs = [f for f in files_and_dirs if not f.startswith('.')]  # 排除隐藏文件
+    return render_template('files.html', files=files_and_dirs, current_path=path)
+
+@app.route('/files/<path:path>/<filename>')
+def download_file(path, filename):
+    return send_from_directory(os.path.join(app.config['UPLOADED_FILES_DEST'], path), filename)
+
 @socketio.on('input')
 def handle_input(input_data):
     os.write(master_fd, (input_data + '\n').encode('utf-8'))
 
 if __name__ == '__main__':
+    if not os.path.exists(app.config['UPLOADED_FILES_DEST']):
+        os.makedirs(app.config['UPLOADED_FILES_DEST'])
     shell_thread = threading.Thread(target=run_shell)
     shell_thread.daemon = True
     shell_thread.start()
-    socketio.run(app, host='0.0.0.0', port=7860, allow_unsafe_werkzeug=True)
+    socketio.run(app, host='0.0.0.0', port=7860, allow_unsafe_werkzeug=True)