"""
class to make a webdriver from selenium wait until the page is stable and has no further changes in its page_source
WebDriverWait(browser, 10).until(selenium_wait_reload(3))
"""
import time
class selenium_wait_reload:
    def __init__(self, desired_repeating_sources):
        self.desired_repeating_sources = desired_repeating_sources
        self.repeated_pages = 0
        self.previous_source = None

    def __call__(self, driver):
        while True:
            current_source = driver.page_source
            if current_source == self.previous_source:
                self.repeated_pages = self.repeated_pages +1
                if self.repeated_pages >= self.desired_repeating_sources:
                    return True
            else:
                self.previous_source = current_source
                self.repeated_pages = 0
            time.sleep(0.3)