Remove old files and rename files with correct extensions

bessagroup · Jun 26, 2024 · 0678959 · 0678959
1 parent 1eaae8b
commit 0678959
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git a/src/f3dasm/_src/design/domain.py b/src/f3dasm/_src/design/domain.py
@@ -279,6 +279,9 @@ def store(self, filename: Path) -> None:
  with open(filename.with_suffix('.tmp'), 'wb') as f:
  pickle.dump(self, f)
 
+ # remove old file if it exists
+ filename.with_suffix('.pkl').unlink(missing_ok=True)
+
  # rename the file to the correct extension
  filename.with_suffix('.tmp').rename(filename.with_suffix('.pkl'))
 

diff --git a/src/f3dasm/_src/experimentdata/_data.py b/src/f3dasm/_src/experimentdata/_data.py
@@ -266,6 +266,9 @@ def store(self, filename: Path) -> None:
  # TODO: The column information is not saved in the .csv!
  self.to_dataframe().to_csv(filename.with_suffix('.tmp'))
 
+ # remove the old file if it exists
+ filename.with_suffix('.csv').unlink(missing_ok=True)
+
  # rename the file to the correct extension
  filename.with_suffix('.tmp').rename(filename.with_suffix('.csv'))
 

diff --git a/src/f3dasm/_src/experimentdata/_jobqueue.py b/src/f3dasm/_src/experimentdata/_jobqueue.py
@@ -199,6 +199,9 @@ def store(self, filename: Path) -> None:
  """
  self.jobs.to_pickle(filename.with_suffix('.tmp'))
 
+ # remove old file if it exists
+ filename.with_suffix('.pkl').unlink(missing_ok=True)
+
  # rename the file to the correct extension
  filename.with_suffix('.tmp').rename(filename.with_suffix('.pkl'))