Git-Mirrors/brozzler
1
0
Fork 0
mirror of https://github.com/internetarchive/brozzler.git synced 2025-02-24 08:39:59 -05:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #42 from galgeek/loginAndReloadSeed

Browse Source 
login and reload original url if navigated away
This commit is contained in:
Noah Levitt 2017-06-02 14:03:51 -07:00 committed by GitHub
commit 5e38a9755e
1 changed files with 12 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
brozzler/browser.py
View File

@ -429,11 +429,18 @@ class Browser:
self.websock_thread.on_response = on_response self.websock_thread.on_response = on_response
try: try:
with brozzler.thread_accept_exceptions(): with brozzler.thread_accept_exceptions():
self.navigate_to_page( self.configure_browser(
page_url, extra_headers=extra_headers, extra_headers=extra_headers,
user_agent=user_agent, timeout=300) user_agent=user_agent)
self.navigate_to_page(page_url, timeout=300)
if password: if password:
self.try_login(username, password, timeout=300) self.try_login(username, password, timeout=300)
# if login redirected us, return to page_url
if page_url != self.url().split('#')[0]:
self.logger.debug(
'login navigated away from %s; returning!',
page_url)
self.navigate_to_page(page_url, timeout=300)
if on_screenshot: if on_screenshot:
jpeg_bytes = self.screenshot() jpeg_bytes = self.screenshot()
on_screenshot(jpeg_bytes) on_screenshot(jpeg_bytes)
@ -479,8 +486,7 @@ class Browser:
# run behavior again with short timeout? # run behavior again with short timeout?
# retrieve outlinks again and append to list? # retrieve outlinks again and append to list?
def navigate_to_page( def configure_browser(self, extra_headers=None, user_agent=None):
self, page_url, extra_headers=None, user_agent=None, timeout=300):
headers = extra_headers or {} headers = extra_headers or {}
headers['Accept-Encoding'] = 'identity' headers['Accept-Encoding'] = 'identity'
self.send_to_chrome( self.send_to_chrome(
@ -492,7 +498,7 @@ class Browser:
method='Network.setUserAgentOverride', method='Network.setUserAgentOverride',
params={'userAgent': user_agent}) params={'userAgent': user_agent})
# navigate to the page! def navigate_to_page(self, page_url, timeout=300):
self.logger.info('navigating to page %s', page_url) self.logger.info('navigating to page %s', page_url)
self.websock_thread.got_page_load_event = None self.websock_thread.got_page_load_event = None
self.send_to_chrome(method='Page.navigate', params={'url': page_url}) self.send_to_chrome(method='Page.navigate', params={'url': page_url})