xenforo.py

#!/usr/bin/env python
 
import httplib, urllib, re

DEBUG = False
VERSION = '0.0.1'

class XenForo(object):
  host      = None
  username  = None
  password  = None
  cookies   = {}
  headers   = {
   'Cache-Control': 'max-age=0',
 'Accept-Language': 'en-US,en;q=0.8',
  'Accept-Charset': 'ISO-8859-1,utf-8;q=0.7,*;q=0.3',
          'Accept': 'application/xml,application/xhtml+xml,' +\
                    'text/html;q=0.9,text/plain;q=0.8,image/png,*/*;q=0.5',
      'User-Agent': 'Mozilla/5.0 '+\
                    '(Macintosh; U; Intel Mac OS X 10_6_7; en-US) '+\
                    'AppleWebKit/534.16 (KHTML, like Gecko) '+\
                    'Chrome/10.0.648.205 Safari/534.16 ' +\
                    'pyXenForo %s' % VERSION,
  }
  
  def __init__(self, username, password, host):
    self.username = username
    self.password = password
    self.host     = host
  
  def _update_cookies(self, cookies):
    if cookies is not None:
      for cookie in cookies.split(';'):
        dset        = cookie.split('=')
        if len(dset) > 1:
          if dset[0] in ['f_user', 'f_session', 'IDstack']:
            self.cookies[dset[0]] = dset[1]
  
  def _logged_in(self, page):
    if page[:100].find('class="Public LoggedOut"') > -1:
      return False
    elif page[:100].find('class="Public LoggedIn"') > -1:
      return True
    else:
      return 'UNKNOWN'
  
  def _get_cookies(self):
    cookies = []
    for cookie in self.cookies:
      cookies.append('%s=%s; ' % (cookie, self.cookies[cookie]))
    return '; '.join(cookies)
  
  def _get(self, loc):
    con           = httplib.HTTPConnection(self.host)
    cookies       = self._get_cookies()
    headers       = self.headers
    if cookies is not '':
      headers['Cookie'] = cookies
    if DEBUG:
      print '\nRequest\n----------\n%s' % headers
    con.request('GET', loc, headers=headers)
    resp          = con.getresponse()
    if DEBUG:
      print '\nResponse\n----------\n%s' % resp.getheaders()
    self._update_cookies(resp.getheader('set-cookie'))
    newloc        = resp.getheader('location')
    if newloc is not None:
      if DEBUG:
        print 'Chasing Referral: %s' % newloc
      page        = self._get(newloc)
    else:
      page        = resp.read()
    return page
  
  def _post(self, loc, formdata):
    con           = httplib.HTTPConnection(self.host)
    cookies       = self._get_cookies()
    headers       = self.headers
    payload       = urllib.urlencode(formdata)
    if cookies is not '':
      headers['Cookie']       = cookies
    headers['Content-Type']   = 'application/x-www-form-urlencoded'
    headers['Content-Length'] = len(payload)
    if DEBUG:
      print '\nRequest\n----------\n%s' % headers
    con.request('POST', loc, headers=headers, body=payload)
    resp          = con.getresponse()
    if DEBUG:
      print '\nResponse\n----------\n%s' % resp.getheaders()
    self._update_cookies(resp.getheader('set-cookie'))
    newloc        = resp.getheader('location')
    if newloc is not None:
      if DEBUG:
        print 'Chasing Referral: %s' % newloc
      page        = self._get(newloc)
    else:
      page        = resp.read()
    return page
  
  def _get_token(self, page):
    token_rex = re.compile(r'\<a href\=\"logout/\?_xfToken\=(.+?[^\"])\"')
    token     = token_rex.findall(page)[0]
    token     = token.replace('%2C',',')
    return token
  
  def _get_resolver(self, page):
    res_rex   = re.compile(r'name\=\"_xfRelativeResolver" value\="(.+[^\"])\"')
    return res_rex.findall(page)[0]
  
  def login(self):
    self._get('/')
    formdata  = {
         'login': self.username,
      'register': 0,
      'password': self.password,
      'remember': 1,
  'cookie_check': 1,
      'redirect': '/',
      '_xfToken': '',
    }
    page  = self._post('/login/login', formdata)
    return self._logged_in(page)
  
  def private_message(self, user, subject, message, locked=False):
    page      = self._get('/conversations/add')
    token     = self._get_token(page)
    resolver  = self._get_resolver(page)
    formdata  = {
               'recipients': user,
                    'title': subject,
             'message_html': message,
      '_xfRelativeResolver': resolver,
      'conversation_locked': int(locked),
                 '_xfToken': token
    }
    page  = self._post('/conversations/insert', formdata)
    return page
  
  def change_title(self, threadid, title):
    threadurl = '/threads/%i/' % threadid
    page = self._get(threadurl)
    token = self._get_token(page)
    formdata = {
    'title': title,
    '_xfToken': token
    }
    page = self._post('%s/save' % threadurl, formdata)
    return page
  
  def change_post(self, postid, messagehtml):
    page = self._get('/posts/%i/edit-inline' % postid)
    token = self._get_token(page)
    resolver = self._get_resolver(page)
    formdata = {
    'message_html': messagehtml,
    '_xfRelativeResolver': resolver,
    'attachment_hash': 'bcf92c353751f0d22f9fa70f92fb8e99',
    'watch_thread': '1',
    'watch_thread_email': '1',
    'watch_thread_state': '1',
    '_xfToken': token
    }
    page = self._post('/posts/%i/save' % postid, formdata)