integrate JourneyDataManager with transactions

1 files changed, 15 insertions, 5 deletions

diff --git a/fietsboek/data.py b/fietsboek/data.py
index e288a03..1e97ada 100644
--- a/fietsboek/data.py
+++ b/fietsboek/data.py
@@ -382,9 +382,10 @@ class UserDataDir:
 class JourneyDataDir:
     """Manager for a single journey's data."""
 
-    def __init__(self, journey_id: int, path: Path):
+    def __init__(self, journey_id: int, path: Path, *, txn: Transaction | None = None):
         self.journey_id = journey_id
         self.path = path
+        self.txn = txn
 
     def purge(self):
         """Purge all data pertaining to the journey.
@@ -392,8 +393,11 @@ class JourneyDataDir:
         This function logs errors but raises no exception, as such it can
         always be used to clean up after a track.
         """
-        if self.path.is_dir():
-            shutil.rmtree(self.path, ignore_errors=False, onerror=_log_deletion_error)
+        if self.txn:
+            self.txn.purge(self.path)
+        else:
+            if self.path.is_dir():
+                shutil.rmtree(self.path, ignore_errors=False, onerror=_log_deletion_error)
 
     def preview_path(self) -> Path:
         """Gets the path to the "preview image".
@@ -407,11 +411,17 @@ class JourneyDataDir:
 
         :param data: The data of the preview image.
         """
-        self.preview_path().write_bytes(data)
+        if self.txn:
+            self.txn.write_bytes(self.preview_path(), data)
+        else:
+            self.preview_path().write_bytes(data)
 
     def remove_preview(self):
         """Deletes the preview image."""
-        self.preview_path().unlink()
+        if self.txn:
+            self.txn.unlink(self.preview_path())
+        else:
+            self.preview_path().unlink()
 
 
 __all__ = ["generate_filename", "DataManager", "TrackDataDir", "UserDataDir", "JourneyDataDir"]