Python class to open and close website using Selenium

Question

Trying to setup Chrome for the open / close a website. Now i can open it, But failed to close it.

Can anyone tell me why? Many thanks!

from selenium import webdriver
from selenium.webdriver.chrome.options import Options

class chromeSetup():
    def __init__(self):
        self.chrome_path    = r'C:\XXXXX\chromedriver.exe'

    def searchWeb(self, url="https://www.google.com.hk/"):
        driver = webdriver.Chrome(self.chrome_path)
        driver.get(url)

    def close(self):
        self.driver.close()

Answer 1

You are not making driver an instance attribute. Change searchWeb method like this:

def searchWeb(self, url="https://www.google.com.hk/"):
    self.driver = webdriver.Chrome(self.chrome_path)
    self.driver.get(url)