diff --git a/handler.py b/handler.py index cbf5c12..58de016 100755 --- a/handler.py +++ b/handler.py @@ -136,13 +136,15 @@ if __name__ == "__main__": db = dbtools.dbHelper() driver = webdriver.Firefox() ret = {"rec": 0,"snt": 0} - if loginHandler(driver): - p = multiprocessing.Process(target=mainHandler, args=([ret])) - p.start() - p.join(1500) - if p.is_alive(): - p.terminate() - p.join() - driver.close() - print("Received", ret["rec"], "messages") - print("Sent", ret["snt"], "messages") + try: + if loginHandler(driver): + p = multiprocessing.Process(target=mainHandler, args=([ret])) + p.start() + p.join(1500) + if p.is_alive(): + p.terminate() + p.join() + finally: + driver.close() + print("Received", ret["rec"], "messages") + print("Sent", ret["snt"], "messages")