From a3a8936780a1ede00f1c94453654821aaae8b491 Mon Sep 17 00:00:00 2001
From: Michael Fincham <michael@hotplate.co.nz>
Date: Wed, 18 Jan 2017 09:16:06 +1300
Subject: [PATCH] Better error handling

---
 battery | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/battery b/battery
index 872c089..a801fdb 100755
--- a/battery
+++ b/battery
@@ -7,6 +7,8 @@ On e.g. Lenovo laptops with multiple batteries this produces a nice one line sum
 Michael Fincham <michael@hotplate.co.nz> 2017-01-16
 """
 
+import sys
+
 import dbus
 
 class DisplayDevice(object):
@@ -22,9 +24,13 @@ class DisplayDevice(object):
     )
 
     def __init__(self):
-        self.bus = dbus.SystemBus()
-        self.display_device = self.bus.get_object('org.freedesktop.UPower', '/org/freedesktop/UPower/devices/DisplayDevice')
-        
+        try:
+            self.bus = dbus.SystemBus()
+            self.display_device = self.bus.get_object('org.freedesktop.UPower', '/org/freedesktop/UPower/devices/DisplayDevice')
+        except:
+            print("Error connecting to dbus")
+            sys.exit(2)
+            
     def get_property(self, property_name):
         """
         Retrieve a dbus property for the DisplayDevice.
-- 
2.38.5