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

import sys
import ClientCookie
from urllib import urlencode

# add loader
cookies = ClientCookie.LWPCookieJar()
opener = ClientCookie.build_opener(
        ClientCookie.HTTPCookieProcessor(cookies),
        ClientCookie.HTTPRefererProcessor,
        ClientCookie.HTTPEquivProcessor,
        ClientCookie.HTTPRefreshProcessor,
        ClientCookie.SeekableProcessor)
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/*, */*")]
ClientCookie.install_opener(opener)

# 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 = ClientCookie.urlopen("http://www.linuxforum.net/forum/start_page.php", urldata)

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

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

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

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