mirror of
https://github.com/nikdoof/test-auth.git
synced 2025-12-14 14:52:15 +00:00
104 lines
2.8 KiB
Python
104 lines
2.8 KiB
Python
import simplejson as json
|
|
import urllib2
|
|
import urllib
|
|
from datetime import datetime
|
|
|
|
class NotLoggedIn(Exception):
|
|
pass
|
|
|
|
class Message(dict):
|
|
""" Abstract for a 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
|
|
|
|
>>> 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))
|
|
|
|
self.login_cookie = jsondoc['json']['data']['cookie']
|
|
self.modhash = jsondoc['json']['data']['modhash']
|
|
|
|
if self.login_cookie:
|
|
return True
|
|
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 self.login_cookie:
|
|
raise NotLoggedIn
|
|
|
|
data = { 'to': to,
|
|
'subject': subject,
|
|
'text': text,
|
|
'uh': self.modhash,
|
|
'thing_id': '' }
|
|
url = "%s" % (self.REDDIT_API_COMPOSE)
|
|
|
|
jsondoc = json.load(self._url_request(url, data))
|