Update app.py

app.py

@@ -1,7 +1,7 @@
-from flask import Flask, render_template, request, redirect, url_for, send_from_directory
+from flask import Flask, render_template, request, redirect, url_for
 from flask_socketio import SocketIO, emit
 from flask_cors import CORS
-from flask_uploads import UploadSet, configure_uploads, ALL
+from flask_dropzone import Dropzone
 import pty
 import os
 import subprocess
@@ -12,9 +12,13 @@ CORS(app)  # 启用 CORS
 app.config['SECRET_KEY'] = 'your_secret_key'
 
 # 配置文件上传
+app.config['DROPZONE_UPLOAD_MULTIPLE'] = False
+app.config['DROPZONE_ALLOWED_FILE_TYPE'] = 'all'
+app.config['DROPZONE_MAX_FILE_SIZE'] = 3  # 最大文件大小为3MB
+app.config['DROPZONE_REDIRECT_VIEW'] = 'list_files'
 app.config['UPLOADED_FILES_DEST'] = 'uploads'  # 上传文件的保存目录
-files = UploadSet('files', ALL)
-configure_uploads(app, files)
+
+dropzone = Dropzone(app)
 
 socketio = SocketIO(app)
 
@@ -33,13 +37,6 @@ 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):
@@ -51,10 +48,6 @@ def list_files(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'))
@@ -65,4 +58,4 @@ if __name__ == '__main__':
     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)