Yes, I was being sarcastic.

1 files changed, 293 insertions, 0 deletions

diff --git a/push.py b/push.py
new file mode 100755
index 0000000..a67d034
--- /dev/null
+++ b/push.py
@@ -0,0 +1,293 @@
+#!/usr/bin/python3
+
+import datetime
+import getpass
+import http.client
+import json
+import os.path
+import ssl
+import subprocess
+import sys
+import time
+import urllib.parse
+import ws4py.client
+import yaml
+
+dev_name = 'test_client'
+cfg_file = 'push.cfg'
+show_stdout = True
+show_kdialog = True
+
+class HTTPError(Exception): pass
+class LoginError(Exception): pass
+class MessageAckError(Exception): pass
+class MessageDeleteError(Exception): pass
+class MessageError(Exception): pass
+class RegisterError(Exception): pass
+class WebSocketError(Exception): pass
+class WebSocketRestart(Exception): pass
+
+class WebSocketClient(ws4py.client.WebSocketBaseClient):
+    def handshake_ok(self):
+        self.send('login:%s:%s' % (self.cfg.deviceid, self.cfg.secret))
+
+    def received_message(self, m):
+        msg = m.data.decode('utf-8')
+
+        if msg == '!': # We got a message
+            msg_handler = MessageHandler(self.cfg)
+            msg_handler.handle_messages()
+        elif msg == 'R': # Restart
+            raise WebSocketRestart()
+        elif msg == 'E': # Error - Requires manual reconnect
+            raise WebSocketError()
+        elif msg == '#': # Keep-alive packet
+            pass
+        else: # Ignore unknown commands
+            pass
+
+    def closed(self, code, reason=None):
+        if code in (1001, 1011): # This might be wrong
+            raise WebSocketRestart()
+
+class Message:
+    def __init__(self, msg):
+        self.acked      = msg.get('acked', 0)
+        self.aid        = msg.get('aid', None)
+        self.app        = msg.get('app', 'Unknown')
+        self.date       = msg.get('date', None)
+        self.html       = msg.get('html', 0)
+        self.icon       = msg.get('icon', None)
+        self.id         = msg.get('id', 0)
+        self.message    = msg.get('message', '')
+        self.priority   = msg.get('priority', 0)
+        self.receipt    = msg.get('receipt', None)
+        self.sound      = msg.get('sound', None)
+        self.title      = msg.get('title', 'Unnamed')
+        self.umid       = msg.get('umid', 0)
+        self.url        = msg.get('url', None)
+        self.url_title  = msg.get('url_title', None)
+
+class MessageHandler:
+    headers = {'Content-Type': 'application/x-www-form-urlencoded'}
+
+    def __init__(self, cfg):
+        self.conn = http.client.HTTPSConnection('api.pushover.net', 443)
+        self.cfg = cfg
+    
+    def __del__(self):
+        self.conn.close()
+
+    def fetch_messages(self):
+        self.conn.request('GET', '/1/messages.json?secret=%s&device_id=%s' % (self.cfg.secret, self.cfg.deviceid))
+        resp = self.conn.getresponse()
+        if resp.status != http.client.OK:
+            raise HTTPError(resp.status)
+
+        data = json.loads(resp.read().decode('utf-8'))
+        if data.get('status', 0) != 1:
+            raise MessageError(data)
+
+        return [Message(msg) for msg in data['messages']] 
+
+    def delete_messages(self, last_msg):
+        params = urllib.parse.urlencode({'secret': self.cfg.secret, 'message': last_msg.id})
+
+        self.conn.request('POST', '/1/devices/%s/update_highest_message.json' % self.cfg.deviceid, params, self.headers)
+        resp = self.conn.getresponse()
+        if resp.status != http.client.OK:
+            raise HTTPError(resp.status)
+
+        data = json.loads(resp.read().decode('utf-8'))
+        if data.get('status', 0) != 1:
+            raise MessageDeleteError(data)
+
+    def handle_messages(self):
+        last_msg = None
+        for message in self.fetch_messages():
+            if last_msg is None or message.id > last_msg.id:
+                last_msg = message
+
+            if message.priority >= 2 and message.acked != 1:
+                self.ack_message(message)
+
+            self.show_message(message)
+
+        if last_msg is not None:
+            self.delete_messages(last_msg)
+
+    def show_message(self, message):
+        if show_stdout:
+            self.show_message_stdout(message)
+
+        if show_kdialog:
+            self.show_message_dialog(message)
+        
+    def show_message_stdout(self, message):
+        ts = datetime.datetime.fromtimestamp(message.date).strftime('%Y-%m-%d %H:%M:%S')
+        msg = '\033[1m[%s - %s] %s\033[0m\n%s' % (ts, message.app, message.title, message.message)
+        
+        if message.url is not None:
+            if message.url_title is not None:
+                msg += '\n[%s] <URL: %s>' % (message.url_title, message.url)
+            else:
+                msg += '\n<URL: %s >' % message.url
+
+        print(msg)
+
+    def show_message_dialog(self, message):
+        msg = message.message
+        title = '[%s] %s' % (message.app, message.title)
+        
+        if message.url is not None:
+            if message.url_title is not None:
+                msg += '\n\n[%s] <URL: %s >' % (message.url_title, message.url)
+            else:
+                msg += '\n\n<URL: %s >' % message.url
+
+        if subprocess.call('kdialog --title "%s" --passivepopup "%s" 10' % (title, msg), shell=True) != 0:
+            print('Failed to show dialog.', file=sys.stderr)
+
+    def ack_message(self, msg):
+        params = urllib.parse.urlencode({'secret': self.cfg.secret})
+
+        self.conn.request('POST', '/1/receipts/%s/acknowledge.json' % msg.receipt, params, self.headers)
+        resp = self.conn.getresponse()
+        if resp.status != http.client.OK:
+            raise HTTPError(resp.status)
+
+        data = json.loads(resp.read().decode('utf-8'))
+        if data.get('status', 0) != 1:
+            raise MessageAckError(data)
+
+class Cfg:
+    def __init__(self, file=None):
+        self.file = file
+        self.secret = None
+        self.deviceid = None
+
+    def load(self, file=None):
+        if file is not None:
+            self.file = file
+
+        try:
+            with open(self.file, 'r') as fh:
+                cfg = yaml.load(fh)
+        except FileNotFoundError:
+            with open(self.file, 'x') as fh:
+                pass
+        else:
+            self.secret = cfg.get('secret')
+            self.deviceid = cfg.get('deviceid')
+
+    def save(self):
+        with open(self.file, 'w') as fh:
+            yaml.dump({'secret': self.secret, 'deviceid': self.deviceid}, fh)
+
+class PushClient:
+    headers = {'Content-Type': 'application/x-www-form-urlencoded'}
+
+    def __init__(self, cfg):
+        self.cfg = cfg
+        self.conn = None
+
+    def get_user_pass(self):
+        print('Visit https://pushover.net/signup to create an account.')
+        username = input('Username: ')
+        password = getpass.getpass()
+        return (username, password)
+
+    def setup_connection(self):
+        self.conn = http.client.HTTPSConnection('api.pushover.net', 443)
+
+    def close_connection(self):
+        self.conn.close()
+        self.conn = None
+
+    def setup(self):
+        self.setup_connection()
+
+        if self.cfg.secret is None:
+            self.cfg.deviceid = None
+
+            (username, password) = self.get_user_pass()
+            self.login(username, password)
+
+        if self.cfg.deviceid is None:
+            self.register_device()
+
+        self.cfg.save()
+
+        if self.conn is not None:
+            self.close_connection()
+
+    def login(self, username, password):
+        params = urllib.parse.urlencode({'email': username, 'password': password})
+
+        conn.request('POST', '/1/users/login.json', params, self.headers)
+        resp = self.conn.getresponse()
+        if resp.status != http.client.OK:
+            raise HTTPError(resp.status)
+
+        data = json.loads(resp.read().decode('utf-8'))
+        if data.get('status', 0) != 1:
+            raise LoginError(data)
+
+        self.cfg.secret = data['secret']
+
+    def register_device(self):
+        params = urllib.parse.urlencode({'secret': secret, 'name': dev_name, 'os': 'O'})
+
+        self.conn.request('POST', '/1/devices.json', params, self.headers)
+        resp = self.conn.getresponse()
+        if resp.status != http.client.OK:
+            raise HTTPError(resp.status)
+
+        data = json.loads(resp.read().decode('utf-8'))
+        if data.get('status', 0) != 1:
+            raise RegisterError(data)
+
+        self.cfg.deviceid = data['id']
+
+    def run(self):
+        if self.cfg.secret is None or self.cfg.deviceid is None:
+            self.setup()
+
+        msg_handler = MessageHandler(self.cfg)
+        msg_handler.handle_messages()
+
+        while True:
+            ws = WebSocketClient('wss://client.pushover.net/push', ssl_options={'cert_reqs': ssl.CERT_NONE})
+            ws.cfg = self.cfg
+            ws.connect()
+
+            try:
+                ws.run()
+            except WebSocketRestart:
+                pass
+            except KeyboardInterrupt: # Handle this gracefully
+                break
+            finally:
+                ws.close()
+
+            for i in reversed(range(1, 6)):
+                print(' %s' % i, flush=True, end='')
+                time.sleep(1)
+            print('\n', end='')
+
+def debug(msg):
+    if enable_debug:
+        ts = datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')
+        print('%s - DEBUG: %s' % (ts, msg), file=sys.stderr)
+
+def main():
+    cfg = Cfg()
+    cfg.load(cfg_file)
+    client = PushClient(cfg)
+    client.run()
+
+if __name__ == '__main__':
+    try:
+        main()
+    except KeyboardInterrupt:
+        pass