core.desktop: Refactor into class

TODO: find better way to handle Impl

i3pystatus/core/desktop.py

@@ -1,17 +1,30 @@
 
 
-def display_notification(title, body, icon="dialog-information", urgency=1, timeout=0):
+class DesktopNotification:
     """
-    Displays a desktop notification
+    Class to display a desktop notification
 
     :param title: Title of the notification
     :param body: Body text of the notification, depending on the users system configuration HTML may be used, but is not recommended
     :param icon: A XDG icon name, see http://standards.freedesktop.org/icon-naming-spec/icon-naming-spec-latest.html
     :param urgency: A value between 1 and 3 with 1 meaning low urgency and 3 high urgency.
     :param timeout: Timeout in seconds for the notification. Zero means it needs to be dismissed by the user.
-    :returns: boolean indicating success
     """
-    return False
+
+    def __init__(self, title, body, icon="dialog-information", urgency=1, timeout=0):
+        self.title = title
+        self.body = body
+        self.icon = icon
+        self.urgency = urgency
+        self.timeout = timeout
+
+    def display(self):
+        """
+        Display this notification
+
+        :returns: boolean indicating success
+        """
+        return False
 
 try:
     from gi.repository import Notify
@@ -21,19 +34,20 @@ else:
     if not Notify.init("i3pystatus"):
         raise ImportError("Couldn't initialize libnotify")
 
-    URGENCY_LUT = (
-        Notify.Urgency.LOW,
-        Notify.Urgency.NORMAL,
-        Notify.Urgency.CRITICAL,
-    )
-
     # List of some useful icon names:
     # battery, battery-caution, battery-low
     # …
 
-    def display_notification(title, body, icon="dialog-information", urgency=1, timeout=0):
-        notification = Notify.Notification.new(title, body, icon)
-        if timeout:
-            notification.set_timeout(timeout)
-        notification.set_urgency(URGENCY_LUT[urgency])
-        return notification.show()
+    class DesktopNotification(DesktopNotification):
+        URGENCY_LUT = (
+            Notify.Urgency.LOW,
+            Notify.Urgency.NORMAL,
+            Notify.Urgency.CRITICAL,
+        )
+
+        def display(self):
+            notification = Notify.Notification.new(self.title, self.body, self.icon)
+            if self.timeout:
+                notification.set_timeout(self.timeout)
+            notification.set_urgency(self.URGENCY_LUT[self.urgency])
+            return notification.show()