Notify after post-backup-tasks borgbase#1632

TheLazron · Feb 21, 2024 · e17e5c0 · e17e5c0
1 parent 472c7c8
commit e17e5c0
Showing 1 changed file with 25 additions and 9 deletions.
diff --git a/src/vorta/scheduler.py b/src/vorta/scheduler.py
@@ -430,16 +430,32 @@ def notify(self, result):
         profile_id = result['params']['profile'].id
 
         if result['returncode'] in [0, 1]:
-            notifier.deliver(
-                self.tr('Vorta Backup'),
-                self.tr('Backup successful for %s.') % profile_name,
-                level='info',
-            )
-            logger.info('Backup creation successful.')
-            # unpause scheduler
-            self.unpause(result['params']['profile_id'])
+            try:
+                notifier.deliver(
+                    self.tr('Vorta Backup'),
+                    self.tr('Backup successful for %s.') % profile_name,
+                    level='info',
+                )
+                logger.info('Backup creation successful.')
+                # unpause scheduler
+                self.unpause(result['params']['profile_id'])
 
-            self.post_backup_tasks(profile_id)
+                self.post_backup_tasks(profile_id)
+                # Notify after successful post_backup_tasks
+                notifier.deliver(
+                    self.tr('Vorta Backup'),
+                    self.tr('Post Backup Tasks successful for %s' % profile_id),
+                    level='info',
+                )
+                logger.info('Post Backup Tasks successful for %s' % profile_id)
+
+            except Exception as e:
+                # Handle exceptions if post_backup_tasks fails
+                notifier.deliver(
+                    self.tr('Vorta Backup'),
+                    self.tr('Error during post backup tasks for %s.') % profile_name,
+                    level='error',
+                )
         else:
             notifier.deliver(
                 self.tr('Vorta Backup'),