diff options
| -rwxr-xr-x[-rw-r--r--] | push.py | 16 |
1 files changed, 10 insertions, 6 deletions
@@ -16,8 +16,8 @@ except IndexError: with open(cfg) as fh: maps = yaml.load(fh) -sender_map = maps.get('senders', None) -recipient_map = maps.get('recipients', None) +sender_map = maps.get('senders', []) +recipient_map = maps.get('recipients', []) recipient = os.environ.get('RECIPIENT', '') sender = os.environ.get('SENDER', '') @@ -46,16 +46,20 @@ parser = email.parser.Parser() mail = parser.parse(sys.stdin) title, encoding = email.header.decode_header(mail.get('subject'))[0] -if title != None and encoding != None: +if title is not None and encoding is not None: title = title.decode(encoding).encode('utf-8') payload = mail.get_payload(decode=True) encoding = mail.get_charsets()[0] -if payload != None and encoding != None: +if payload is not None and encoding is not None: payload = payload.decode(encoding).encode('utf-8') msg = payload.strip() -for user_token in user_tokens: - p = pushover.Client(user_token, api_token=api_token) +if type(user_tokens) is list: + for user_token in user_tokens: + p = pushover.Client(user_token, api_token=api_token) + p.send_message(msg, title=title) +else: + p = pushover.Client(user_tokens, api_token=api_token) p.send_message(msg, title=title) |