Running a chrome extension from Selenium WebDriver

问题:

I am aware of how I can load a chrome extension in selenium webdriver. But I am not seeing any posts/blogs which describe how I can run a chrome extension from Selenium.

I need to explicitly make a chrome extension run/make it perform it's function from selenium. For example, I want to clear the cache of Chrome browser using this extension with Selenium Webdriver.

Can I do it in the first place? Or will Selenium WebDriver help me only with loading an extension into the browser instance and leave it there?

回答1:

When a Chrome extension is activated, it is already "running" (its background/event page, at least). There is no API to programatically click on the button.

If you want to use the functionality of an existing extension with little efforts, then I suggest to download the source code of the extension and insert an additional event listener in the extension's source code.

1. Get the extension's source (e.g. using the Chrome extension source viewer aka CRX Viewer).
2. Unpack the zip file.

3. Create a new HTML file, example_name.html, and let it contain:

   <script src="example_name.js"></script>
   

4. Create a new script file, example_name.js, and let it call the extension's functionality, e.g.:

   chrome.runtime.getBackgroundPage(function(bg) {
       // Relevant function at the background page. In your specific example:
       bg.clearCache();
   });
   

5. Add the previous HTML file to web_accessible_resources in the manifest file.

6. Pack the extension again, e.g. using the GUI at chrome://extensions or using

   chrome.exe --pack-extension=directorycontainingextension
   

   After creating directorycontainingextension.crx, load this crx file in Chrome to know the extension ID. If you don't know how to load the crx file in Chrome, just visit https://robwu.nl/crxviewer/, select the crx file, open the F12 developer tools and copy the 32-character string at "Calculated extension ID: [extension ID here]".

   (Starting from ChromeDriver 2.11, you can just zip the extension instead of packing it as a CRX file, and hard-code the extension ID in the manifest file by setting the "key" attribute (this "key" attribute is also printed to the F12 console by the CRX Viewer).)

After modifying the extension, you will have an extension with the same functionality as the original one, plus an additional HTML page. This new HTML page will invoke the extension's functionality when it is opened.

After doing this, "running" the extension is as easy as opening chrome-extension://[EXTENSION ID HERE]/example_name.html in a new tab.

If you don't like these new tabs, then you could also use the chrome.webRequest or chrome.declarativeWebRequest API to intercept custom URLs and activate the desired functionality whenever a page requests this resource. Then, you can just put the URL in an <img> to activate the extension's functionality.

回答2:

The above solution is technically sound seem to be complicated , so I thought another way to do it. Because many times I need a lot of things that are better done manually, authentications, certain cookies, etc

I use folders as profiles, I run:

chrome_options = Options()
chrome_options.add_argument("user-data-dir=selenium") 
driver = webdriver.Chrome(chrome_options=chrome_options)
driver.get("www.google.com")

Then I manually install the Extensions and do the logins I need now every-time I start the Webdriver with that folder everything is there

chrome_options = Options()
chrome_options.add_argument("user-data-dir=selenium") 
driver = webdriver.Chrome(chrome_options=chrome_options)
driver.get("www.google.com") #Now you can see the Extensions and the logins done are present

The advantage is you can use multiple folders with different setting and Extensions without the need to install and uninstall Extensions, change settings, change logins etc