12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152 |
- try:
- import Cookie
- except:
- import http.cookies as Cookie
- class SimpleCookieJar(object):
- def __init__(self):
- self.jar = dict()
- def add(self, set_cookie):
- if set_cookie:
- try:
- simpleCookie = Cookie.SimpleCookie(set_cookie)
- except:
- simpleCookie = Cookie.SimpleCookie(set_cookie.encode('ascii', 'ignore'))
- for k, v in simpleCookie.items():
- domain = v.get("domain")
- if domain:
- if not domain.startswith("."):
- domain = "." + domain
- cookie = self.jar.get(domain) if self.jar.get(domain) else Cookie.SimpleCookie()
- cookie.update(simpleCookie)
- self.jar[domain.lower()] = cookie
- def set(self, set_cookie):
- if set_cookie:
- try:
- simpleCookie = Cookie.SimpleCookie(set_cookie)
- except:
- simpleCookie = Cookie.SimpleCookie(set_cookie.encode('ascii', 'ignore'))
- for k, v in simpleCookie.items():
- domain = v.get("domain")
- if domain:
- if not domain.startswith("."):
- domain = "." + domain
- self.jar[domain.lower()] = simpleCookie
- def get(self, host):
- if not host:
- return ""
- cookies = []
- for domain, simpleCookie in self.jar.items():
- host = host.lower()
- if host.endswith(domain) or host == domain[1:]:
- cookies.append(self.jar.get(domain))
- return "; ".join(filter(None, ["%s=%s" % (k, v.value) for cookie in filter(None, sorted(cookies)) for k, v in
- sorted(cookie.items())]))
|