Merge pull request #164 from jdi-testing/1554-add-logs-for-determinin…

…g-visibility-locators

1554-add-logs-for-determining-visibility-locators

app/selenium_app.py

@@ -6,6 +6,8 @@
 from selenium.webdriver.support import expected_conditions as EC
 from selenium.webdriver.support.ui import WebDriverWait
 
+from app.logger import logger
+
 
 def get_element_id_to_is_displayed_mapping(page_content_str):
     chrome_options = Options()
@@ -22,15 +24,16 @@ def get_element_id_to_is_displayed_mapping(page_content_str):
         wait = WebDriverWait(driver, 10)
         wait.until(EC.presence_of_element_located((By.TAG_NAME, "body")))
 
-        all_elements = driver.find_elements(by=By.XPATH, value='//*')
+        all_elements = driver.find_elements(by=By.XPATH, value="//*")
 
         result = {}
 
         for element in all_elements:
-            element_id = element.get_attribute('jdn-hash')  # got this format \"0000000000000000000000000000\"
+            element_id = element.get_attribute("jdn-hash")  # got this format \"0000000000000000000000000000\"
             element_id = str(element_id).replace('\\"', '')  # need to remove \"
             is_shown = element.is_displayed()
             result[element_id] = is_shown
+            logger.info(f"Element with jdn-hash {element_id} {'Visible' if is_shown else 'Invisible'}")
 
         driver.quit()
         return result

ds_methods/html5_predict.py

@@ -93,6 +93,8 @@ async def html5_predict_elements(body):
         del model
         gc.collect()
         result = results_df[columns_to_publish].to_dict(orient="records")
+
+        logger.info("Determining visibility locators")
         element_id_to_is_displayed_map = get_element_id_to_is_displayed_mapping(document_json)
         for element in result:
             element["is_shown"] = element_id_to_is_displayed_map.get(element["element_id"], None)