properly fix #24 using try/catch statement

VainlyStrain · Sep 25, 2021 · ca5fad3 · ca5fad3
1 parent 921cacb
commit ca5fad3
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 9 deletions.
diff --git a/Vailyn b/Vailyn
@@ -40,7 +40,10 @@ import sys
 
 if adv_li:
     import setproctitle
-    import notify2
+    try:
+        import notify2
+    except ModuleNotFoundError:
+        pass
 
 
 # initialize colorama
@@ -67,7 +70,10 @@ if __name__ == "__main__":
 
         if variables.adv_li and DESKTOP_NOTIFY:
             # initialize notifications
-            notify2.init("Vailyn")
+            try:
+                notify2.init("Vailyn")
+            except NameError:
+                pass
 
         # set global variables and determine
         # main method

diff --git a/core/methods/notify.py b/core/methods/notify.py
@@ -26,7 +26,10 @@
 from core.variables import SEPARATOR
 
 if variables.adv_li and DESKTOP_NOTIFY:
-    import notify2
+    try:
+        import notify2
+    except ModuleNotFoundError:
+        pass
 
 
 def notify(message):
@@ -48,10 +51,13 @@ def notify(message):
             "tid.png",
         ])
 
-        notification = notify2.Notification(
-            "Vailyn",
-            message=message,
-            icon=icon,
-        )
+        try:
+            notification = notify2.Notification(
+                "Vailyn",
+                message=message,
+                icon=icon,
+            )
 
-        notification.show()
+            notification.show()
+        except NameError:
+            pass