run selenium as headless mode with argument

Author: joonas-yoon

util.py

@@ -1,7 +1,7 @@
 import os, bs4
 from selenium import webdriver
 
-APP_DIR = os.path.dirname(os.path.dirname("__filename__"))
+APP_DIR = os.path.dirname(os.path.dirname('__filename__'))
 
 
 def get_text(element):
@@ -17,19 +17,19 @@ class Chrome:
     _chrome_options = webdriver.ChromeOptions()
     _driver = None
 
-    def __init__(self):
-        # Uncomment if you DO NOT want GUI
-        # self._chrome_options.add_argument('--headless')
-        # self._chrome_options.add_argument('--no-sandbox')
-        # self._chrome_options.add_argument('--disable-dev-shm-usage')
-        # self._chrome_options.add_argument('--disable-gpu')
+    def __init__(self, headless=False):
+        if headless:
+            self._chrome_options.add_argument('--headless')
+            self._chrome_options.add_argument('--no-sandbox')
+            self._chrome_options.add_argument('--disable-dev-shm-usage')
+            self._chrome_options.add_argument('--disable-gpu')
         self._chrome_options.add_argument('--window-size=1920x1080')
         self._chrome_options.add_argument('Mozilla/5.0 (Macintosh; Intel Mac OS X 10_13_6) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/78.0.3904.108 Safari/537.36')
         self._driver = self.create_driver()
 
     def create_driver(self):
-        driver = webdriver.Chrome(self._chrome_driver, chrome_options=self._chrome_options)
+        driver = webdriver.Chrome(self._chrome_driver, options=self._chrome_options)
         return driver
 
     def get_driver(self):
-        return self._driver
+        return self._driver