1 files changed, 100 insertions, 0 deletions
diff --git a/inboxapp-service/app/inboxapp/client.py b/inboxapp-service/app/inboxapp/client.py
new file mode 100644
index 00000000..cf96a498
--- /dev/null
+++ b/inboxapp-service/app/inboxapp/client.py
@@ -0,0 +1,100 @@
+import json
+import urllib2
+import requests
+
+
+class Client:
+
+    INBOX_APP_ROOT = 'http://localhost:5555/n'
+
+    def _get_user(self, account):
+        accounts = json.load(urllib2.urlopen(self.INBOX_APP_ROOT))
+        return [a for a in accounts if a['email_address'] == account][0]
+
+    def __init__(self, account):
+        self.user = self._get_user(account)
+        self.namespace = self.user['namespace']
+
+    def _get(self, append_url):
+        url = "%s/%s/%s" % (self.INBOX_APP_ROOT, self.namespace, append_url)
+        return requests.get(url).json()
+
+    def _post(self, append_url, body):
+        url = "%s/%s/%s" % (self.INBOX_APP_ROOT, self.namespace, append_url)
+        return requests.post(url, json.dumps(body)).json()
+
+    def _put(self, append_url, body):
+        url = "%s/%s/%s" % (self.INBOX_APP_ROOT, self.namespace, append_url)
+        return requests.put(url, json.dumps(body)).json()
+
+    def mails(self, query):
+        url = "messages"
+        if('tags' in query and len(query['tags']) > 0):
+            url = url + "?tag=%s" % ",".join(query['tags'])
+        return self._get(url)
+
+    def drafts(self):
+        return self._get("drafts")
+
+    def mail(self, mail_id):
+        return self._get("messages/%s" % mail_id)
+
+    def thread(self, thread_id):
+        return self._get("threads/%s" % thread_id)
+
+    def mark_as_read(self, mail_id):
+        mail_to_mark = self.mail(mail_id)
+        self._put("messages/%s" % mail_id, {"unread": False})
+        self.remove_tag_from_thread(mail_to_mark["thread"], "unread")
+
+    def tags_for_thread(self, thread):
+        url = "threads/%s" % thread
+        tags = self._get(url)['tags']
+        return [tag['name'] for tag in tags]
+
+    def add_tag_to_thread(self, thread_id, tag):
+        url = "threads/%s" % thread_id
+        response = self._put(url, {'add_tags': [tag]})
+        return response
+
+    def remove_tag_from_thread(self, thread_id, tag):
+        url = "threads/%s" % thread_id
+        response = self._put(url, {'remove_tags': [tag]})
+        return response
+
+    def delete_mail(self, mail_id):
+        thread_id = self.mail(mail_id)['thread']
+        tags = self.tags_for_thread(thread_id)
+        if('trash' in tags):
+            self.add_tag_to_thread(thread_id, 'delete')
+        else:
+            self.add_tag_to_thread(thread_id, 'trash')
+        return None
+
+    def save_draft(self, draft):
+        if 'id' in draft and draft['id']:
+            url = 'drafts/%s' % draft["id"]
+        else:
+            url = "drafts"
+        result = self._post(url, draft)
+        self.mark_as_read(result['id'])
+        return result['id']
+
+    def send_draft(self, draft):
+        new_draft_id = self.save_draft(draft)
+        response = self._post("send", {"draft_id": new_draft_id})
+        return response
+
+    def draft_reply_for(self, mail_id):
+        thread = self.thread(self.mail(mail_id)["thread"])
+        if thread['drafts']:
+            response = self.mail(thread['drafts'][0])
+        else:
+            response = None
+        return response
+
+    def all_tags(self):
+        return self._get("tags")
+
+    def all_contacts(self, query):
+        return self._get("contacts?filter=%s&order_by=rank" % query['general'])