mirror of
https://github.com/Hopiu/linkchecker.git
synced 2026-03-17 06:20:27 +00:00
e9805dbd8a
git-svn-id: https://linkchecker.svn.sourceforge.net/svnroot/linkchecker/trunk/linkchecker@3887 e7d03fd6-7b0d-0410-9947-9c21f3af8025
73 lines
2.5 KiB
Python
73 lines
2.5 KiB
Python
# -*- coding: iso-8859-1 -*-
|
|
# Copyright (C) 2006-2009 Bastian Kleineidam
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation; either version 2 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; if not, write to the Free Software
|
|
# Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
|
|
"""
|
|
Store and retrieve cookies.
|
|
"""
|
|
from .. import log, LOG_CACHE, cookies
|
|
from ..decorators import synchronized
|
|
from ..lock import get_lock
|
|
|
|
|
|
_lock = get_lock("cookie")
|
|
|
|
class CookieJar (object):
|
|
"""
|
|
Cookie storage, implementing the default cookie handling policy for
|
|
LinkChecker.
|
|
"""
|
|
|
|
def __init__ (self):
|
|
self.cache = {}
|
|
|
|
@synchronized(_lock)
|
|
def add (self, headers, scheme, host, path):
|
|
"""
|
|
Parse cookie values, add to cache.
|
|
"""
|
|
jar = set()
|
|
for h in headers.getallmatchingheaders("Set-Cookie"):
|
|
# RFC 2109 (Netscape) cookie type
|
|
try:
|
|
c = cookies.NetscapeCookie(h, scheme, host, path)
|
|
jar.add(c)
|
|
except cookies.CookieError:
|
|
log.debug(LOG_CACHE,
|
|
"Invalid cookie header for %s:%s%s: %r", scheme, host, path, h)
|
|
for h in headers.getallmatchingheaders("Set-Cookie2"):
|
|
# RFC 2965 cookie type
|
|
try:
|
|
c = cookies.Rfc2965Cookie(h, scheme, host, path)
|
|
jar.add(c)
|
|
except cookies.CookieError:
|
|
log.debug(LOG_CACHE,
|
|
"Invalid cookie2 header for %s:%s%s: %r", scheme, host, path, h)
|
|
self.cache[host] = jar
|
|
return jar
|
|
|
|
@synchronized(_lock)
|
|
def get (self, scheme, host, port, path):
|
|
"""
|
|
Cookie cache getter function.
|
|
"""
|
|
log.debug(LOG_CACHE, "Get cookies for host %r path %r", host, path)
|
|
jar = self.cache.setdefault(host, set())
|
|
return [x for x in jar if x.check_expired() and \
|
|
x.is_valid_for(scheme, host, port, path)]
|
|
|
|
@synchronized(_lock)
|
|
def __str__ (self):
|
|
return "<CookieJar with %s>" % self.cache
|