refactor(services): add a generic storage counter

restic-rewrite-api
Houkime 2023-04-17 14:02:07 +00:00
parent a34b14449b
commit 789fd71e2f
2 changed files with 9 additions and 5 deletions

View File

@ -113,7 +113,7 @@ class Gitea(Service):
def get_storage_usage() -> int:
storage_usage = 0
for folder in Gitea.get_folders():
storage_usage += get_storage_usage()
storage_usage += get_storage_usage(folder)
return storage_usage
@staticmethod

View File

@ -8,6 +8,8 @@ from selfprivacy_api.jobs import Job
from selfprivacy_api.utils.block_devices import BlockDevice
from selfprivacy_api.services.generic_size_counter import get_storage_usage
class ServiceStatus(Enum):
"""Enum for service status"""
@ -120,10 +122,12 @@ class Service(ABC):
def get_logs():
pass
@staticmethod
@abstractmethod
def get_storage_usage() -> int:
pass
@classmethod
def get_storage_usage(cls) -> int:
storage_used = 0
for folder in cls.get_folders():
storage_used += get_storage_usage(folder)
return storage_used
@staticmethod
@abstractmethod