#!/usr/bin/python
"""get lists from tadalist"""
import sys, os, re, pwd
import urllib, urllib2, cookielib
from BeautifulSoup import BeautifulSoup
website = 'jcomeauictx.tadalist.com'

def loginform():
 loginform = 'no login form found'
 page = fetch('http://%s/session/new' % website)
 forms = page.findAll('form')
 for form in forms:
  if form.get('id') and 'login' in form.get('id').lower():
   loginform = form
   break
 return loginform

def makepost(inputs):
 postdata = {}
 for input in inputs:
  postdata[input.get('name') or 'none'] = input.get('value') or ''
 return postdata

def login():
 pattern = '^machine\s+%s\s+login\s+(\S+)\s+password\s+(.*\S)\s*$' % website
 login = filter(re.compile(pattern).match, open(os.path.join(
  pwd.getpwuid(os.geteuid())[5], '.netrc')).readlines())
 login = re.compile(pattern).match(login[-1]).groups()  # use last one in list
 username, password = login
 form = loginform()
 url = 'http://' + website + form.get('action')
 inputs = form.findAll('input')
 postdata = makepost(inputs)
 postdata['username'], postdata['password'] = username, password
 sys.stderr.write('logging in...\n')
 page = post(url, urllib.urlencode(postdata)).prettify()
 if username in page:
  sys.stderr.write('login successful\n')
 else:
  print page
  raise Exception, 'failed to log in; did you set up $HOME/.netrc correctly?'
 return True

def init():
 cookiejar = cookielib.CookieJar()
 opener = urllib2.build_opener(urllib2.HTTPCookieProcessor(cookiejar))
 urllib2.install_opener(opener)
 login()

def fetch(url):
 socket = urllib2.urlopen(url)
 page = BeautifulSoup(socket.read())
 socket.close()
 return page

def post(url, data):
 socket = urllib2.urlopen(url, data)
 page = BeautifulSoup(socket.read())
 socket.close()
 return page

def getlists():
 page = fetch('http://%s/lists' % website)
 atags = page.findAll('a')
 for tag in atags:
  href = tag.get('href')
  if href.startswith('/lists/') and not href.endswith('/new'):
   url = 'http://%s%s/email' % (website, href)
   post(url, '')

init()  # no matter what, we have to login and set up handler first
if __name__ == '__main__':
 getlists()
else:
 sys.stderr.write('ready for testing: %s\n' % dir())