Update app.py

app.py

@@ -1,34 +1,33 @@
-from flask import Flask, render_template, send_from_directory
+from flask import Flask, send_from_directory, render_template_string
 import os
 
 app = Flask(__name__)
 
-# 指定要浏览的目录
-base_directory = '/app'
-
 @app.route('/')
-def list_files():
-    # 获取目录下的文件和子目录
-    entries = os.listdir(base_directory)
+@app.route('/<path:req_path>')
+def dir_listing(req_path=''):
+    base_dir = os.path.abspath(os.getcwd())  # 使用绝对路径
+    abs_path = os.path.join(base_dir, req_path)
 
-    # 构建一个 HTML 列表来显示所有条目，并为每个条目创建一个链接
-    file_list = "<ul>"
-    for entry in entries:
-        full_path = os.path.join(base_directory, entry)
-        if os.path.isdir(full_path):
-            # 如果是子目录，创建链接
-            file_list += f"<li><a href='{entry}/'>{entry}/</a></li>"
-        else:
-            # 如果是文件，直接显示文件名
-            file_list += f"<li>{entry}</li>"
-    file_list += "</ul>"
+    # 检查路径是否存在
+    if not os.path.exists(abs_path):
+        return render_template_string('<h1>404 Not Found</h1>'), 404
 
-    return f"<h1>当前目录下的文件和子目录：</h1>{file_list}"
+    # 检查路径是否是文件
+    if os.path.isfile(abs_path):
+        # 如果是文件，则下载文件
+        return send_from_directory(base_dir, req_path, as_attachment=True)
 
-@app.route('/<path:filename>')
-def download_file(filename):
-    # 发送文件内容给用户
-    return send_from_directory(base_directory, filename)
+    # 显示目录内容
+    files = os.listdir(abs_path)
+    return render_template_string('''
+        <h1>Directory listing for {{req_path}}</h1>
+        <ul>
+            {% for file in files %}
+                <li><a href="{{ req_path|safe }}/{{ file|safe }}">{{ file }}</a></li>
+            {% endfor %}
+        </ul>
+    ''', files=files, req_path=req_path)
 
 if __name__ == '__main__':
     app.run(debug=True)