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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
|
#!/usr/local/bin/python
import os
import re
import sys
import yaml
import pushover
import email.header
import email.parser
try:
# Use argv[1] as path to config file if specified
cfg = sys.argv[1]
except IndexError:
# Fall back to push.cfg in the same direcetory
cfg = os.path.dirname(os.path.realpath(__file__)) + '/push.cfg'
with open(cfg, 'r') as fh:
maps = yaml.load(fh)
sender_map = maps.get('senders', [])
recipient_map = maps.get('recipients', [])
try:
recipient = os.environ['RECIPIENT']
except KeyError:
print >>sys.stderr, "Recipient missing."
sys.exit(1)
try:
sender = os.environ['SENDER']
except KeyError:
print >>sys.stderr, "Sender missing."
sys.exit(1)
api_token = None
user_tokens = None
# Select api key from map based on sender
for s in sender_map:
try:
if re.match(s['re'], sender):
try:
api_token = s['key']
except KeyError:
print >>sys.stderr, "Found no key for sender."
sys.exit(1)
else:
break
except KeyError:
next
else:
print >>sys.stderr, "Found no matching sender."
sys.exit(1)
# Select user key(s) from map based on sender
for r in recipient_map:
try:
if re.match(r['re'], recipient):
try:
user_tokens = r['keys']
except KeyError:
print >>sys.stderr, "Found no key(s) for recipient."
sys.exit(1)
else:
break
except KeyError:
next
else:
print >>sys.stderr, "Found no matching receiver."
sys.exit(1)
# Read an email from stdin
parser = email.parser.Parser()
mail = parser.parse(sys.stdin)
# Try to get the subject from the email and recode to utf-8 if possible
title, encoding = email.header.decode_header(mail.get('subject'))[0]
try:
title = title.decode(encoding).encode('utf-8')
except (NameError, TypeError):
pass
# Lets at least try to handle mime multipart
if mail.is_multipart():
# Try to find something plain text
for current in mail.walk():
if current.get_content_type() == 'text/plain':
payload = get_payload(decode=True)
break
else:
# A realy stupid fallback
payload = mail.get_payload(0)
else:
# Hurray no multipart
payload = mail.get_payload(decode=True)
# Recode payload to utf-8 if possible
try:
encoding = payload.get_charsets()[0]
payload = payload.decode(encoding).encode('utf-8')
except (NameError, TypeError, AttributeError):
pass
msg = payload.strip()
# Check for multiple user keys
if type(user_tokens) is list:
# Send notification to each user key
for user_token in user_tokens:
p = pushover.Client(user_token, api_token=api_token)
p.send_message(msg, title=title)
else:
# Send notification to the only user key
p = pushover.Client(user_tokens, api_token=api_token)
p.send_message(msg, title=title)
|
