MatthiasQuintern/cpdctrl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Auto add number when dir exists

Browse Source
This commit is contained in:
CPD 2025-03-04 17:50:59 +01:00
parent 4846608b4b
commit 9d2c35f466
1 changed files with 14 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
cpdctrl/utility/data.py
View File

@ -11,12 +11,12 @@ METADATA_FILENAME = "_measurement_metadata.pkl"
class DataCollector:
columns = ["idx", "t [s]", "V [V]", "LED [%]"]
def __init__(self,
def __init__(self,
data_path: str,
data_name: str="CPData",
data_name: str="CPData",
metadata: dict[str, str]={},
dirname: str|None=None,
dir_exists_is_ok=False,
add_number_if_dir_exists=True,
):
self.data = []
self.fulldata = None # if loaded, this contains the final numpy array
@ -30,10 +30,17 @@ class DataCollector:
self.dirpath = os.path.join(self.path, self.dirname)
if os.path.exists(self.dirpath):
if not dir_exists_is_ok:
raise Exception(f"Directory '{self.dirname}' already exists. Provide a different directory or pass `dir_exists_is_ok=True` to ignore this")
else:
os.makedirs(self.dirpath)
if not add_number_if_dir_exists:
raise Exception(f"Directory '{self.dirname}' already exists. Provide a different directory or pass `add_number_if_dir_exists=True` to ignore this")
else:
i = 1
dirpath = f"{self.dirpath}-{i}"
while os.path.exists(dirpath):
i += 1
dirpath = f"{self.dirpath}-{i}"
print(f"Directory '{self.dirname}' already exists. Trying '{dirpath}' instead")
self.dirpath = dirpath
os.makedirs(self.dirpath)
self.flushed = False