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: class DataCollector:
columns = ["idx", "t [s]", "V [V]", "LED [%]"] columns = ["idx", "t [s]", "V [V]", "LED [%]"]
def __init__(self, def __init__(self,
data_path: str, data_path: str,
data_name: str="CPData", data_name: str="CPData",
metadata: dict[str, str]={}, metadata: dict[str, str]={},
dirname: str|None=None, dirname: str|None=None,
dir_exists_is_ok=False, add_number_if_dir_exists=True,
): ):
self.data = [] self.data = []
self.fulldata = None # if loaded, this contains the final numpy array 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) self.dirpath = os.path.join(self.path, self.dirname)
if os.path.exists(self.dirpath): if os.path.exists(self.dirpath):
if not dir_exists_is_ok: if not add_number_if_dir_exists:
raise Exception(f"Directory '{self.dirname}' already exists. Provide a different directory or pass `dir_exists_is_ok=True` to ignore this") raise Exception(f"Directory '{self.dirname}' already exists. Provide a different directory or pass `add_number_if_dir_exists=True` to ignore this")
else: else:
os.makedirs(self.dirpath) 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 self.flushed = False