Improved detecting loading page for pushing to Archive Today.

oduwsdl · Jan 24, 2021 · 90230c2 · 90230c2
1 parent 17bdfb8
commit 90230c2
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/archivenow/handlers/is_handler.py b/archivenow/handlers/is_handler.py
@@ -50,18 +50,14 @@ def push(self, uri_org, p_args=[], session=requests.Session()):
             # The page takes a while to archive, so keep checking if the loading page is still displayed.
             loading = True
             while loading:
-                try:
-                    loadingPage = driver.find_element_by_xpath("/html/body/div/img")
-
-                except:
+
+                if not 'wip' in driver.current_url:
                     loading = False
 
             # After the loading screen is gone and the page is archived, the current URL
             # will be the URL to the archived page.
             print(driver.current_url)
 
-            msg = driver.current_url
-
             driver.quit()
 
         except: