52 lines
1.7 KiB
Python
52 lines
1.7 KiB
Python
import asyncio
|
|
import os
|
|
import logging
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
# Allow overriding the zpool binary path via env (for bind-mounted host tools)
|
|
ZPOOL_BIN = os.environ.get("ZPOOL_BIN", "zpool")
|
|
|
|
|
|
async def get_zfs_pool_map() -> dict[str, str]:
|
|
"""Return a dict mapping device names to ZFS pool names.
|
|
|
|
e.g. {"sda": "tank", "sdb": "tank", "sdc": "fast"}
|
|
"""
|
|
pool_map = {}
|
|
try:
|
|
env = os.environ.copy()
|
|
host_lib = os.environ.get("ZFS_HOST_LIB")
|
|
if host_lib:
|
|
env["LD_LIBRARY_PATH"] = host_lib
|
|
|
|
proc = await asyncio.create_subprocess_exec(
|
|
ZPOOL_BIN, "status", "-P",
|
|
stdout=asyncio.subprocess.PIPE,
|
|
stderr=asyncio.subprocess.PIPE,
|
|
env=env,
|
|
)
|
|
stdout, _ = await proc.communicate()
|
|
if proc.returncode != 0:
|
|
return pool_map
|
|
|
|
current_pool = None
|
|
for line in stdout.decode().splitlines():
|
|
stripped = line.strip()
|
|
if stripped.startswith("pool:"):
|
|
current_pool = stripped.split(":", 1)[1].strip()
|
|
elif current_pool and "/dev/" in stripped:
|
|
parts = stripped.split()
|
|
dev_path = parts[0]
|
|
try:
|
|
real = os.path.realpath(dev_path)
|
|
dev_name = os.path.basename(real)
|
|
# Strip partition numbers (sda1 -> sda)
|
|
dev_name = dev_name.rstrip("0123456789")
|
|
pool_map[dev_name] = current_pool
|
|
except Exception:
|
|
pass
|
|
except FileNotFoundError:
|
|
logger.debug("zpool not available")
|
|
return pool_map
|