rawfile-localpv/declarative.py

89 lines
2.0 KiB
Python
Raw Normal View History

2020-06-13 23:47:32 +00:00
import os
import subprocess
2020-06-13 23:47:32 +00:00
from pathlib import Path
from util import run
def be_absent(path):
path = Path(path)
if path.is_symlink():
path.unlink()
elif path.is_file():
2020-06-13 23:47:32 +00:00
path.unlink()
elif path.is_dir():
path.rmdir()
# XXX: should we `shutil.rmtree(path)` instead?
elif not path.exists():
return
2020-06-13 23:47:32 +00:00
else:
raise Exception("Unknown file type")
def be_symlink(path, to):
path = Path(path)
to = Path(to)
if path.is_symlink():
if os.readlink(path) == str(to):
return
be_absent(path)
2020-06-13 23:47:32 +00:00
path.symlink_to(to)
def be_mounted(dev, mountpoint):
dev = Path(dev).resolve()
mountpoint = Path(mountpoint)
if mountpoint.is_mount():
if True: # TODO: verify that the right device is mounted
return
# noinspection PyUnreachableCode
be_unmounted(mountpoint)
run(f"mount {dev} {mountpoint}")
def be_unmounted(path):
path = Path(path)
while path.is_mount():
run(f"umount {path}")
def current_fs(device):
res = subprocess.run(
f"blkid -o value -s TYPE {device}", shell=True, capture_output=True
)
if res.returncode == 2: # specified token was not found
return None
return res.stdout.decode().strip()
def be_formatted(dev, fs):
2021-06-25 21:07:52 +00:00
def init_fs(device):
if fs == "ext4":
run(f"mkfs.ext4 {device}")
elif fs == "btrfs":
run(f"mkfs.btrfs {device}")
else:
2021-06-25 21:07:52 +00:00
raise Exception(f"Unsupported fs type: {fs}")
dev = Path(dev).resolve()
current = current_fs(dev)
if current is None:
2021-06-25 21:07:52 +00:00
init_fs(dev)
else:
if current != fs:
raise Exception(f"Existing filesystem does not match: {current}/{fs}")
2021-06-25 21:07:52 +00:00
def be_fs_expanded(dev, path):
dev = Path(dev).resolve()
2021-06-25 21:07:52 +00:00
fs = current_fs(dev)
path = Path(path).resolve()
if fs == "ext4":
run(f"resize2fs {dev}")
elif fs == "btrfs":
run(f"btrfs filesystem resize max {path}")
else:
raise Exception(f"Unsupported fsType: {fs}")