Add __getitem__ to PickleableCookieJar

resources/lib/utils/cookies.py

@@ -44,6 +44,41 @@ def __setstate__(self, state):
  if '_cookies_lock' not in self.__dict__:
  self._cookies_lock = RLock()
 
+ def __getitem__(self, name):
+ """Dict-like __getitem__() for compatibility with client code. Throws
+ exception if there are more than one cookie with name. In that case,
+ use the more explicit get() method instead.
+
+ .. warning:: operation is O(n), not O(1).
+ """
+ return self._find_no_duplicates(name)
+
+ def _find_no_duplicates(self, name, domain=None, path=None):
+ """Both ``__get_item__`` and ``get`` call this function: it's never
+ used elsewhere in Requests.
+
+ :param name: a string containing name of cookie
+ :param domain: (optional) string containing domain of cookie
+ :param path: (optional) string containing path of cookie
+ :raises KeyError: if cookie is not found
+ :raises CookieConflictError: if there are multiple cookies
+ that match name and optionally domain and path
+ :return: cookie.value
+ """
+ to_return = None
+ for cookie in iter(self):
+ if cookie.name == name:
+ if domain is None or cookie.domain == domain:
+ if path is None or cookie.path == path:
+ if to_return is not None:
+ # if there are multiple cookies that meet passed in criteria
+ raise Exception(f"There are multiple cookies with name, {name!r}") # pylint: disable=broad-except
+ # we will eventually return this as long as no cookie conflict
+ to_return = cookie.value
+ if to_return:
+ return to_return
+ raise KeyError(f"name={name!r}, domain={domain!r}, path={path!r}")
+
 def save(cookie_jar, log_output=True):
  """Save a cookie jar to file and in-memory storage"""
  if log_output: