pesterchum/pytwmn.py

91 lines
2.4 KiB
Python
Raw Normal View History

import os
import socket
2021-03-25 03:49:27 -04:00
2021-03-25 03:49:27 -04:00
class TwmnError(Exception):
UNWN_ERR = -1
NO_TWMND = -2
NO_CONF = -3
2021-03-25 03:49:27 -04:00
def __init__(self, code):
self.code = code
2021-03-25 03:49:27 -04:00
def __str__(self):
if self.code == TwmnError.NO_TWMND:
return "Unable to connect to twmnd"
elif self.code == TwmnError.NO_CONF:
return "Could not find twmn configuration file"
else:
return "Unknown twmn error"
def confExists():
# FIXME
2021-03-25 03:49:27 -04:00
try:
from xdg import BaseDirectory
except ImportError:
return False
try:
return os.path.join(BaseDirectory.xdg_config_home, "twmn/twmn.conf")
2021-03-25 03:49:27 -04:00
except:
return False
2023-02-12 19:34:13 -05:00
2021-03-25 03:49:27 -04:00
def init(host="127.0.0.1", port=None):
if not port:
port = 9797
try:
fn = confExists()
if not fn:
return False
with open(fn) as f:
for line in f.readlines():
if line.startswith("port=") and line[5:-1].isdigit():
2021-03-25 03:49:27 -04:00
port = int(line[5:-1])
break
except OSError:
2021-03-25 03:49:27 -04:00
raise TwmnError(TwmnError.NO_CONF)
if type(port) == type(""):
2021-03-25 03:49:27 -04:00
port = int(port)
global s
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
s.connect((host, port))
class Notification:
2021-03-25 03:49:27 -04:00
def __init__(self, title="", msg="", icon=""):
self.title = str(title)
self.msg = str(msg)
if icon.startswith("file://"):
icon = icon[7:]
self.icon = icon
self.time = None
def set_duration(self, time):
self.time = time
def show(self):
try:
if self.time is None:
s.send(
"<root><title>" + self.title + "</title>"
2021-03-25 03:49:27 -04:00
"<content>" + self.msg + "</content>"
"<icon>" + self.icon + "</icon></root>"
)
2021-03-25 03:49:27 -04:00
else:
s.send(
"<root><title>" + self.title + "</title>"
2021-03-25 03:49:27 -04:00
"<content>" + self.msg + "</content>"
"<icon>" + self.icon + "</icon>"
"<duration>" + str(self.time) + "</duration></root>"
)
2021-03-25 03:49:27 -04:00
except:
raise TwmnError(TwmnError.NO_TWMND)
2021-03-25 03:49:27 -04:00
if __name__ == "__main__":
init()
n = Notification("PyTwmn", "This is a notification!")
n.set_duration(1000)
n.show()