forked from Dpeta/pesterchum-alt-servers
-
Notifications
You must be signed in to change notification settings - Fork 0
/
pytwmn.py
executable file
·77 lines (69 loc) · 2.25 KB
/
pytwmn.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
import os
import socket
class TwmnError(Exception):
UNWN_ERR = -1
NO_TWMND = -2
NO_CONF = -3
def __init__(self, code):
self.code = code
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():
try:
from xdg import BaseDirectory
return os.path.join(BaseDirectory.xdg_config_home,"twmn/twmn.conf")
except:
return False
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():
port = int(line[5:-1])
break
except IOError:
raise TwmnError(TwmnError.NO_CONF)
if type(port) == type(str()):
port = int(port)
global s
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
s.connect((host, port))
class Notification(object):
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>"
"<content>" + self.msg + "</content>"
"<icon>" + self.icon + "</icon></root>")
else:
s.send("<root><title>" + self.title + "</title>"
"<content>" + self.msg + "</content>"
"<icon>" + self.icon + "</icon>"
"<duration>" + str(self.time) + "</duration></root>")
except:
raise TwmnError(TwmnError.NO_TWMND)
if __name__ == "__main__":
init()
n = Notification("PyTwmn", "This is a notification!")
n.set_duration(1000)
n.show()