Additional Python type hints (#9192)

AlbertoCentonze · AutomatedTester · web-flow · commit 4024861170d2 · 2021-03-01T15:46:42.000Z
Co-authored-by: David Burns &lt;david.burns@theautomatedtester.co.uk&gt;
diff --git a/py/selenium/webdriver/remote/webdriver.py b/py/selenium/webdriver/remote/webdriver.py
@@ -367,7 +367,7 @@ def title(self) -> str:
         resp = self.execute(Command.GET_TITLE)
         return resp['value'] if resp['value'] else ""
 
-    def find_element_by_id(self, id_):
+    def find_element_by_id(self, id_) -> WebElement:
         """Finds an element by id.
 
         :Args:
@@ -387,7 +387,7 @@ def find_element_by_id(self, id_):
         warnings.warn("find_element_by_* commands are deprecated. Please use find_element() instead")
         return self.find_element(by=By.ID, value=id_)
 
-    def find_elements_by_id(self, id_):
+    def find_elements_by_id(self, id_) -> WebElement:
         """
         Finds multiple elements by id.
 
@@ -406,7 +406,7 @@ def find_elements_by_id(self, id_):
         warnings.warn("find_elements_by_* commands are deprecated. Please use find_elements() instead")
         return self.find_elements(by=By.ID, value=id_)
 
-    def find_element_by_xpath(self, xpath):
+    def find_element_by_xpath(self, xpath) -> WebElement:
         """
         Finds an element by xpath.
 
@@ -427,7 +427,7 @@ def find_element_by_xpath(self, xpath):
         warnings.warn("find_element_by_* commands are deprecated. Please use find_element() instead")
         return self.find_element(by=By.XPATH, value=xpath)
 
-    def find_elements_by_xpath(self, xpath):
+    def find_elements_by_xpath(self, xpath) -> WebElement:
         """
         Finds multiple elements by xpath.
 
@@ -446,7 +446,7 @@ def find_elements_by_xpath(self, xpath):
         warnings.warn("find_elements_by_* commands are deprecated. Please use find_elements() instead")
         return self.find_elements(by=By.XPATH, value=xpath)
 
-    def find_element_by_link_text(self, link_text):
+    def find_element_by_link_text(self, link_text) -> WebElement:
         """
         Finds an element by link text.
 
@@ -467,7 +467,7 @@ def find_element_by_link_text(self, link_text):
         warnings.warn("find_element_by_* commands are deprecated. Please use find_element() instead")
         return self.find_element(by=By.LINK_TEXT, value=link_text)
 
-    def find_elements_by_link_text(self, text):
+    def find_elements_by_link_text(self, text) -> WebElement:
         """
         Finds elements by link text.
 
@@ -486,7 +486,7 @@ def find_elements_by_link_text(self, text):
         warnings.warn("find_elements_by_* commands are deprecated. Please use find_elements() instead")
         return self.find_elements(by=By.LINK_TEXT, value=text)
 
-    def find_element_by_partial_link_text(self, link_text):
+    def find_element_by_partial_link_text(self, link_text) -> WebElement:
         """
         Finds an element by a partial match of its link text.
 
@@ -507,7 +507,7 @@ def find_element_by_partial_link_text(self, link_text):
         warnings.warn("find_element_by_* commands are deprecated. Please use find_element() instead")
         return self.find_element(by=By.PARTIAL_LINK_TEXT, value=link_text)
 
-    def find_elements_by_partial_link_text(self, link_text):
+    def find_elements_by_partial_link_text(self, link_text) -> WebElement:
         """
         Finds elements by a partial match of their link text.
 
@@ -526,7 +526,7 @@ def find_elements_by_partial_link_text(self, link_text):
         warnings.warn("find_elements_by_* commands are deprecated. Please use find_elements() instead")
         return self.find_elements(by=By.PARTIAL_LINK_TEXT, value=link_text)
 
-    def find_element_by_name(self, name):
+    def find_element_by_name(self, name) -> WebElement:
         """
         Finds an element by name.
 
@@ -547,7 +547,7 @@ def find_element_by_name(self, name):
         warnings.warn("find_element_by_* commands are deprecated. Please use find_element() instead")
         return self.find_element(by=By.NAME, value=name)
 
-    def find_elements_by_name(self, name):
+    def find_elements_by_name(self, name) -> WebElement:
         """
         Finds elements by name.
 
@@ -566,7 +566,7 @@ def find_elements_by_name(self, name):
         warnings.warn("find_elements_by_* commands are deprecated. Please use find_elements() instead")
         return self.find_elements(by=By.NAME, value=name)
 
-    def find_element_by_tag_name(self, name):
+    def find_element_by_tag_name(self, name) -> WebElement:
         """
         Finds an element by tag name.
 
@@ -587,7 +587,7 @@ def find_element_by_tag_name(self, name):
         warnings.warn("find_element_by_* commands are deprecated. Please use find_element() instead")
         return self.find_element(by=By.TAG_NAME, value=name)
 
-    def find_elements_by_tag_name(self, name):
+    def find_elements_by_tag_name(self, name) -> WebElement:
         """
         Finds elements by tag name.
 
@@ -606,7 +606,7 @@ def find_elements_by_tag_name(self, name):
         warnings.warn("find_elements_by_* commands are deprecated. Please use find_elements() instead")
         return self.find_elements(by=By.TAG_NAME, value=name)
 
-    def find_element_by_class_name(self, name):
+    def find_element_by_class_name(self, name) -> WebElement:
         """
         Finds an element by class name.
 
@@ -627,7 +627,7 @@ def find_element_by_class_name(self, name):
         warnings.warn("find_element_by_* commands are deprecated. Please use find_element() instead")
         return self.find_element(by=By.CLASS_NAME, value=name)
 
-    def find_elements_by_class_name(self, name):
+    def find_elements_by_class_name(self, name) -> WebElement:
         """
         Finds elements by class name.
 
@@ -646,7 +646,7 @@ def find_elements_by_class_name(self, name):
         warnings.warn("find_elements_by_* commands are deprecated. Please use find_elements() instead")
         return self.find_elements(by=By.CLASS_NAME, value=name)
 
-    def find_element_by_css_selector(self, css_selector):
+    def find_element_by_css_selector(self, css_selector) -> WebElement:
         """
         Finds an element by css selector.
 
@@ -667,7 +667,7 @@ def find_element_by_css_selector(self, css_selector):
         warnings.warn("find_element_by_* commands are deprecated. Please use find_element() instead")
         return self.find_element(by=By.CSS_SELECTOR, value=css_selector)
 
-    def find_elements_by_css_selector(self, css_selector):
+    def find_elements_by_css_selector(self, css_selector) -> WebElement:
         """
         Finds elements by css selector.
 
