From 9157c7071f9900c2608f5ca40ae4f518ed373b96 Mon Sep 17 00:00:00 2001 From: Dave Hunt Date: Fri, 21 Oct 2016 10:49:46 +0100 Subject: [PATCH] [py] Only attempt to remove the profile path if one was specified --- py/selenium/webdriver/firefox/webdriver.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/py/selenium/webdriver/firefox/webdriver.py b/py/selenium/webdriver/firefox/webdriver.py index 65ec514f39796..ab4a9b3d90601 100644 --- a/py/selenium/webdriver/firefox/webdriver.py +++ b/py/selenium/webdriver/firefox/webdriver.py @@ -186,12 +186,13 @@ def quit(self): self.service.stop() else: self.binary.kill() - try: - shutil.rmtree(self.profile.path) - if self.profile.tempfolder is not None: - shutil.rmtree(self.profile.tempfolder) - except Exception as e: - print(str(e)) + if self.profile is not None: + try: + shutil.rmtree(self.profile.path) + if self.profile.tempfolder is not None: + shutil.rmtree(self.profile.tempfolder) + except Exception as e: + print(str(e)) @property def firefox_profile(self):