Fixed the incompatibility between Python 3.7 and 3.9

phuhl · Mar 19, 2021 · 99415b4 · 99415b4
1 parent eb070f6
commit 99415b4
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/notify_send_py/notify_send_py.py b/notify_send_py/notify_send_py.py
@@ -123,13 +123,20 @@ def notify(
             # address = ('localhost', 6000)
             try:
                 with open('/tmp/notify-send.py.address', 'rb') as pidf:
-                    conn = Client(pidf.read())
+                    address = pidf.read()
+                    try:
+                        conn = Client(address)
+                    except ValueError:
+                        conn = Client(address.decode('utf8'))
                     conn.send([n, replaces_process])
                     conn.close()
             except Exception:
                 listener = Listener()
                 with open('/tmp/notify-send.py.address', 'wb') as pidf:
-                    pidf.write(listener.address)
+                    try:
+                        pidf.write(listener.address)
+                    except TypeError:
+                        pidf.write(listener.address.encode('utf8'))
                 replaces_processes = {}
                 n.show()
                 replaces_processes[replaces_process] = n.id