update docstring and argname of Dirmanager

This commit is contained in:
Daniel 2023-11-22 14:23:35 +01:00
parent 0d1dd4ca17
commit ea51e508d7
2 changed files with 19 additions and 5 deletions

View file

@ -2,19 +2,33 @@ from pathlib import Path
class DirManager: class DirManager:
def __init__(self, root_tests_dir: Path, session_id: str): """Manages directories for the tests and should be used to create and find
and use the correct directories.
The structures is as follows:
tests dir/
session_dir-1/
records
states
results
session_dir-2/
records
...
"""
def __init__(self, tests_dir: Path, session_id: str):
# root test dir # root test dir
self.root_tests_dir = root_tests_dir self.tests_dir = tests_dir
self.session_id = session_id self.session_id = session_id
self.get_all_dirs() self.get_all_dirs()
def get_all_dirs(self): def get_all_dirs(self):
self.session_dir = self.root_tests_dir / f"test-{self.session_id}" self.session_dir = self.tests_dir / f"test-{self.session_id}"
self.session_sub_dirs = self.get_subdirs(self.session_dir) self.session_sub_dirs = self.get_subdirs(self.session_dir)
def create_all_dirs(self): def create_all_dirs(self):
self.create_dirs(self.root_tests_dir, exist_ok=True) self.create_dirs(self.tests_dir, exist_ok=True)
self.create_dirs(self.session_dir) self.create_dirs(self.session_dir)
self.create_dirs(self.session_sub_dirs) self.create_dirs(self.session_sub_dirs)

View file

@ -42,7 +42,7 @@ class Wrapper:
self.run_test() self.run_test()
def setup_test(self, session_id: str): def setup_test(self, session_id: str):
self.dir_manager = DirManager(root_tests_dir=TESTS_DIR, session_id=session_id) self.dir_manager = DirManager(tests_dir=TESTS_DIR, session_id=session_id)
self.dir_manager.create_all_dirs() self.dir_manager.create_all_dirs()
def run_test(self): def run_test(self):