diff --git a/linkedin_matrix/db/upgrade/v08_splat_pickle_data.py b/linkedin_matrix/db/upgrade/v08_splat_pickle_data.py index 6f0479d..e96f251 100644 --- a/linkedin_matrix/db/upgrade/v08_splat_pickle_data.py +++ b/linkedin_matrix/db/upgrade/v08_splat_pickle_data.py @@ -25,8 +25,10 @@ async def upgrade_v8(conn: Connection): cookies = pickle.loads(client_pickle) jsessionid, li_at = None, None for cookies in cookies.values(): - jsessionid = cookies.get("JSESSIONID").value - li_at = cookies.get("li_at").value + if j := cookies.get("JSESSIONID"): + jsessionid = j.value + if li := cookies.get("li_at"): + li_at = li.value if not jsessionid or not li_at: log.warning(f"User {user_id} doesn't have JSESSIONID or li_at")