From b037748b5eb212fdb517694a414cd19ba7ca4570 Mon Sep 17 00:00:00 2001 From: Klaus-Uwe Mitterer Date: Sun, 7 Aug 2016 21:59:23 +0200 Subject: [PATCH] Make handler stop automatically after 25 minutes in case it stalls --- handler.py | 15 +++++++++++---- 1 file changed, 11 insertions(+), 4 deletions(-) diff --git a/handler.py b/handler.py index 0b1c18c..088202d 100755 --- a/handler.py +++ b/handler.py @@ -4,7 +4,7 @@ from bs4 import BeautifulSoup from selenium import webdriver from selenium.webdriver.common.keys import Keys from selenium.common.exceptions import NoSuchElementException -import urllib.request, urllib.error, urllib.parse, time, os +import multiprocessing, urllib.request, urllib.error, urllib.parse, time, os import dbtools, phototools, setuptools currentRun = [] @@ -109,11 +109,18 @@ def siteHandler(driver, mode = 0, p = 0, db = dbtools.dbHelper()): if pageHandler(driver, db): siteHandler(driver, mode, p+1, db) +def mainHandler(): + siteHandler(driver, db=db) + siteHandler(driver, 1, db=db) + if __name__ == "__main__": db = dbtools.dbHelper() driver = webdriver.Firefox() if loginHandler(driver): - siteHandler(driver, db=db) - siteHandler(driver, 1, db=db) - + p = multiprocessing.Process(target=mainHandler) + p.start() + p.join(1500) + if p.is_alive(): + p.terminate() + p.join() driver.close()