Stub fixes

Author: aaltat

gen_stub.py

@@ -106,19 +106,33 @@ def keyword_line(keyword_arguments, keyword_types, method_name):
 from datetime import timedelta
 from typing import Any, Optional, Union
 
+from selenium.webdriver.remote.webdriver import WebDriver
+from selenium.webdriver.remote.webelement import WebElement
 
 class SeleniumLibrary:
 """
 pyi_boilerplate_append = """
     def add_library_components(self, library_components): ...
     def get_keyword_names(self): ...
-    def run_keyword(self, name, args, kwargs=None): ...
-    def get_keyword_arguments(self, name): ...
-    def get_keyword_tags(self, name): ...
-    def get_keyword_documentation(self, name): ...
-    def get_keyword_types(self, name): ...
-    def get_keyword_source(self, keyword_name): ...
-
+    def run_keyword(self, name: str, args: tuple, kwargs: Optional[dict] = None): ...
+    def get_keyword_arguments(self, name: str): ...
+    def get_keyword_tags(self, name: str): ...
+    def get_keyword_documentation(self, name: str): ...
+    def get_keyword_types(self, name: str): ...
+    def get_keyword_source(self, keyword_name: str): ...
+    def failure_occurred(self): ...
+    def register_driver(self, driver: WebDriver, alias: str): ...
+    @property
+    def driver(self) -> WebDriver: ...
+    def find_element(self, locator: str, parent: Optional[WebElement] = None): ...
+    def find_elements(self, locator: str, parent: WebElement = None): ...
+    def _parse_plugins(self, plugins: Any): ...
+    def _parse_plugin_doc(self): ...
+    def _get_intro_documentation(self): ...
+    def _parse_listener(self, event_firing_webdriver: Any): ...
+    def _string_to_modules(self, modules: Any): ...
+    def _store_plugin_keywords(self, plugin): ...
+    def _resolve_screenshot_root_directory(self): ...
 """
 
 INIT_METHOD = KeywordBuilder.build(SL.__init__)
@@ -130,5 +144,5 @@ def get_keyword_source(self, keyword_name): ...
         )
     )
     stub_file.writelines(FUNCTION_LIST)
-    stub_file.write("\n    # methods from PythonLibCore")
+    stub_file.write("\n    # methods from library.")
     stub_file.writelines(pyi_boilerplate_append.splitlines(keepends=True))

src/SeleniumLibrary/__init__.pyi

@@ -1,6 +1,9 @@
 from datetime import timedelta
 from typing import Any, Optional, Union
 
+from selenium.webdriver.remote.webdriver import WebDriver
+from selenium.webdriver.remote.webelement import WebElement
+
 class SeleniumLibrary:
     def __init__(
         self,
@@ -401,12 +404,25 @@ class SeleniumLibrary:
         error: Optional[str] = None,
         limit: Optional[int] = None,
     ): ...
-    # methods from PythonLibCore
+    # methods from library.
     def add_library_components(self, library_components): ...
     def get_keyword_names(self): ...
-    def run_keyword(self, name, args, kwargs=None): ...
-    def get_keyword_arguments(self, name): ...
-    def get_keyword_tags(self, name): ...
-    def get_keyword_documentation(self, name): ...
-    def get_keyword_types(self, name): ...
-    def get_keyword_source(self, keyword_name): ...
+    def run_keyword(self, name: str, args: tuple, kwargs: Optional[dict] = None): ...
+    def get_keyword_arguments(self, name: str): ...
+    def get_keyword_tags(self, name: str): ...
+    def get_keyword_documentation(self, name: str): ...
+    def get_keyword_types(self, name: str): ...
+    def get_keyword_source(self, keyword_name: str): ...
+    def failure_occurred(self): ...
+    def register_driver(self, driver: WebDriver, alias: str): ...
+    @property
+    def driver(self) -> WebDriver: ...
+    def find_element(self, locator: str, parent: Optional[WebElement] = None): ...
+    def find_elements(self, locator: str, parent: WebElement = None): ...
+    def _parse_plugins(self, plugins: Any): ...
+    def _parse_plugin_doc(self): ...
+    def _get_intro_documentation(self): ...
+    def _parse_listener(self, event_firing_webdriver: Any): ...
+    def _string_to_modules(self, modules: Any): ...
+    def _store_plugin_keywords(self, plugin): ...
+    def _resolve_screenshot_root_directory(self): ...