[py] PEP 484 type hints for selenium.webdriver.common.timeouts (#9524)

Signed-off-by: oleg.hoefling <[email protected]>

Co-authored-by: David Burns <[email protected]>

Author: hoefling

py/selenium/webdriver/common/timeouts.py

@@ -16,9 +16,29 @@
 # under the License.
 
 
+from typing import TYPE_CHECKING
+
+
+if TYPE_CHECKING:
+    import sys
+    if sys.version_info >= (3, 8):
+        from typing import TypedDict
+    else:
+        from typing_extensions import TypedDict
+
+    class JSONTimeouts(TypedDict, total=False):
+        implicit: int
+        pageLoad: int
+        script: int
+
+else:
+    from typing import Dict
+    JSONTimeouts = Dict[str, int]
+
+
 class Timeouts(object):
 
-    def __init__(self, implicit_wait=0, page_load=0, script=0):
+    def __init__(self, implicit_wait: float = 0, page_load: float = 0, script: float = 0) -> None:
         """
         Create a new Timeout object.
 
@@ -35,55 +55,55 @@ def __init__(self, implicit_wait=0, page_load=0, script=0):
         self._script = self._convert(script)
 
     @property
-    def implicit_wait(self):
+    def implicit_wait(self) -> float:
         """
         Return the value for the implicit wait. This does not return the value on the remote end
         """
         return self._implicit_wait / 1000
 
     @implicit_wait.setter
-    def implicit_wait(self, _implicit_wait):
+    def implicit_wait(self, _implicit_wait: float) -> None:
         """
         Sets the value for the implicit wait. This does not set the value on the remote end
         """
         self._implicit_wait = self._convert(_implicit_wait)
 
     @property
-    def page_load(self):
+    def page_load(self) -> float:
         """
         Return the value for the page load wait. This does not return the value on the remote end
         """
         return self._page_load / 1000
 
     @page_load.setter
-    def page_load(self, _page_load):
+    def page_load(self, _page_load: float) -> None:
         """
         Sets the value for the page load wait. This does not set the value on the remote end
         """
         self._page_load = self._convert(_page_load)
 
     @property
-    def script(self):
+    def script(self) -> float:
         """
         Return the value for the script wait. This does not return the value on the remote end
         """
         return self._script / 1000
 
     @script.setter
-    def script(self, _script):
+    def script(self, _script: float) -> None:
         """
         Sets the value for the script wait. This does not set the value on the remote end
         """
         self._script = self._convert(_script)
 
-    def _convert(self, timeout):
+    def _convert(self, timeout: float) -> int:
         if isinstance(timeout, (int, float)):
             return int(float(timeout) * 1000)
         else:
             raise TypeError("Timeouts can only be an int or a float")
 
-    def _to_json(self):
-        timeouts = {}
+    def _to_json(self) -> JSONTimeouts:
+        timeouts: JSONTimeouts = {}
         if self._implicit_wait:
             timeouts["implicit"] = self._implicit_wait
         if self._page_load: