From d028dea189dd8c0f8432bde9417e297012d9eb57 Mon Sep 17 00:00:00 2001
From: Mathew Guest <mathewguest@gmail.com>
Date: Sun, 23 Feb 2020 03:21:21 -0700
Subject: [PATCH] added phantom support

---
 wabot/api.py  | 43 ++++++++++++++++++++++++++++++++++---------
 wabot/page.py |  2 --
 2 files changed, 34 insertions(+), 11 deletions(-)

diff --git a/wabot/api.py b/wabot/api.py
index 496482b..a13e3be 100644
--- a/wabot/api.py
+++ b/wabot/api.py
@@ -17,13 +17,22 @@ EXECUTOR_PORT = 4444
 REMOTE_EXECUTOR = 'http://127.0.0.1:%s/wd/hub'
 
 class BrowserProxy:
-    def __init__(self, session_name='webdriver', pickle_filename=PICKLE_FILENAME):
+    def __init__(
+        self,
+        session_name='webdriver',
+        pickle_filename=PICKLE_FILENAME,
+        phantom=False
+    ):
         self._pickle_filename = pickle_filename
         LOGGER.info('creating selenium browser: session name = %s', session_name)
         try:
             driver_type = "remote_chromium2"
             self.driver_type = driver_type
             print('session_name:', session_name)
+            if phantom:
+                pass
+                print('phantom detected')
+                driver_type = "phantomjs"
             self.driver = self.get_driver(driver_type, session_name)
             if not self.driver:
                 LOGGER.error("failed to get selenium webdriver")
@@ -169,7 +178,7 @@ class BrowserProxy:
         # nhsn actively rejects http requests that do not request with an
         # IE user agent. All of these browser instances have to change the
         # user agent in various ways.
-        LOGGER.debug("creating selenium driver: %s" % (browser))
+        LOGGER.debug('creating selenium driver: %s' % (browser))
         # todo(mathew guest) turn USER_AGENT into local variable so that it
         # could be parameterized someday
         # user_agent = USER_AGENT
@@ -178,24 +187,27 @@ class BrowserProxy:
 
         driver = None
 
-        if browser == "chromium2": # Selenium 2 - Chrome
+        if browser == 'chromium2': # Selenium 2 - Chrome
             driver = self._create_driver_chromium2()
 
         elif browser == 'remote_chromium2':
             driver = self._create_driver_remote_chromium2(session_name)
 
-        elif browser == "chromium1": # Selenium 1 - Chrome without working user agent switch
+        elif browser == 'chromium1': # Selenium 1 - Chrome without working user agent switch
             driver = self._create_driver_chromium1()
 
-        elif browser == "firefox2": # Selenium 2 - Firefox
+        elif browser == 'firefox2': # Selenium 2 - Firefox
             driver = self._create_driver_firefox2()
 
-        elif browser == "firefox1": # Selenium 1 - Firefox
+        elif browser == 'firefox1': # Selenium 1 - Firefox
             driver = self._create_driver_firefox1()
+
+        elif browser == 'phantomjs':
+            driver = self._create_driver_phantomjs()
         else:
-            LOGGER.error("an attempt was made to request an "\
-                              "unsupported (by this product) selenium "\
-                              "webdriver; refusing. requested = %s"\
+            LOGGER.error('an attempt was made to request an '\
+                              'unsupported (by this product) selenium '\
+                              'webdriver; refusing. requested = %s'\
                               % (browser))
             
         driver.implicitly_wait(10)
@@ -301,3 +313,16 @@ class BrowserProxy:
             selenium.webdriver.DesiredCapabilities.FIREFOX.copy(), browser_profile = profile )
         return driver
 
+    def _create_driver_phantomjs(self):
+        # Note(MG): Selenium support for PhantomJS has been deprecated, please use headless
+        # driver = selenium.webdriver.PhantomJS()
+        # driver.set_window_size(1280, 960)
+        # return driver
+        opt = selenium.webdriver.chrome.options.Options()
+        opt.add_argument("--user-agent=" + USER_AGENT)
+        opt.add_argument("--kiosk-printing")
+        opt.add_argument("--headless")
+        driver = selenium.webdriver.Chrome(chrome_options = opt)
+        self.driver = driver
+        return driver
+
diff --git a/wabot/page.py b/wabot/page.py
index 9a68f35..dae17c0 100644
--- a/wabot/page.py
+++ b/wabot/page.py
@@ -1,5 +1,3 @@
-# import nhsn_lo.pages
-
 import enum
 import random
 import selenium.webdriver