From 8ae18c1578d7c191a87228ef6c78a34df0352c0d Mon Sep 17 00:00:00 2001
From: naisanzaa <naisanza@gmail.com>
Date: Tue, 8 Oct 2024 04:27:39 +0800
Subject: [PATCH] selenium: add caching for find_anything

---
 .../integrations/seleniumWrapper/browser.py   | 50 +++++++++++++++----
 1 file changed, 41 insertions(+), 9 deletions(-)

diff --git a/automon/integrations/seleniumWrapper/browser.py b/automon/integrations/seleniumWrapper/browser.py
index d883878a..1f5ac6d8 100644
--- a/automon/integrations/seleniumWrapper/browser.py
+++ b/automon/integrations/seleniumWrapper/browser.py
@@ -40,6 +40,7 @@ def __init__(self, config: SeleniumConfig = None):
         self.autosaved = None
 
         self.logs = {}
+        self.cache = {}
 
     def __repr__(self):
         try:
@@ -524,6 +525,7 @@ def find_anything(
             case_sensitive: bool = False,
             exact_match: bool = False,
             return_first: bool = False,
+            caching: bool = True,
             **kwargs) -> list:
         """fuzzy search through everything
 
@@ -531,7 +533,15 @@ def find_anything(
         find all matches within meta data
         """
         logger.debug(
-            f'find_anything :: {match=} :: {value=} :: {by=} : {case_sensitive=} :: {exact_match=} :: {return_first=} :: {kwargs=}')
+            f'find_anything :: '
+            f'{match=} :: '
+            f'{value=} :: '
+            f'{by=} : {case_sensitive=} :: '
+            f'{exact_match=} :: '
+            f'{return_first=} :: '
+            f'{caching=} :: '
+            f' {kwargs=}'
+        )
 
         by_types = [
             self.by.TAG_NAME,
@@ -553,7 +563,8 @@ def find_anything(
         MATCHED = []
 
         for by_ in by_types:
-            elements = self.find_elements(value=value, by=by_, **kwargs)
+            elements = self.find_elements(value=value, by=by_, caching=caching, **kwargs)
+
             for element in elements:
                 dirs = dir(element)
                 dir_meta = []
@@ -612,11 +623,25 @@ def find_elements(
             self,
             value: str,
             by: selenium.webdriver.common.by.By,
+            caching: bool = True,
             **kwargs) -> list:
         """find elements"""
         logger.debug(f'find_elements :: {self.current_url} :: {value=} :: {by=} :: {kwargs=}')
 
-        find_elements = self.webdriver.find_elements(value=value, by=by, **kwargs)
+        # try caching the elements
+        if caching:
+            if by not in self.cache.keys():
+                self.cache[by] = {}
+                self.cache[by][value] = self.webdriver.find_elements(value=value, by=by, **kwargs)
+
+            elif value not in self.cache[by].keys():
+                self.cache[by][value] = self.webdriver.find_elements(value=value, by=by, **kwargs)
+
+        else:
+            self.cache[by] = {}
+            self.cache[by][value] = self.webdriver.find_elements(value=value, by=by, **kwargs)
+
+        find_elements = self.cache[by][value]
         logger.debug(f'find_elements :: {len(find_elements)} elements found')
 
         logger.info(f'find_elements :: done')
@@ -922,7 +947,7 @@ def wait_for_anything(
 
             logger.debug(
                 f'wait_for_anything :: '
-                f'timeout {timeout_elapsed}/{timeout} :: '
+                f'timeout {timeout_elapsed}/{timeout} sec :: '
                 f'{self.current_url=} :: '
                 f'{value=} :: '
                 f'{by=}'
@@ -947,7 +972,6 @@ def wait_for_anything(
                 logger.error(f'wait_for_anything :: failed :: {error=} :: {match=} :: {value=} :: {by=}')
 
             timeout_elapsed = round(abs(timeout_start - time.time()), 1)
-            logger.debug(f'wait_for_anything :: {timeout_elapsed} seconds elapsed')
 
         raise ElementNotFoundException(f'wait_for_anything :: failed :: {match=} :: {value=} :: {by=}')
 
@@ -966,7 +990,12 @@ def wait_for_element(
         while timeout_elapsed < timeout:
 
             logger.debug(
-                f'wait_for_element :: {f"{timeout_elapsed}/{timeout}"} :: {by=} :: {self.current_url} :: {value=}')
+                f'wait_for_element :: '
+                f'timeout {timeout_elapsed}/{timeout} sec :: '
+                f'{by=} :: '
+                f'{self.current_url} :: '
+                f'{value=}'
+            )
 
             try:
                 find = self.find_element(
@@ -983,7 +1012,6 @@ def wait_for_element(
                 logger.error(f'wait_for_element :: failed :: {error=} :: {value=} :: {by=}')
 
             timeout_elapsed = round(abs(timeout_start - time.time()), 1)
-            logger.debug(f'wait_for_element :: {timeout_elapsed} seconds elapsed')
 
         raise ElementNotFoundException(f'wait_for_element :: failed :: {value=} :: {by=}')
 
@@ -1002,7 +1030,12 @@ def wait_for_elements(
         while timeout_elapsed < timeout:
 
             logger.debug(
-                f'wait_for_element :: {f"{timeout_elapsed}/{timeout}"} :: {by=} :: {self.current_url} :: {value=}')
+                f'wait_for_element :: '
+                f'timeout {timeout_elapsed}/{timeout} sec :: '
+                f'{by=} :: '
+                f'{self.current_url} :: '
+                f'{value=}'
+            )
 
             try:
                 find = self.find_elements(
@@ -1019,7 +1052,6 @@ def wait_for_elements(
                 logger.error(f'wait_for_elements :: failed :: {error=} :: {value=} :: {by=}')
 
             timeout_elapsed = round(abs(timeout_start - time.time()), 1)
-            logger.debug(f'wait_for_elements :: {timeout_elapsed} seconds elapsed')
 
         raise ElementNotFoundException(f'wait_for_elements :: failed :: {value=} :: {by=}')