File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments