[schema] Add an API for sending/receiving messages.

(imported from commit 209d525dc5892fc4c392a8ced1588c838cbb17c4)

api/common.py

@@ -0,0 +1,55 @@
+#!/usr/bin/python
+import mechanize
+import urllib
+import simplejson
+from urllib2 import HTTPError
+import time
+
+class HumbugAPI():
+    def __init__(self, email, api_key, verbose=False, site="https://app.humbughq.com"):
+        self.browser = mechanize.Browser()
+        self.browser.set_handle_robots(False)
+        self.browser.add_password("https://app.humbughq.com/", "tabbott", "xxxxxxxxxxxxxxxxx", "wiki")
+        self.api_key = api_key
+        self.email = email
+        self.verbose = verbose
+        self.base_url = site
+
+    def send_message(self, submit_hash):
+        submit_hash["email"] = self.email
+        submit_hash["api-key"] = self.api_key
+        submit_data = urllib.urlencode([(k, v.encode('utf-8')) for k,v in submit_hash.items()])
+        res = self.browser.open(self.base_url + "/api/v1/send_message", submit_data)
+        return simplejson.loads(res.read())
+
+    def get_messages(self, last_received = None):
+        submit_hash = {}
+        submit_hash["email"] = self.email
+        submit_hash["api-key"] = self.api_key
+        if last_received is not None:
+            submit_hash["first"] = "0"
+            submit_hash["last"] = str(last_received)
+        submit_data = urllib.urlencode([(k, v.encode('utf-8')) for k,v in submit_hash.items()])
+        res = self.browser.open(self.base_url + "/api/v1/get_updates", submit_data)
+        return simplejson.loads(res.read())['zephyrs']
+
+    def call_on_each_message(self, callback):
+        max_message_id = None
+        while True:
+            try:
+                messages = self.get_messages(max_message_id)
+            except HTTPError, e:
+                # 502/503 typically means the server was restarted; sleep
+                # a bit, then try again
+                if self.verbose:
+                    print "HTTP Error getting zephyrs; trying again soon."
+                    print e
+                time.sleep(1)
+            except Exception, e:
+                # For other errors, just try again
+                print e
+                time.sleep(2)
+                continue
+            for message in sorted(messages, key=lambda x: x["id"]):
+                max_message_id = max(max_message_id, message["id"])
+                callback(message)