From cb9e59483792bc99ebfc496d7152b87ff2d31d85 Mon Sep 17 00:00:00 2001 From: Lars Noack Date: Thu, 24 Apr 2025 11:00:54 +0200 Subject: [PATCH] feat: added steps dir --- secure_pixelation/data_classes.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/secure_pixelation/data_classes.py b/secure_pixelation/data_classes.py index 9bbdfe7..6f6baf7 100644 --- a/secure_pixelation/data_classes.py +++ b/secure_pixelation/data_classes.py @@ -17,6 +17,8 @@ class RawImage: self.meta_data = self.read_meta() self.image = self.get_image() + + self.steps_dir = self._get_dir("steps") def _get_path(self, ending: str, original_suffix: bool = False) -> Path: if original_suffix: @@ -24,6 +26,11 @@ class RawImage: else: 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: if not self.meta_file.exists(): return {}