timmy / 磁碟使用量查詢 API

import json from http.server import BaseHTTPRequestHandler, HTTPServer from urllib.parse import urlparse, parse_qs import shutil class DiskUsage: def __init__(self, path="/"): self.path = path def get_usage(self): try: total, used, free = shutil.disk_usage(self.path) usage_percent = (used / total) * 100 if total > 0 else 0 return { "path": self.path, "total_space": total, "used_space": used, "free_space": free, "usage_percent": usage_percent, } except Exception as e: return {"error": str(e)} @staticmethod def format_size(size_in_bytes): for unit in ["B", "KB", "MB", "GB", "TB"]: if size_in_bytes < 1024: return f"{size_in_bytes:.2f} {unit}" size_in_bytes /= 1024 return f"{size_in_bytes:.2f} PB" class DiskUsageHandler(BaseHTTPRequestHandler): def do_GET(self): parsed_path = urlparse(self.path) if parsed_path.path == "/disk-usage": # 取得 query string 中的 path 參數 query = parse_qs(parsed_path.query) path = query.get("path", ["/"])[0] # 執行 DiskUsage 並取得結果 disk_usage = DiskUsage(path) usage = disk_usage.get_usage() # 回傳 JSON 結果 self.send_response(200) self.send_header("Content-Type", "application/json") self.end_headers() self.wfile.write(json.dumps(usage).encode("utf-8")) else: self.send_response(404) self.send_header("Content-Type", "application/json") self.end_headers() self.wfile.write(json.dumps({"error": "Not Found"}).encode("utf-8")) def run(server_class=HTTPServer, handler_class=DiskUsageHandler, port=8000): server_address = ("", port) httpd = server_class(server_address, handler_class) print(f"Starting server on port {port}...") httpd.serve_forever() if __name__ == "__main__": run()