From c8912232a2ae203c5bf23fcd3a54e8dfc480161e Mon Sep 17 00:00:00 2001
From: Groupr <grop3r@protonmail.com>
Date: Tue, 20 May 2025 17:42:56 +0000
Subject: [PATCH] Update send_node_digest_notifications.py

---
 .../scripts/send_node_digest_notifications.py | 23 +++++++++++++++----
 1 file changed, 18 insertions(+), 5 deletions(-)

diff --git a/remarkbox/scripts/send_node_digest_notifications.py b/remarkbox/scripts/send_node_digest_notifications.py
index eee9a94..d76ace9 100644
--- a/remarkbox/scripts/send_node_digest_notifications.py
+++ b/remarkbox/scripts/send_node_digest_notifications.py
@@ -4,10 +4,10 @@ from . import base_parser
 import transaction
 
 
-def get_arg_parser():
-    parser = base_parser("Send Node Notification Digests.")
-    return parser
 
+def get_arg_parser():
+    parser = base_parser(" Send Node Notification Digests.")
+    return parser
 
 def main():
     parser = get_arg_parser()
@@ -15,5 +15,18 @@ def main():
     setup_logging(args.config)
 
     with bootstrap(args.config) as env:
-        deliver_scheduled_notifications(env["request"])
-        transaction.commit()
+        request = env["request"]
+        with transaction.manager:
+            try:
+                deliver_scheduled_notifications(request)
+                print("Committing transaction.")
+                transaction.commit()
+                print("Committed transaction.")
+                raise SystemExit(0)
+            except Exception as e:
+                print("Aborting transaction.")
+                transaction.abort()
+                raise SystemExit(1)
+    
+if __name__ == "__main__":
+    main()