SelfPrivacy/selfprivacy-rest-api
1
0
Fork 0
mirror of https://git.selfprivacy.org/SelfPrivacy/selfprivacy-rest-api.git synced 2024-11-29 15:31:28 +00:00
Code Issues Projects Releases Wiki Activity

feature(backup):calculate needed space for inplace restoration

Browse source
This commit is contained in:
Houkime 2023-07-14 12:34:45 +00:00 committed by Inex Code
parent f711275a5e
commit 45011450c5
1 changed files with 18 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
selfprivacy_api/backup/__init__.py
View file

@ -1,6 +1,6 @@
from datetime import datetime, timedelta from datetime import datetime, timedelta
from operator import add from operator import add
from os import statvfs from os import statvfs, path, walk
from typing import List, Optional from typing import List, Optional
from selfprivacy_api.utils import ReadUserData, WriteUserData from selfprivacy_api.utils import ReadUserData, WriteUserData
@ -277,14 +277,28 @@ class Backups:
Jobs.update(job, status=JobStatus.FINISHED) Jobs.update(job, status=JobStatus.FINISHED)
@staticmethod @staticmethod
def _assert_restorable(snapshot: Snapshot): def _assert_restorable(
snapshot: Snapshot, strategy=RestoreStrategy.DOWNLOAD_VERIFY_OVERWRITE
):
service = get_service_by_id(snapshot.service_name) service = get_service_by_id(snapshot.service_name)
if service is None: if service is None:
raise ValueError( raise ValueError(
f"snapshot has a nonexistent service: {snapshot.service_name}" f"snapshot has a nonexistent service: {snapshot.service_name}"
) )
needed_space = Backups.snapshot_restored_size(snapshot.id) restored_snap_size = Backups.snapshot_restored_size(snapshot.id)
if strategy == RestoreStrategy.DOWNLOAD_VERIFY_OVERWRITE:
needed_space = restored_snap_size
elif strategy == RestoreStrategy.INPLACE:
needed_space = restored_snap_size - service.get_storage_usage()
else:
raise NotImplementedError(
"""
We do not know if there is enough space for restoration because there is some novel restore strategy used!
This is a developer's fault, open a issue please
"""
)
available_space = Backups.space_usable_for_service(service) available_space = Backups.space_usable_for_service(service)
if needed_space > available_space: if needed_space > available_space:
raise ValueError( raise ValueError(
@ -466,6 +480,7 @@ class Backups:
if folders == []: if folders == []:
raise ValueError("unallocated service", service.get_id()) raise ValueError("unallocated service", service.get_id())
# We assume all folders of one service live at the same volume
fs_info = statvfs(folders[0]) fs_info = statvfs(folders[0])
usable_bytes = fs_info.f_frsize * fs_info.f_bavail usable_bytes = fs_info.f_frsize * fs_info.f_bavail
return usable_bytes return usable_bytes