Use uppercase instead of lowercase

This commit is contained in:
2025-02-17 11:27:35 -08:00
parent f60db7c1d4
commit a6413348ba

View File

@@ -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)