Merge pull request #35 from joao-voltarelli/extensions_support

ENH: added support to be able to use extensions in browsers

Author: hhslepicka

botcity/web/bot.py

@@ -1120,6 +1120,20 @@ def leave_iframe(self):
         """
         self._driver.switch_to.default_content()
 
+    def install_firefox_extension(self, extension):
+        """
+        Install an extension in the Firefox browser.
+        This will start the browser if it was not started yet.
+
+        Args:
+            extension (str): The path of the .xpi extension to be loaded.
+        """
+        if self.browser != Browser.FIREFOX:
+            raise ValueError("install_firefox_extension only works with Firefox.")
+        if not self._driver:
+            self.start_browser()
+        self._driver.install_addon(os.path.abspath(extension))
+
     #######
     # Mouse
     #######

botcity/web/browsers/chrome.py

@@ -32,7 +32,6 @@ def default_options(headless=False, download_folder_path=None, user_data_dir=Non
     chrome_options.add_argument("--disable-background-timer-throttling")
     chrome_options.add_argument("--disable-client-side-phishing-detection")
     chrome_options.add_argument("--disable-default-apps")
-    chrome_options.add_argument("--disable-extensions")
     chrome_options.add_argument("--disable-hang-monitor")
     chrome_options.add_argument("--disable-popup-blocking")
     chrome_options.add_argument("--disable-prompt-on-repost")

botcity/web/browsers/edge.py

@@ -32,7 +32,6 @@ def default_options(headless=False, download_folder_path=None, user_data_dir=Non
     edge_options.add_argument("--disable-background-timer-throttling")
     edge_options.add_argument("--disable-client-side-phishing-detection")
     edge_options.add_argument("--disable-default-apps")
-    edge_options.add_argument("--disable-extensions")
     edge_options.add_argument("--disable-hang-monitor")
     edge_options.add_argument("--disable-popup-blocking")
     edge_options.add_argument("--disable-prompt-on-repost")