SelfPrivacy/selfprivacy-rest-api
1
0
Fork 0
mirror of https://git.selfprivacy.org/SelfPrivacy/selfprivacy-rest-api.git synced 2024-11-25 21:41:27 +00:00
Code Issues Projects Releases Wiki Activity

refactor(services): add a generic storage counter

Browse source
This commit is contained in:
Houkime 2023-04-17 14:02:07 +00:00 committed by Inex Code
parent 0a9848be47
commit c2cd972805
2 changed files with 9 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
selfprivacy_api/services/gitea/__init__.py
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

12
selfprivacy_api/services/service.py
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