def getPage(self, *args, **kwargs):
"""Open url with debugging support. Return status, headers, body.
Automatically sends cookies; clear manually with self.clearCookies().
Also automatically converts a body passed as a dictionary to url
params, and appends params to a GET request's url as needed."
"""
if 'headers' in kwargs:
kwargs['headers'] += self.cookies
else:
# be sure to copy cookies here so we don't add non-cookies to
# the cookies collection
kwargs['headers'] = self.cookies[:]
if 'body' in kwargs:
body = kwargs['body']
if isinstance(body, dict):
kwargs['body'] = urlencode(body)
if 'method' not in kwargs or kwargs['method'] == 'GET':
body = kwargs['body']
del kwargs['body']
argsList = list(args)
argsList[0] += '?' + body
args = tuple(argsList)
old_cookies = self.cookies # self.cookies is re-set by getPage
result = CPWebCase.getPage(self, *args, **kwargs)
# Re-merge old_cookies into self.cookies, since the server might
# not have re-set all of our cookies
allCookies = [ z
for c,v in self.cookies
for z in v.replace('\r', '').split('\n')
]
cookiesSet = {}
for cookie in allCookies:
name, value = cookie.split(';', 1)[0].split('=', 1)
cookiesSet[name] = value
for (c, v) in old_cookies:
name, value = v.split('=', 1)
cookiesSet.setdefault(name, value)
self.cookies = [
(
'Cookie'
, '; '.join([ '='.join([ n, v ])
for n,v in cookiesSet.items() ])
)
]
return result
def __init__(self, *args, **kwargs):
CPWebCase.__init__(self, *args, **kwargs)
self.clearCookies()