Search code examples
pythonseleniumfirefoxgeckodriverselenium-firefoxdriver

Selenium using too much RAM with Firefox


I am using selenium with Firefox to automate some tasks on Instagram. It basically goes back and forth between user profiles and notifications page and does tasks based on what it finds.

It has one infinite loop that makes sure that the task keeps on going. I have sleep() function every few steps but the memory usage keeps increasing. I have something like this in Python:

while(True):
    expected_conditions()
    ...doTask()
    driver.back()
    expected_conditions()
    ...doAnotherTask()
    driver.forward()
    expected_conditions()

I never close the driver because that will slow down the program by a lot as it has a lot of queries to process. Is there any way to keep the memory usage from increasing overtime without closing or quitting the driver?

EDIT: Added explicit conditions but that did not help either. I am using headless mode of Firefox.


Solution

  • To start with Selenium have very little control over the amount of RAM used by Firefox. As you mentioned the Browser Client i.e. Mozilla goes back and forth between user profiles and notifications page on Instagram and does tasks based on what it finds is too broad as a single usecase. So, the first and foremost task would be to break up the infinite loop pertaining to your usecase into smaller Tests.


    time.sleep()

    Inducing time.sleep() virtually puts a blanket over the underlying issue. However while using Selenium and WebDriver to execute tests through your Automation Framework, using time.sleep() without any specific condition defeats the purpose of automation and should be avoided at any cost. As per the documentation:

    time.sleep(secs) suspends the execution of the current thread for the given number of seconds. The argument may be a floating point number to indicate a more precise sleep time. The actual suspension time may be less than that requested because any caught signal will terminate the sleep() following execution of that signal’s catching routine. Also, the suspension time may be longer than requested by an arbitrary amount because of the scheduling of other activity in the system.

    You can find a detailed discussion in How to sleep webdriver in python for milliseconds


    Analysis

    There were previous instances when Firefox consumed about 80% of the RAM.

    Firefox_RAM

    However as per this discussion some of the users feels that the more memory is used the better because it means you don't have RAM wasted. Firefox uses RAM to make its processes faster since application data is transferred much faster in RAM.


    Solution

    You can implement either/all of the generic/specific steps as follows:

    • Upgrade Selenium to current levels Version 3.141.59.
    • Upgrade GeckoDriver to GeckoDriver v0.24.0 level.
    • Upgrade Firefox version to Firefox v65.0.2 levels.
    • Clean your Project Workspace through your IDE and Rebuild your project with required dependencies only.
    • If your base Web Client version is too old, then uninstall it and install a recent GA and released version of Web Client.
    • Some extensions allow you to block such unnecessary content, as an example:

      • uBlock Origin allows you to hide ads on websites.
      • NoScript allows you to selectively enable and disable all scripts running on websites.
      • To open the Firefox client with an extension you can download the extension i.e. the XPI file from https://addons.mozilla.org and use the add_extension(extension='webdriver.xpi') method to add the extension in a FirefoxProfile as follows:

        from selenium import webdriver
        
        profile = webdriver.FirefoxProfile() 
        profile.add_extension(extension='extension_name.xpi')
        driver = webdriver.Firefox(firefox_profile=profile, executable_path=r'C:\path\to\geckodriver.exe') 
        
    • If your Tests doesn't requires the CSS you can disable the CSS following the this discussion.