feat: added steps dir

This commit is contained in:
Hazel 2025-04-24 11:00:54 +02:00
parent 0895256dc4
commit cb9e594837

View File

@ -17,6 +17,8 @@ class RawImage:
self.meta_data = self.read_meta() self.meta_data = self.read_meta()
self.image = self.get_image() self.image = self.get_image()
self.steps_dir = self._get_dir("steps")
def _get_path(self, ending: str, original_suffix: bool = False) -> Path: def _get_path(self, ending: str, original_suffix: bool = False) -> Path:
if original_suffix: if original_suffix:
@ -24,6 +26,11 @@ class RawImage:
else: else:
return self.file.with_name(self.file.stem + "_" + ending) return self.file.with_name(self.file.stem + "_" + ending)
def _get_dir(self, name: str) -> Path:
p = self._get_path(ending=name, original_suffix=False)
p.mkdir(exist_ok=True, parents=True)
return p
def read_meta(self) -> dict: def read_meta(self) -> dict:
if not self.meta_file.exists(): if not self.meta_file.exists():
return {} return {}