#!/usr/bin/env python
# -*- coding: GB2312 -*-
# xyb at linuxforum.net

import sys
from urllib import urlencode
import cookielib, urllib2
cj = cookielib.LWPCookieJar()
opener = urllib2.build_opener(urllib2.HTTPCookieProcessor(cj))
urllib2.install_opener(opener)
opener.addheaders = [
        ("User-agent", "Mozilla/5.0 (X11; U; Linux i686; en-US; rv:1.5) Gecko/20031107 Debian/1.5-3"),
        ("Accept", "text/html, image/jpeg, image/png, text/*, image/*, */*")]

# check args
if len(sys.argv) <= 2:
    print "Usage: %s USERNAME PASSWORD" % (sys.argv[0])
    sys.exit(1)
else:
    UserName = sys.argv[1]
    Password = sys.argv[2]

# login
data = {
        'Loginname': UserName,
        'Loginpass': Password,
        'firstlogin': 1,
        'option': 'µÇÈëÂÛÌ³'
        }
urldata = urlencode(data)
r = opener.open("http://www.linuxforum.net/forum/start_page.php", urldata)

# show result
results = r.read()
open('start.html', 'w').write(results)

# then, user page
r = urllib2.urlopen("http://www.linuxforum.net/forum/login.php?Cat=")
results = r.read()
open('user.html', 'w').write(results)

# read messages
r = urllib2.urlopen("http://www.linuxforum.net/forum/viewmessages.php?Cat=&box=received")
results = r.read()
open('inbox.html', 'w').write(results)

# save cookies to file
cj.save("./cookies")
