Update app.py

app.py

@@ -1,48 +1,56 @@
+import json
 import os
+from pathlib import Path
 from http.server import BaseHTTPRequestHandler, HTTPServer
 
-DATA_DIR = "/data"
-
-def list_files(base):
-    found = []
-    for root, dirs, files in os.walk(base):
-        for name in files:
-            full = os.path.join(root, name)
-            rel = os.path.relpath(full, base)
-            found.append(rel)
-    return sorted(found)
-
-print("=== STARTUP ===")
-print(f"Checking {DATA_DIR}")
-
-if os.path.exists(DATA_DIR):
-    files = list_files(DATA_DIR)
-    if files:
-        print("=== FILE NAMES ===")
-        for f in files:
-            print(f"[FILE] {f}")
-    else:
-        print("No files found in /data")
-else:
-    print("/data does not exist")
+# ĐẶT biến này đúng bằng mount path bạn chọn trong Space Settings
+MOUNT_PATH = os.getenv("BUCKET_MOUNT_PATH", "/bucket")
+
+def scan(base: str):
+    p = Path(base)
+    if not p.exists():
+        return {
+            "exists": False,
+            "path": base,
+            "count": 0,
+            "files": []
+        }
+
+    files = []
+    for x in sorted(p.rglob("*")):
+        if x.is_file():
+            st = x.stat()
+            files.append({
+                "path": str(x.relative_to(p)),
+                "size": st.st_size
+            })
+
+    return {
+        "exists": True,
+        "path": base,
+        "count": len(files),
+        "files": files
+    }
 
 class Handler(BaseHTTPRequestHandler):
     def do_GET(self):
-        files = []
-        if os.path.exists(DATA_DIR):
-            files = list_files(DATA_DIR)
-
-        body = "<h1>/data files</h1>"
-        if files:
-            body += "<ul>" + "".join(f"<li>{f}</li>" for f in files) + "</ul>"
-        else:
-            body += "<p>No files found in /data</p>"
+        data = scan(MOUNT_PATH)
+        body = json.dumps(data, ensure_ascii=False, indent=2).encode("utf-8")
 
         self.send_response(200)
-        self.send_header("Content-Type", "text/html; charset=utf-8")
+        self.send_header("Content-Type", "application/json; charset=utf-8")
+        self.send_header("Content-Length", str(len(body)))
         self.end_headers()
-        self.wfile.write(body.encode("utf-8"))
+        self.wfile.write(body)
+
+    def log_message(self, fmt, *args):
+        print(fmt % args, flush=True)
+
+if __name__ == "__main__":
+    data = scan(MOUNT_PATH)
+
+    print("=== BUCKET_SCAN_START ===", flush=True)
+    print(json.dumps(data, ensure_ascii=False, indent=2), flush=True)
+    print("=== BUCKET_SCAN_END ===", flush=True)
 
-port = int(os.environ.get("PORT", "7860"))
-print(f"Serving on 0.0.0.0:{port}")
-HTTPServer(("0.0.0.0", port), Handler).serve_forever()
+    HTTPServer(("0.0.0.0", 7860), Handler).serve_forever()