mirror of
https://github.com/nikdoof/test-auth.git
synced 2025-12-14 14:52:15 +00:00
Move reddit api to api, oops.
This commit is contained in:
67
reddit/api.py
Normal file
67
reddit/api.py
Normal file
@@ -0,0 +1,67 @@
|
||||
import simplejson as json
|
||||
import urllib2
|
||||
import urllib
|
||||
from datetime import datetime
|
||||
|
||||
class Message(dict):
|
||||
""" Reddit Message """
|
||||
|
||||
def __init__(self, dict=None):
|
||||
if dict:
|
||||
self.dictitems = dict
|
||||
|
||||
def __getattr__(self, name):
|
||||
return self.dictitems[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 """
|
||||
|
||||
REDDIT = "http://www.reddit.com"
|
||||
REDDIT_API_LOGIN = "%s/api/login" % REDDIT
|
||||
REDDIT_API_INBOX = "%s/message/inbox/.json?mark=false" % REDDIT
|
||||
|
||||
def __init__(self, username, password):
|
||||
|
||||
self.opener = urllib2.build_opener(urllib2.HTTPCookieProcessor())
|
||||
urllib2.install_opener(self.opener)
|
||||
|
||||
|
||||
data = { 'user': username,
|
||||
'passwd': password,
|
||||
'api_type': 'json' }
|
||||
|
||||
|
||||
url = "%s/%s" % (self.REDDIT_API_LOGIN, username)
|
||||
req = urllib2.Request( url, urllib.urlencode(data))
|
||||
jsondoc = json.load(self.opener.open(req))
|
||||
|
||||
self.login_cookie = jsondoc['json']['data']['cookie']
|
||||
|
||||
@property
|
||||
def _inbox_data(self):
|
||||
|
||||
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 __len__(self):
|
||||
return len(self._inbox_data)
|
||||
|
||||
def __getitem__(self, name):
|
||||
return self._inbox_data[name]
|
||||
|
||||
def __iter__(self):
|
||||
return self._inbox_data.__iter__()
|
||||
|
||||
Reference in New Issue
Block a user