diff --git a/src/dirmanager.py b/src/dirmanager.py index 4fa4844..dfc149d 100644 --- a/src/dirmanager.py +++ b/src/dirmanager.py @@ -21,18 +21,19 @@ class DirManager: self.tests_dir = tests_dir self.session_id = session_id - self.get_all_dirs() - - def get_all_dirs(self): - self.session_dir = self.tests_dir / f"test-{self.session_id}" - self.session_sub_dirs = self.get_subdirs(self.session_dir) + self.dirs = self._get_all_dirs() def create_all_dirs(self): self.create_dirs(self.tests_dir, exist_ok=True) - self.create_dirs(self.session_dir) - self.create_dirs(self.session_sub_dirs) + self.create_dirs(self.dirs) - def get_subdirs(self, session_dir: Path): + def _get_all_dirs(self): + dirs = {} + dirs["session"] = self.tests_dir / f"test-{self.session_id}" + dirs.update(self._get_subdirs(session_dir=dirs["session"])) + return dirs + + def _get_subdirs(self, session_dir: Path): return { "records": session_dir / Path("records"), "states": session_dir / Path("states"),