def get_session(username, password, cookie_path=COOKIE_PATH):
"""Get session, existing or new."""
class USPSAuth(AuthBase): # pylint: disable=too-few-public-methods
"""USPS authorization storage."""
def __init__(self, username, password, cookie_path):
"""Init."""
self.username = username
self.password = password
self.cookie_path = cookie_path
def __call__(self, r):
"""Call is no-op."""
return r
session = requests.session()
session.auth = USPSAuth(username, password, cookie_path)
session.headers.update({'User-Agent': USER_AGENT})
if os.path.exists(cookie_path):
_LOGGER.debug("cookie found at: %s", cookie_path)
session.cookies = _load_cookies(cookie_path)
else:
_login(session)
return session
评论列表
文章目录