File size: 2,269 Bytes
5efaf20 d556ae4 5efaf20 d556ae4 5efaf20 d556ae4 5efaf20 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 |
import os
import shutil
from fastapi import FastAPI
from typing import Dict, Any
app = FastAPI(title="Hugging Face Space Monitor", version="1.0.0")
@app.get("/")
def greet_json():
return {"Hello": "World!"}
@app.get("/storage")
def get_storage_info() -> Dict[str, Any]:
"""取得目前 Space 的存儲使用情況"""
try:
# 取得根目錄的磁碟使用情況
total, used, free = shutil.disk_usage("/")
# 取得當前工作目錄大小
app_size = get_directory_size("/app")
# 取得 /tmp 目錄使用情況（臨時檔案）
tmp_size = get_directory_size("/tmp")
return {
"disk_usage": {
"total_bytes": total,
"used_bytes": used,
"free_bytes": free,
"total_gb": round(total / (1024**3), 2),
"used_gb": round(used / (1024**3), 2),
"free_gb": round(free / (1024**3), 2),
"usage_percentage": round((used / total) * 100, 2)
},
"application": {
"app_size_bytes": app_size,
"app_size_mb": round(app_size / (1024**2), 2),
"tmp_size_bytes": tmp_size,
"tmp_size_mb": round(tmp_size / (1024**2), 2)
},
"environment": {
"current_dir": os.getcwd(),
"space_id": os.getenv("SPACE_ID", "unknown"),
"space_author": os.getenv("SPACE_AUTHOR_NAME", "unknown")
}
}
except Exception as e:
return {"error": str(e)}
def get_directory_size(path: str) -> int:
"""計算目錄大小"""
total_size = 0
try:
for dirpath, dirnames, filenames in os.walk(path):
for filename in filenames:
filepath = os.path.join(dirpath, filename)
try:
total_size += os.path.getsize(filepath)
except (OSError, FileNotFoundError):
continue
except (OSError, FileNotFoundError):
pass
return total_size
@app.get("/health")
def health_check():
"""健康檢查端點"""
return {
"status": "healthy",
"message": "Space is running normally"
}
|