74 lines
1.7 KiB
Python
74 lines
1.7 KiB
Python
import contextlib
|
|
import http.cookies as cookies
|
|
import logging
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
__all__ = ["CookieJar"]
|
|
|
|
|
|
class CookieJar:
|
|
"""
|
|
Keeps your cookies in a file.
|
|
"""
|
|
|
|
def __init__(self, filename=None):
|
|
self._filename = filename
|
|
self._cookies = cookies.SimpleCookie()
|
|
|
|
if not self._filename:
|
|
logger.warning("Could not load cookies, no filename given.")
|
|
return
|
|
|
|
with contextlib.suppress(FileNotFoundError):
|
|
logger.info(f"Loading cookies from {self._filename!r}")
|
|
with open(self._filename, "r") as f:
|
|
for line in f:
|
|
self._cookies.load(line)
|
|
|
|
def sniff(self):
|
|
"""
|
|
Returns a list of Cookie headers containing all current cookies.
|
|
"""
|
|
|
|
return [morsel.OutputString(attrs=[]) for morsel in self._cookies.values()]
|
|
|
|
def bake(self, cookie_string):
|
|
"""
|
|
Parse cookie and add it to the jar.
|
|
Does not automatically save to the cookie file.
|
|
|
|
Example cookie: "a=bcd; Path=/; Expires=Wed, 24 Jul 2019 14:57:52 GMT; HttpOnly; Secure"
|
|
"""
|
|
|
|
logger.debug(f"Baking cookie: {cookie_string!r}")
|
|
|
|
self._cookies.load(cookie_string)
|
|
|
|
def save(self):
|
|
"""
|
|
Saves all current cookies to the cookie jar file.
|
|
"""
|
|
|
|
if not self._filename:
|
|
logger.warning("Could not save cookies, no filename given.")
|
|
return
|
|
|
|
logger.info(f"Saving cookies to {self._filename!r}")
|
|
|
|
with open(self._filename, "w") as f:
|
|
for morsel in self._cookies.values():
|
|
cookie_string = morsel.OutputString()
|
|
#f.write(f"{cookie_string}\n")
|
|
f.write(cookie_string)
|
|
f.write("\n")
|
|
|
|
def monster(self):
|
|
"""
|
|
Removes all cookies from the cookie jar.
|
|
Does not automatically save to the cookie file.
|
|
"""
|
|
|
|
logger.debug("OMNOMNOM, cookies are all gone!")
|
|
|
|
self._cookies = cookies.SimpleCookie()
|