diff --git a/selenium_cookie_extractor_json.py b/selenium_cookie_extractor_json.py index 58c7bef..8b8d822 100644 --- a/selenium_cookie_extractor_json.py +++ b/selenium_cookie_extractor_json.py @@ -33,28 +33,28 @@ input("Press Enter to extract cookies...") all_cookies = driver.get_cookies() # Define the cookie names we want (case-insensitive) -target_names = {"compass", "ssid", "sid", "osid", "hsid"} +target_names = {"COMPASS", "SSID", "SID", "OSID", "HSID"} extracted = {} # For COMPASS cookie, if multiple are present, prefer the one with path == "/" compass_cookie = None for cookie in all_cookies: - name_lower = cookie["name"].lower() - if name_lower in target_names: + name_upper = cookie["name"].upper() + if name_upper in target_names: # Process COMPASS cookie separately to pick the one with path '/' - if name_lower == "compass": + if name_upper == "COMPASS": if cookie.get("path", "") == "/": compass_cookie = cookie # select this cookie elif not compass_cookie: compass_cookie = cookie # take the first if none with '/' are found yet else: # For others, simply store the value - extracted[name_lower] = cookie["value"] + extracted[name_upper] = cookie["value"] # If we found a COMPASS cookie, add it to the extracted dict. if compass_cookie: - extracted["compass"] = compass_cookie["value"] + extracted["COMPASS"] = compass_cookie["value"] # Form JSON object of the extracted cookies. json_data = json.dumps(extracted, indent=2)