feat: add adapter usage tracking and retrieval methods across various adapters

This commit is contained in:
shiyu
2026-05-02 21:55:35 +08:00
parent dcc8aa139e
commit a8737b883e
11 changed files with 225 additions and 2 deletions
+5
View File
@@ -21,3 +21,8 @@ class BaseAdapter(Protocol):
async def stream_file(self, root: str, rel: str, range_header: str | None): ...
async def stat_file(self, root: str, rel: str): ...
def get_effective_root(self, sub_path: str | None) -> str: ...
@runtime_checkable
class UsageCapableAdapter(Protocol):
async def get_usage(self, root: str) -> Dict: ...
+17 -1
View File
@@ -455,6 +455,23 @@ class DropboxAdapter:
return StreamingResponse(iterator(), status_code=resp.status_code, headers=out_headers, media_type=content_type)
async def get_usage(self, root: str):
resp = await self._api_json("/users/get_space_usage", {})
resp.raise_for_status()
payload = resp.json() or {}
allocation = payload.get("allocation") or {}
allocated = allocation.get("allocated")
used = payload.get("used")
total = int(allocated) if allocated is not None else None
used_bytes = int(used) if used is not None else None
return {
"used_bytes": used_bytes,
"total_bytes": total,
"free_bytes": total - used_bytes if total is not None and used_bytes is not None else None,
"source": "dropbox",
"scope": "account",
}
ADAPTER_TYPE = "dropbox"
CONFIG_SCHEMA = [
@@ -468,4 +485,3 @@ CONFIG_SCHEMA = [
def ADAPTER_FACTORY(rec): return DropboxAdapter(rec)
+16
View File
@@ -541,6 +541,22 @@ class GoogleDriveAdapter:
except Exception:
return None
async def get_usage(self, root: str):
resp = await self._request("GET", "/about", params={"fields": "storageQuota"})
resp.raise_for_status()
quota = (resp.json() or {}).get("storageQuota") or {}
limit = quota.get("limit")
usage = quota.get("usage")
total = int(limit) if limit is not None else None
used = int(usage) if usage is not None else None
return {
"used_bytes": used,
"total_bytes": total,
"free_bytes": total - used if total is not None and used is not None else None,
"source": "googledrive",
"scope": "drive",
}
ADAPTER_TYPE = "googledrive"
+23
View File
@@ -329,6 +329,29 @@ class LocalAdapter:
info["exif"] = exif
return info
async def get_usage(self, root: str):
root_path = Path(root).resolve()
def _usage():
used = 0
for dirpath, dirnames, filenames in os.walk(root_path):
for filename in filenames:
fp = Path(dirpath) / filename
try:
used += fp.stat().st_size
except OSError:
continue
disk = shutil.disk_usage(root_path)
return {
"used_bytes": used,
"total_bytes": disk.total,
"free_bytes": disk.free,
"source": "local",
"scope": "mount",
}
return await asyncio.to_thread(_usage)
ADAPTER_TYPE = "local"
CONFIG_SCHEMA = [
+15
View File
@@ -443,6 +443,21 @@ class OneDriveAdapter:
resp.raise_for_status()
return self._format_item(resp.json())
async def get_usage(self, root: str):
resp = await self._request("GET", full_url=f"{MS_GRAPH_URL}/me/drive?$select=quota")
resp.raise_for_status()
quota = (resp.json() or {}).get("quota") or {}
used = quota.get("used")
total = quota.get("total")
remaining = quota.get("remaining")
return {
"used_bytes": int(used) if used is not None else None,
"total_bytes": int(total) if total is not None else None,
"free_bytes": int(remaining) if remaining is not None else None,
"source": "onedrive",
"scope": "drive",
}
ADAPTER_TYPE = "onedrive"
+15
View File
@@ -776,6 +776,21 @@ class PikPakAdapter:
return None
return resp.content
async def get_usage(self, root: str):
data = await self._request("GET", "/about")
quota = data.get("quota") or {}
limit = quota.get("limit")
usage = quota.get("usage")
total = int(limit) if limit is not None else None
used = int(usage) if usage is not None else None
return {
"used_bytes": used,
"total_bytes": total,
"free_bytes": total - used if total is not None and used is not None else None,
"source": "pikpak",
"scope": "drive",
}
async def mkdir(self, root: str, rel: str):
rel = (rel or "").strip("/")
if not rel:
+17
View File
@@ -840,6 +840,23 @@ class QuarkAdapter:
async def copy(self, root: str, src_rel: str, dst_rel: str, overwrite: bool = False):
raise NotImplementedError("QuarkOpen does not support copy via open API")
async def get_usage(self, root: str):
data = await self._request("GET", "/capacity/growth/info")
payload = (data or {}).get("data") or {}
if isinstance(payload.get("member"), dict):
payload = payload["member"]
used = payload.get("use_capacity") or payload.get("used_capacity")
total = payload.get("total_capacity")
used_bytes = int(used) if used is not None else None
total_bytes = int(total) if total is not None else None
return {
"used_bytes": used_bytes,
"total_bytes": total_bytes,
"free_bytes": total_bytes - used_bytes if total_bytes is not None and used_bytes is not None else None,
"source": "quark",
"scope": "account",
}
# -----------------
# STAT / EXISTS / 辅助
# -----------------