diff --git a/scrape.py b/scrape.py index 6b04b47..0912b63 100644 --- a/scrape.py +++ b/scrape.py @@ -50,8 +50,9 @@ def get_index(self): driver.get(service_manual_index) time.sleep(10) - window1 = driver.window_handles[1] - driver.switch_to.window(window1) + if len(driver.window_handles) > 1: + window1 = driver.window_handles[-1] + driver.switch_to.window(window1) source = driver.find_element(By.TAG_NAME, "html").get_attribute("outerHTML") append_upcoming_and_img_urls(source) os.makedirs(os.path.dirname('docs/index.html'), exist_ok=True)