From 08e02d12dbf47c317d948843c1188815e68af932 Mon Sep 17 00:00:00 2001
From: John Lancaster <32917998+jsl12@users.noreply.github.com>
Date: Mon, 3 Jun 2024 21:10:21 -0500
Subject: [PATCH] improved logic in forget

---
 src/restic/forget.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/src/restic/forget.py b/src/restic/forget.py
index def37ed..b1437b4 100644
--- a/src/restic/forget.py
+++ b/src/restic/forget.py
@@ -86,9 +86,12 @@ def main(loki_url: str, dry_run: bool, **kwargs):
     except Exception:
         console.print_exception()
     else:
-        snapshots.snapshot(loki_url)
         prune(loki_url, dry_run)
-        size.get_size(loki_url)
+        if not dry_run:
+            snapshots.snapshot(loki_url)
+            size.get_size(loki_url)
+        else:
+            logger.debug('Skipping getting size because of dry run')
 
 
 if __name__ == '__main__':