adding few methods and screenshot path

Author: nareshnavinash

Library/driver.py

@@ -1,5 +1,7 @@
 from selenium import webdriver
+from selenium.common.exceptions import NoSuchElementException
 from selenium.webdriver.remote import webelement
+import os
 from Library.store import Store
 
 
@@ -26,9 +28,35 @@ def get(self, url: str) -> None:
 		self.driver.get(url)
 
 	def find_element(self, by, value) -> webelement.WebElement:
-		return self.driver.find_element(by, value)
+		try:
+			return self.driver.find_element(by, value)
+		except NoSuchElementException:
+			print("Element not found \n\n" + by + "\n" + value)
+		except Exception as e:
+			print("Error in finding the element \n\n" + by + "\n" + value + "\nException: \n" + str(e))
 
-	def quit(self):
-		self.driver.quit()
+	def find_elements(self, by, value) -> list[webelement.WebElement]:
+		try:
+			return self.driver.find_elements(by, value)
+		except NoSuchElementException:
+			print("Element not found \n\n" + by + "\n" + value)
+		except Exception as e:
+			print("Error in finding the element \n\n" + by + "\n" + value + "\nException: \n" + str(e))
+
+	def refresh(self) -> None:
+		self.driver.refresh()
+
+	def execute_script(self, script, locator) -> None:
+		self.driver.execute_script(self, script, locator)
 
+	def current_url(self) -> str:
+		return self.driver.current_url
 
+	def save_screenshot(self) -> bool:
+		ROOT_DIR = os.path.dirname(os.path.abspath(__file__))
+		CONFIG_PATH = os.path.join(ROOT_DIR, 'configuration.conf')
+		self.driver.save_screenshot(self, )
+
+
+	def quit(self):
+		self.driver.quit()

Library/locator.py

@@ -1,11 +1,143 @@
+from selenium.webdriver import ActionChains
+import selenium.webdriver.support.ui as ui
+from selenium.webdriver.remote import webelement
 from Library.store import Store
 
 
 class Locator(Store):
 
     def __init__(self, by: str, value: str) -> None:
+        super().__init__()
         self.by = by
         self.value = value
 
-    def send_keys(self, string) -> None:
-        Store.current_driver.find_element(self.by, self.value).send_keys(string)
+    def send_keys(self, string) -> bool:
+        try:
+            Store.current_driver.find_element(self.by, self.value).send_keys(string)
+        except Exception as e:
+            print("Send keys not worked at \n" + self.by + "\n" + self.value + "\n Exception: \n" + str(e))
+            return False
+        else:
+            return True
+
+    def click(self) -> bool:
+        try:
+            Store.current_driver.find_element(self.by, self.value).click
+        except Exception as e:
+            print("click not worked at \n" + self.by + "\n" + self.value + "\n Exception: \n" + str(e))
+            return False
+        else:
+            return True
+
+    def mouse_over(self) -> bool:
+        hovering_element = Store.current_driver.find_element(self.by, self.value)
+        hover = ActionChains(Store.current_driver).move_to_element(hovering_element)
+        try:
+            hover.perform()
+        except Exception as e:
+            print("mouse_over not worked at \n" + self.by + "\n" + self.value + "\n Exception: \n" + str(e))
+            return False
+        else:
+            return True
+
+    def text(self) -> str:
+        try:
+            txt_value = Store.current_driver.find_element(self.by, self.value).text
+        except Exception as e:
+            print("get text not worked at \n" + self.by + "\n" + self.value + "\n Exception: \n" + str(e))
+            return False
+        else:
+            return txt_value
+
+    def texts(self) -> list:
+        try:
+            arr_text = []
+            elements = Store.current_driver.find_elements(self.by, self.value)
+            for ele in elements:
+                arr_text.append(ele.text)
+        except Exception as e:
+            print("get texts not worked at \n" + self.by + "\n" + self.value + "\n Exception: \n" + str(e))
+            return [False]
+        else:
+            return arr_text
+
+    def is_displayed(self) -> bool:
+        try:
+            return Store.current_driver.find_elements(self.by, self.value).is_displayed()
+        except Exception as e:
+            print("is_displayed not worked at \n" + self.by + "\n" + self.value + "\n Exception: \n" + str(e))
+            return False
+
+    def is_enabled(self) -> bool:
+        try:
+            return Store.current_driver.find_elements(self.by, self.value).is_enabled()
+        except Exception as e:
+            print("is_enabled not worked at \n" + self.by + "\n" + self.value + "\n Exception: \n" + str(e))
+            return False
+
+    def is_displayed_with_wait(self, timeout=10) -> bool:
+        try:
+            wait = ui.WebDriverWait(Store.current_driver, timeout)
+            return wait.until(lambda element: Store.current_driver.find_element(self.by, self.value).is_displayed())
+        except Exception as e:
+            print("is_displayed_with_wait not worked at \n" + self.by + "\n" + self.value + "\n Exception: \n" + str(e))
+            return False
+
+    def click_if_displayed(self) -> bool:
+        try:
+            self.is_displayed_with_wait()
+            var = Store.current_driver.find_element(self.by, self.value).click
+        except Exception as e:
+            print("click if displayed not worked at \n" + self.by + "\n" + self.value + "\n Exception: \n" + str(e))
+            return False
+        else:
+            return True
+
+    def scroll_to_locator(self) -> bool:
+        try:
+            scroll_locator = Store.current_driver.find_element(self.by, self.value)
+            actions = ActionChains(Store.current_driver)
+            actions.move_to_element(scroll_locator)
+            actions.perform()
+        except Exception as e:
+            print("scroll to locator not worked at \n" + self.by + "\n" + self.value + "\n Exception: \n" + str(e))
+            return False
+        else:
+            return True
+
+    def scroll_to_locator_using_js(self) -> bool:
+        try:
+            scroll_locator = Store.current_driver.find_element(self.by, self.value)
+            Store.current_driver.execute_script('arguments[0].scrollIntoView(true);', scroll_locator)
+        except Exception as e:
+            print("scroll to locator using js not worked at \n" + self.by + "\n" + self.value + "\n Exception: \n" + str(e))
+            return False
+        else:
+            return True
+
+    def move_and_click(self) -> bool:
+        try:
+            self.scroll_to_locator()
+            self.mouse_over()
+            self.click()
+        except Exception as e:
+            print("move and click not worked at \n" + self.by + "\n" + self.value + "\n Exception: \n" + str(e))
+            return False
+        else:
+            return True
+
+    def get_element(self) -> webelement.WebElement:
+        try:
+            return Store.current_driver.find_element(self.by, self.value)
+        except Exception as e:
+            print("scroll to locator using js not worked at \n" + self.by + "\n" + self.value + "\n Exception: \n" + str(e))
+
+    def clear_and_send_keys(self, string) -> bool:
+        try:
+            var = Store.current_driver.find_element(self.by, self.value).clear
+            Store.current_driver.find_element(self.by, self.value).send_keys(string)
+        except Exception as e:
+            print("Clear and Send keys not worked at \n" + self.by + "\n" + self.value + "\n Exception: \n" + str(e))
+            return False
+        else:
+            return True

Library/store.py

@@ -1,6 +1,7 @@
 from selenium import webdriver
 from selenium.webdriver.chrome.webdriver import WebDriver
 
+
 class Store:
 	drivers = []
 	current_driver = None

Pages/google.py

@@ -4,7 +4,7 @@
 class GooglePage(GoogleLocator):
 
     def __init__(self):
-        super
+        super().__init__()
 
     @classmethod
     def enter_search_text(cls, string):