From 693d2ebdb702be536cdc393f66384f0451966904 Mon Sep 17 00:00:00 2001
From: Kenny Keslar <r3dey3@r3dey3.com>
Date: Wed, 2 Mar 2016 15:57:57 -0500
Subject: [PATCH] Catch exceptions in button handlers

---
 i3pystatus/core/modules.py | 41 ++++++++++++++++++++------------------
 1 file changed, 22 insertions(+), 19 deletions(-)

diff --git a/i3pystatus/core/modules.py b/i3pystatus/core/modules.py
index f60ac3e..952b221 100644
--- a/i3pystatus/core/modules.py
+++ b/i3pystatus/core/modules.py
@@ -94,25 +94,28 @@ class Module(SettingsBase):
         else:
             args = []
 
-        our_method = is_method_of(cb, self)
-        if callable(cb) and not our_method:
-            self.__log_button_event(button, cb, args, "Python callback")
-            cb(*args)
-        elif our_method:
-            cb(self, *args)
-        elif hasattr(self, cb):
-            if cb is not "run":
-                # CommandEndpoint already calls run() after every
-                # callback to instantly update any changed state due
-                # to the callback's actions.
-                self.__log_button_event(button, cb, args, "Member callback")
-                getattr(self, cb)(*args)
-        else:
-            self.__log_button_event(button, cb, args, "External command")
-            if hasattr(self, "data"):
-                args = [arg.format(**self.data) for arg in args]
-                cb = cb.format(**self.data)
-            execute(cb + " " + " ".join(args), detach=True)
+        try:
+            our_method = is_method_of(cb, self)
+            if callable(cb) and not our_method:
+                self.__log_button_event(button, cb, args, "Python callback")
+                cb(*args)
+            elif our_method:
+                cb(self, *args)
+            elif hasattr(self, cb):
+                if cb is not "run":
+                    # CommandEndpoint already calls run() after every
+                    # callback to instantly update any changed state due
+                    # to the callback's actions.
+                    self.__log_button_event(button, cb, args, "Member callback")
+                    getattr(self, cb)(*args)
+            else:
+                self.__log_button_event(button, cb, args, "External command")
+                if hasattr(self, "data"):
+                    args = [arg.format(**self.data) for arg in args]
+                    cb = cb.format(**self.data)
+                execute(cb + " " + " ".join(args), detach=True)
+        except Exception as e:
+            self.logger.critical("Exception while processing button callback: {!r}".format(e))
 
         # Notify status handler
         try: