Fix stub file generation when timedelta is used

Author: aaltat

gen_stub.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 # Copied from: https://github.com/MarketSquare/robotframework-browser/blob/master/Browser/gen_stub.py
 import sys
+from datetime import timedelta
 from pathlib import Path
 from typing import Any
 
@@ -82,7 +83,9 @@ def keyword_line(keyword_arguments, keyword_types, method_name):
                 arg_str = arg_str + f": {arg_type_str}"
             elif isinstance(default_value, str):
                 default_value = f"'{default_value}'"
-            arg_str = arg_str + f" = {default_value}"
+            elif isinstance(default_value, timedelta):
+                default_value = f"timedelta(seconds={default_value.total_seconds()})"
+            arg_str = f"{arg_str} = {default_value}"
         else:
             arg_str = argument
             arg_type_str = get_type_sting_from_argument(arg_str, keyword_types)
@@ -99,7 +102,11 @@ def keyword_line(keyword_arguments, keyword_types, method_name):
 FUNCTION_LIST = get_function_list_from_keywords(SL.get_keyword_names())
 
 
-pyi_boilerplate = """
+pyi_boilerplate = """\
+from datetime import timedelta
+from typing import Any, Optional, Union
+
+
 class SeleniumLibrary:
 """
 pyi_boilerplate_append = """

src/SeleniumLibrary/__init__.pyi

@@ -1,6 +1,9 @@
+from datetime import timedelta
+from typing import Any, Optional, Union
+
 
 class SeleniumLibrary:
-    def __init__(self, timeout = 0:00:05, implicit_wait = 0:00:00, run_on_failure = 'Capture Page Screenshot', screenshot_root_directory: Optional[str] = None, plugins: Optional[str] = None, event_firing_webdriver: Optional[str] = None): ...
+    def __init__(self, timeout = timedelta(seconds=5.0), implicit_wait = timedelta(seconds=0.0), run_on_failure = 'Capture Page Screenshot', screenshot_root_directory: Optional[str] = None, plugins: Optional[str] = None, event_firing_webdriver: Optional[str] = None): ...
     def add_cookie(self, name: str, value: str, path: Optional[str] = None, domain: Optional[str] = None, secure: Optional[bool] = None, expiry: Optional[str] = None): ...
     def add_location_strategy(self, strategy_name: str, strategy_keyword: str, persist: bool = False): ...
     def alert_should_be_present(self, text: str = '', action: str = 'ACCEPT', timeout: Optional[timedelta] = None): ...