refactoring storages

This commit is contained in:
msramalho
2022-06-03 15:46:00 +02:00
parent 10f03cb888
commit d33daabee1
4 changed files with 85 additions and 137 deletions

View File

@@ -5,6 +5,7 @@ from pathlib import Path
class Storage(ABC):
TMP_FOLDER = "tmp/"
@abstractmethod
def __init__(self, config): pass
@@ -28,18 +29,18 @@ class Storage(ABC):
and others not, but that all can call
"""
for k, v in kwargs.items():
if k in self.get_allowed_properties():
if k in self._get_allowed_properties():
setattr(self, k, v)
else:
logger.warning(f'[{self.__class__.__name__}] does not accept dynamic property "{k}"')
def get_allowed_properties(self):
def _get_allowed_properties(self):
"""
child classes should specify which properties they allow to be set
"""
return set(["subfolder"])
def clean_path(self, folder, default="", add_forward_slash=True):
def _clean_path(self, folder, default="", add_forward_slash=True):
if folder is None or type(folder) != str or len(folder.strip()) == 0:
return default
return str(Path(folder)) + ("/" if add_forward_slash else "")