Skip to content

Commit c81d47f

Browse files
committed
Do not save WAF cookie, fix dead code
1 parent cccc1f0 commit c81d47f

1 file changed

Lines changed: 10 additions & 2 deletions

File tree

pytr/api.py

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -286,7 +286,12 @@ def complete_weblogin(self, verify_code):
286286
def save_websession(self):
287287
# Saves session cookies too (expirydate=0).
288288
if self._save_cookies:
289-
self._websession.cookies.save(ignore_discard=True, ignore_expires=True)
289+
# Save a copy without the WAF token - it's fetched fresh on every startup
290+
save_jar = MozillaCookieJar(self._cookies_file)
291+
for cookie in self._websession.cookies:
292+
if cookie.name != "aws-waf-token":
293+
save_jar.set_cookie(cookie)
294+
save_jar.save(ignore_discard=True, ignore_expires=True)
290295

291296
def resume_websession(self):
292297
"""
@@ -301,11 +306,14 @@ def resume_websession(self):
301306
# Loads session cookies too (expirydate=0).
302307
self._websession.cookies.load(ignore_discard=True, ignore_expires=True)
303308
self._weblogin = True
309+
# Re-apply fresh WAF token over any stale one from the cookie file
310+
if self._waf_token:
311+
self._set_waf_cookie(self._waf_token)
304312
try:
305313
self.settings()
306314
except requests.exceptions.HTTPError:
307-
return False
308315
self._weblogin = False
316+
return False
309317
else:
310318
return True
311319
return False

0 commit comments

Comments
 (0)