H.P 琥珀

from urllib2 import build_opener, HTTPCookieProcessor
import copy
from cookielib import CookieJar
from ClientForm import ParseFile

class Browser:
    '''A simple browser supporting filling forms and handling cookies'''
    cj = CookieJar()
    opener = build_opener(HTTPCookieProcessor(cj))

    def __init__(self):
        self.page = u'no page'
        self.url = u'blank'
        self.forms = None

    def go(self, url='blank'):
        '''url should be a str or urllib2.Request object'''
        if url == 'blank':
            self.page = u'welcome to Browser'
        response = self.opener.open(url)
        self.url = response.geturl()
        raw_page = response.read()
        self._dump_file(raw_page, 'feedform')
        self.page = raw_page.decode('utf-8')
        print "=> ", self.url
        self.forms = self.getForms()
        response.close()

    def show(self):
        '''return current page HTML source'''
        return self.page

    def getForms(self):
        '''return all forms of current page.'''
        fp = open('feedform', 'r')
        self.forms = ParseFile(fp, self.url, backwards_compat=False)
        fp.close()
        return self.forms
       
    def setForm(self, position=0, **kargs):
        """Example:
            browser.setForm(0, user='jason', password='secret'"""
       
        for key in kargs.keys():
            self.forms[position][key] = kargs[key]

    def submitForm(self, position=0):
        """Submit the form"""
        self.go(self.forms[position].click())
   
    def _copy_response(self, response):
        """Copy a response-like object"""
        cresponse = copy.copy(response)
       
    def _dump_file(self, html, filename='temp'):
        """Write the self.page into a file object"""
        fp = open(filename, 'w')
        fp.write(html)
        fp.close()