backup/pywb
1
0
Fork 0
mirror of https://github.com/webrecorder/pywb.git synced 2025-03-24 06:59:52 +01:00
Code Issues Projects Releases Activity

cookie_rewriter: catch CookieError and ignore erroring cookies

Browse Source
This commit is contained in:
Ilya Kreymer 2014-05-15 22:37:08 -07:00
parent 1d8c68b745
commit 5285723ccf
2 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
pywb/rewrite/cookie_rewriter.py
View File

@ -1,4 +1,4 @@
import Cookie from Cookie import SimpleCookie, CookieError
#================================================================= #=================================================================
@ -11,10 +11,12 @@ class WbUrlCookieRewriter(object):
self.url_rewriter = url_rewriter self.url_rewriter = url_rewriter
def rewrite(self, cookie_str, header='Set-Cookie'): def rewrite(self, cookie_str, header='Set-Cookie'):
cookie = Cookie.SimpleCookie()
cookie.load(cookie_str)
results = [] results = []
cookie = SimpleCookie()
try:
cookie.load(cookie_str)
except CookieError:
return results
for name, morsel in cookie.iteritems(): for name, morsel in cookie.iteritems():
if morsel.get('domain'): if morsel.get('domain'):

4
pywb/rewrite/test/test_cookie_rewriter.py
View File

@ -12,6 +12,10 @@ r"""
>>> rewrite_cookie('abc=def; Path=file.html; Expires=Wed, 13 Jan 2021 22:23:01 GMT') >>> rewrite_cookie('abc=def; Path=file.html; Expires=Wed, 13 Jan 2021 22:23:01 GMT')
[('Set-Cookie', 'abc=def; Path=/pywb/20131226101010/http://example.com/some/path/file.html')] [('Set-Cookie', 'abc=def; Path=/pywb/20131226101010/http://example.com/some/path/file.html')]
# Cookie with invalid chars, not parsed
>>> rewrite_cookie('abc@def=123')
[]
""" """