Reorganise the file structure into a project tree

app/reddit/api.py

@@ -0,0 +1,141 @@
+import simplejson as json
+import urllib2
+import urllib
+from datetime import datetime
+import unicodedata 
+
+class NotLoggedIn(Exception):
+    pass
+
+class LoginError(Exception):
+    pass
+
+class Comment(dict):
+    """ Abstraction for comment data provided by JSON 
+        Comments can be identifed by Kind = 1 """
+
+
+    def __init__(self, data):
+
+        dict.__init__(self)
+        self['kind'] = 1
+        self['id'] = data['id']
+        self['post'] = data['link_id'][3:]
+        self['body'] = data['body']
+        self['ups'] = data['likes']
+        self['downs'] = data['downs']
+        self['subreddit_id'] = data['subreddit_id']
+        self['subreddit'] = data['subreddit']
+        self['author'] = data['author']
+        self['permalink'] = u'http://reddit.com/comments/%s/c/%s' % (self['post'], self['id'])
+
+    def __getattr__(self, name):
+        return dict.__getitem__(self, name)
+
+    def __unicode__(self):
+        return u'/r/%s - %s' % (self['subreddit'], ['self.author'])
+
+    def __str__(self):
+        return self.__unicode__()
+
+
+class Message(dict):
+    """ Abstract for a Reddit Message """
+
+    def __init__(self, msg=None):
+        if msg:
+            for k in msg.keys():
+                self[k] = msg[k]
+
+    def __getattr__(self, name):
+        return dict.__getitem__(self, name)
+
+    def __unicode__(self):
+        return u"%s: %s" % (self.author, self.subject)
+
+    def __str__(self):
+        return self.__unicode__()
+
+class Inbox():
+    """
+    Reddit Inbox class, accesses a user's inbox and provides a iterable
+    list of messages
+
+    >>> inbox = Inbox(username='testuser', password='testpassword')
+    >>> len(inbox)
+    5
+
+    """
+
+    REDDIT = "http://www.reddit.com"
+    REDDIT_API_LOGIN = "%s/api/login" % REDDIT
+    REDDIT_API_INBOX = "%s/message/inbox/.json?mark=false" % REDDIT
+    REDDIT_API_COMPOSE = "%s/api/compose/" % REDDIT
+
+    def __init__(self, username=None, password=None):
+        if username and password:
+            self.login(username, password)
+
+    def login(self, username, password):
+        data = { 'user': username,
+                 'passwd': password, 
+                 'api_type': 'json' }
+        url = "%s/%s" % (self.REDDIT_API_LOGIN, username)
+
+        jsondoc = json.load(self._url_request(url, data))
+
+        if jsondoc and 'json' in jsondoc:
+            if 'data' in jsondoc['json']:
+                self.login_cookie = jsondoc['json']['data']['cookie']
+                self.modhash = jsondoc['json']['data']['modhash']
+                if self.login_cookie:
+                    return True
+            elif 'errors' in jsondoc['json']:
+                raise LoginError(jsondoc['json']['errors'])
+
+        return False            
+
+    @property
+    def _inbox_data(self):
+
+        if not self.login_cookie:
+            raise NotLoggedIn
+
+        if not hasattr(self, '__inbox_cache'):
+            inbox = json.load(self._opener.open(self.REDDIT_API_INBOX))['data']
+            
+            self.__inbox_cache = []
+            for msg in inbox['children']:
+                self.__inbox_cache.append(Message(msg['data']))
+             
+        return self.__inbox_cache
+
+    def _url_request(self, url, data):
+        if not hasattr(self, '_opener'):
+            self._opener = urllib2.build_opener(urllib2.HTTPCookieProcessor())
+            urllib2.install_opener(self._opener)
+
+        req = urllib2.Request(url, urllib.urlencode(data))
+        return self._opener.open(req)
+
+    def __len__(self):
+        return len(self._inbox_data)
+
+    def __getitem__(self, name):
+        return self._inbox_data[name]
+
+    def __iter__(self):
+        return self._inbox_data.__iter__()
+
+    def send(self, to, subject, text):
+        if not hasattr(self, 'login_cookie') or not self.login_cookie:
+            raise NotLoggedIn
+
+        data = { 'to': to,
+                 'subject': subject.encode('utf-8'),
+                 'text': text.encode('utf-8'), 
+                 'uh': self.modhash,
+                 'thing_id': '' }
+        url = "%s" % (self.REDDIT_API_COMPOSE)
+
+        jsondoc = json.load(self._url_request(url, data))