mirror of
https://git.selfprivacy.org/SelfPrivacy/selfprivacy-rest-api.git
synced 2024-11-09 12:43:11 +00:00
111 lines
3.6 KiB
Python
111 lines
3.6 KiB
Python
"""Generic handler for moving services"""
|
|
|
|
from __future__ import annotations
|
|
import subprocess
|
|
import pathlib
|
|
import shutil
|
|
from typing import List
|
|
|
|
from selfprivacy_api.jobs import Job, report_progress
|
|
from selfprivacy_api.utils.block_devices import BlockDevice
|
|
from selfprivacy_api.services.owned_path import OwnedPath
|
|
|
|
|
|
class MoveError(Exception):
|
|
"""Move failed"""
|
|
|
|
|
|
def get_foldername(p: OwnedPath) -> str:
|
|
return p.path.split("/")[-1]
|
|
|
|
|
|
def check_volume(volume: BlockDevice, space_needed: int) -> bool:
|
|
# Check if there is enough space on the new volume
|
|
if int(volume.fsavail) < space_needed:
|
|
raise MoveError("Not enough space on the new volume.")
|
|
|
|
# Make sure the volume is mounted
|
|
if not volume.is_root() and f"/volumes/{volume.name}" not in volume.mountpoints:
|
|
raise MoveError("Volume is not mounted.")
|
|
|
|
|
|
def check_folders(current_volume: BlockDevice, folders: List[OwnedPath]) -> None:
|
|
# Make sure current actual directory exists and if its user and group are correct
|
|
for folder in folders:
|
|
path = pathlib.Path(f"/volumes/{current_volume}/{get_foldername(folder)}")
|
|
|
|
if not path.exists():
|
|
raise MoveError(f"directory {path} is not found.")
|
|
if not path.is_dir():
|
|
raise MoveError(f"{path} is not a directory.")
|
|
if path.owner() != folder.owner:
|
|
raise MoveError(f"{path} is not owned by {folder.owner}.")
|
|
|
|
|
|
def unbind_folders(owned_folders: List[OwnedPath]) -> None:
|
|
for folder in owned_folders:
|
|
try:
|
|
subprocess.run(
|
|
["umount", folder.path],
|
|
check=True,
|
|
)
|
|
except subprocess.CalledProcessError:
|
|
raise MoveError(f"Unable to unmount folder {folder.path}.")
|
|
|
|
|
|
def move_folders_to_volume(
|
|
folders: List[OwnedPath],
|
|
old_volume: BlockDevice,
|
|
new_volume: BlockDevice,
|
|
job: Job,
|
|
) -> None:
|
|
current_progress = job.progress
|
|
folder_percentage = 50 // len(folders)
|
|
for folder in folders:
|
|
folder_name = get_foldername(folder)
|
|
shutil.move(
|
|
f"/volumes/{old_volume}/{folder_name}",
|
|
f"/volumes/{new_volume.name}/{folder_name}",
|
|
)
|
|
progress = current_progress + folder_percentage
|
|
report_progress(progress, job, "Moving data to new volume...")
|
|
|
|
|
|
def ensure_folder_ownership(folders: List[OwnedPath], volume: BlockDevice) -> None:
|
|
for folder in folders:
|
|
true_location = f"/volumes/{volume.name}/{get_foldername(folder)}"
|
|
try:
|
|
subprocess.run(
|
|
[
|
|
"chown",
|
|
"-R",
|
|
f"{folder.owner}:{folder.group}",
|
|
# Could we just chown the binded location instead?
|
|
true_location,
|
|
],
|
|
check=True,
|
|
)
|
|
except subprocess.CalledProcessError as error:
|
|
error_message = (
|
|
f"Unable to set ownership of {true_location} :{error.output}"
|
|
)
|
|
print(error.output)
|
|
raise MoveError(error_message)
|
|
|
|
|
|
def bind_folders(folders: List[OwnedPath], volume: BlockDevice) -> None:
|
|
for folder in folders:
|
|
try:
|
|
subprocess.run(
|
|
[
|
|
"mount",
|
|
"--bind",
|
|
f"/volumes/{volume.name}/{get_foldername(folder)}",
|
|
folder.path,
|
|
],
|
|
check=True,
|
|
)
|
|
except subprocess.CalledProcessError as error:
|
|
print(error.output)
|
|
raise MoveError(f"Unable to mount new volume:{error.output}")
|