From 830b0eef8936823090f36eb19b0de4cd751a9c2e Mon Sep 17 00:00:00 2001 From: Barbara Miller Date: Fri, 2 Jun 2017 12:45:21 -0700 Subject: [PATCH] undo post-login nav (ARI-5385 and/or ARI-5386) --- brozzler/browser.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/brozzler/browser.py b/brozzler/browser.py index 76036a7..7d0fb0e 100644 --- a/brozzler/browser.py +++ b/brozzler/browser.py @@ -434,6 +434,12 @@ class Browser: user_agent=user_agent, timeout=300) if password: self.try_login(username, password, timeout=300) + # if login redirected us, return to page_url + if page_url != self.url().split('#')[0]: + self.logger.info('login navigated to %s, away from %s; returning!', self.url(), page_url) + self.navigate_to_page( + page_url, extra_headers=extra_headers, + user_agent=user_agent, timeout=300) if on_screenshot: jpeg_bytes = self.screenshot() on_screenshot(jpeg_bytes)