Headless Bool

mjdaoudi · Nov 13, 2023 · 23cfca0 · 23cfca0
1 parent 22b1c6e
commit 23cfca0
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/main.py b/main.py
@@ -32,7 +32,9 @@ def main() -> None:
     for region, business_type in product(regions, business_types):
         logging.info(f"Scrapping {business_type} in {region}")
         try:
-            Scrapper = WebDriver(CHROME_DRIVER_PATH=os.path.abspath(CHROME_DRIVER_PATH))
+            Scrapper = WebDriver(
+                CHROME_DRIVER_PATH=os.path.abspath(CHROME_DRIVER_PATH), 
+                headless=False)
             data = Scrapper.scrape(url, f"{business_type} in {region}")
             data = list(
                 map(

diff --git a/src/Scrapper.py b/src/Scrapper.py
@@ -18,10 +18,11 @@
 class WebDriver:
     adresses = []
 
-    def __init__(self, CHROME_DRIVER_PATH: str):
+    def __init__(self, CHROME_DRIVER_PATH: str, headless:bool=True):
         self.PATH = CHROME_DRIVER_PATH
         self.options = Options()
-        self.options.add_argument("--headless")
+        if headless:
+            self.options.add_argument("--headless")
         self.service = Service(executable_path=self.PATH)
         self.driver = webdriver.Chrome(service=self.service, options=self.options)