osn-selenium 0.0.0__py3-none-any.whl → 1.1.0__py3-none-any.whl

osn_selenium/webdrivers/unified/core/window.py

@@ -0,0 +1,135 @@
+from typing import (
+	Any,
+	List,
+	Literal,
+	Optional,
+	Union
+)
+from osn_selenium.webdrivers._decorators import requires_driver
+from selenium.common.exceptions import (
+	InvalidSessionIdException
+)
+from osn_selenium.webdrivers.unified.core.base import UnifiedCoreBaseMixin
+from osn_selenium.exceptions.window import (
+	InvalidWindowHandleError,
+	InvalidWindowIndexError,
+	NoWindowHandlesFoundError
+)
+
+
+__all__ = ["UnifiedCoreWindowMixin"]
+
+
+class UnifiedCoreWindowMixin(UnifiedCoreBaseMixin):
+	@requires_driver
+	def _window_handles_impl(self) -> List[str]:
+		return self._driver_impl.window_handles
+	
+	@requires_driver
+	def _close_impl(self) -> None:
+		self._driver_impl.close()
+	
+	@requires_driver
+	def _switch_to_impl(self) -> Any:
+		return self._driver_impl.switch_to
+	
+	@requires_driver
+	def _current_window_handle_impl(self) -> str:
+		return self._driver_impl.current_window_handle
+	
+	def _get_window_handle_impl(self, window: Optional[Union[str, int]] = None) -> str:
+		handles = self._window_handles_impl()
+		
+		if not handles:
+			raise NoWindowHandlesFoundError()
+		
+		if isinstance(window, str):
+			if window not in handles:
+				raise InvalidWindowHandleError(handle=window)
+		
+			return window
+		
+		if isinstance(window, int):
+			idx = window if window >= 0 else len(handles) + window
+			if idx < 0 or idx >= len(handles):
+				raise InvalidWindowIndexError(index=window, handles_length=len(handles))
+			
+			return handles[idx]
+		
+		return self._current_window_handle_impl()
+	
+	def _close_window_impl(self, window: Optional[Union[str, int]] = None) -> None:
+		current = self._current_window_handle_impl()
+		target = self._get_window_handle_impl(window)
+		legacy_switch_to = self._switch_to_impl()
+		
+		if target == current:
+			self._close_impl()
+		
+			try:
+				remaining = self._window_handles_impl()
+		
+				if remaining:
+					legacy_switch_to.window(remaining[-1])
+			except InvalidSessionIdException:
+				pass
+		else:
+			legacy_switch_to.window(target)
+		
+			self._close_impl()
+		
+			legacy_switch_to.window(current)
+	
+	def _close_all_windows_impl(self) -> None:
+		for window_handle in reversed(self._window_handles_impl()):
+			self._close_window_impl(window_handle)
+	
+	@requires_driver
+	def _fullscreen_window_impl(self) -> None:
+		self._driver_impl.fullscreen_window()
+	
+	@requires_driver
+	def _get_window_position_impl(self, windowHandle: str = "current") -> Any:
+		return self._driver_impl.get_window_position(windowHandle=windowHandle)
+	
+	@requires_driver
+	def _get_window_rect_impl(self) -> Any:
+		return self._driver_impl.get_window_rect()
+	
+	@requires_driver
+	def _get_window_size_impl(self, windowHandle: str = "current") -> Any:
+		return self._driver_impl.get_window_size(windowHandle=windowHandle)
+	
+	@requires_driver
+	def _maximize_window_impl(self) -> None:
+		self._driver_impl.maximize_window()
+	
+	@requires_driver
+	def _minimize_window_impl(self) -> None:
+		self._driver_impl.minimize_window()
+	
+	@requires_driver
+	def _orientation_get_impl(self) -> Literal["LANDSCAPE", "PORTRAIT"]:
+		return self._driver_impl.orientation["orientation"]
+	
+	@requires_driver
+	def _orientation_set_impl(self, value: Literal["LANDSCAPE", "PORTRAIT"]) -> None:
+		setattr(self._driver_impl, "orientation", value)
+	
+	@requires_driver
+	def _set_window_position_impl(self, x: int, y: int, windowHandle: str = "current") -> Any:
+		return self._driver_impl.set_window_position(x=x, y=y, windowHandle=windowHandle)
+	
+	@requires_driver
+	def _set_window_rect_impl(
+			self,
+			x: Optional[int] = None,
+			y: Optional[int] = None,
+			width: Optional[int] = None,
+			height: Optional[int] = None,
+	) -> Any:
+		return self._driver_impl.set_window_rect(x=x, y=y, width=width, height=height)
+	
+	@requires_driver
+	def _set_window_size_impl(self, width: int, height: int, windowHandle: str = "current") -> None:
+		self._driver_impl.set_window_size(width=width, height=height, windowHandle=windowHandle)

osn_selenium/webdrivers/unified/edge/__init__.py

@@ -0,0 +1 @@
+

osn_selenium/webdrivers/unified/edge/base.py

@@ -0,0 +1,56 @@
+from typing import Mapping, Optional, Type
+from osn_selenium.models import WindowRect
+from selenium.webdriver import Edge as legacyEdge
+from osn_selenium.flags.models.edge import EdgeFlags
+from osn_selenium.flags.edge import EdgeFlagsManager
+from osn_selenium._typehints import (
+	ARCHITECTURES_TYPEHINT,
+	PATH_TYPEHINT
+)
+from osn_selenium.webdrivers.unified.blink.base import (
+	UnifiedBlinkBaseMixin
+)
+
+
+__all__ = ["UnifiedEdgeBaseMixin"]
+
+
+class UnifiedEdgeBaseMixin(UnifiedBlinkBaseMixin):
+	def __init__(
+			self,
+			webdriver_path: str,
+			architecture: ARCHITECTURES_TYPEHINT,
+			flags_manager_type: Type[EdgeFlagsManager] = EdgeFlagsManager,
+			use_browser_exe: bool = True,
+			browser_name_in_system: str = "Microsoft Edge",
+			browser_exe: Optional[PATH_TYPEHINT] = None,
+			flags: Optional[EdgeFlags] = None,
+			start_page_url: str = "about:blank",
+			implicitly_wait: int = 5,
+			page_load_timeout: int = 5,
+			script_timeout: int = 5,
+			window_rect: Optional[WindowRect] = None,
+			cdp_versioned_packages_paths: Optional[Mapping[int, PATH_TYPEHINT]] = None,
+			ignore_cdp_version_package_missing: bool = True,
+	):
+		UnifiedBlinkBaseMixin.__init__(
+				self,
+				browser_exe=browser_exe,
+				browser_name_in_system=browser_name_in_system,
+				webdriver_path=webdriver_path,
+				architecture=architecture,
+				use_browser_exe=use_browser_exe,
+				flags_manager_type=flags_manager_type,
+				flags=flags,
+				start_page_url=start_page_url,
+				implicitly_wait=implicitly_wait,
+				page_load_timeout=page_load_timeout,
+				script_timeout=script_timeout,
+				window_rect=window_rect,
+				cdp_versioned_packages_paths=cdp_versioned_packages_paths,
+				ignore_cdp_version_package_missing=ignore_cdp_version_package_missing,
+		)
+	
+	@property
+	def _driver_impl(self) -> Optional[legacyEdge]:
+		return super()._driver_impl

osn_selenium/webdrivers/unified/edge/lifecycle.py

@@ -0,0 +1,88 @@
+from typing import Optional
+from selenium import webdriver
+from osn_selenium.models import WindowRect
+from osn_selenium._typehints import PATH_TYPEHINT
+from selenium.webdriver.edge.service import Service
+from osn_selenium.flags.models.edge import EdgeFlags
+from osn_selenium.webdrivers.unified.core.window import (
+	UnifiedCoreWindowMixin
+)
+from osn_selenium.webdrivers.unified.core.timeouts import (
+	UnifiedCoreTimeoutsMixin
+)
+from osn_selenium.webdrivers.unified.blink.lifecycle import (
+	UnifiedBlinkLifecycleMixin
+)
+
+
+__all__ = ["UnifiedEdgeLifecycleMixin"]
+
+
+class UnifiedEdgeLifecycleMixin(
+		UnifiedBlinkLifecycleMixin,
+		UnifiedCoreTimeoutsMixin,
+		UnifiedCoreWindowMixin
+):
+	def _create_driver_impl(self) -> None:
+		webdriver_options = self._webdriver_flags_manager.options
+		webdriver_service = Service(
+				executable_path=self._webdriver_path,
+				port=self._debugging_port_impl
+				if self._browser_exe_impl is None
+				else 0,
+				service_args=self._webdriver_flags_manager.start_args
+				if self._browser_exe_impl is None
+				else None
+		)
+		
+		self._driver = webdriver.Edge(options=webdriver_options, service=webdriver_service)
+		
+		if self._window_rect is not None:
+			self._set_window_rect_impl(
+					x=self._window_rect.x,
+					y=self._window_rect.y,
+					width=self._window_rect.width,
+					height=self._window_rect.height,
+			)
+		
+		self._set_driver_timeouts_impl(
+				page_load_timeout=self._base_page_load_timeout,
+				implicit_wait_timeout=self._base_implicitly_wait,
+				script_timeout=self._base_implicitly_wait,
+		)
+	
+	def _restart_webdriver_impl(
+			self,
+			flags: Optional[EdgeFlags] = None,
+			browser_exe: Optional[PATH_TYPEHINT] = None,
+			browser_name_in_system: Optional[str] = None,
+			use_browser_exe: Optional[bool] = None,
+			start_page_url: Optional[str] = None,
+			window_rect: Optional[WindowRect] = None,
+	) -> None:
+		super()._restart_webdriver_impl(
+				flags=flags,
+				browser_exe=browser_exe,
+				browser_name_in_system=browser_name_in_system,
+				use_browser_exe=use_browser_exe,
+				start_page_url=start_page_url,
+				window_rect=window_rect,
+		)
+	
+	def _start_webdriver_impl(
+			self,
+			flags: Optional[EdgeFlags] = None,
+			browser_exe: Optional[PATH_TYPEHINT] = None,
+			browser_name_in_system: Optional[str] = None,
+			use_browser_exe: Optional[bool] = None,
+			start_page_url: Optional[str] = None,
+			window_rect: Optional[WindowRect] = None,
+	) -> None:
+		super()._start_webdriver_impl(
+				flags=flags,
+				browser_exe=browser_exe,
+				browser_name_in_system=browser_name_in_system,
+				use_browser_exe=use_browser_exe,
+				start_page_url=start_page_url,
+				window_rect=window_rect,
+		)

osn_selenium/webdrivers/unified/edge/settings.py

@@ -0,0 +1,48 @@
+from typing import Optional
+from osn_selenium.models import WindowRect
+from osn_selenium._typehints import PATH_TYPEHINT
+from osn_selenium.flags.models.edge import EdgeFlags
+from osn_selenium.webdrivers.unified.blink.settings import (
+	UnifiedBlinkSettingsMixin
+)
+
+
+__all__ = ["UnifiedEdgeSettingsMixin"]
+
+
+class UnifiedEdgeSettingsMixin(UnifiedBlinkSettingsMixin):
+	def _reset_settings_impl(
+			self,
+			flags: Optional[EdgeFlags] = None,
+			browser_exe: Optional[PATH_TYPEHINT] = None,
+			browser_name_in_system: Optional[str] = None,
+			use_browser_exe: Optional[bool] = None,
+			start_page_url: str = "",
+			window_rect: Optional[WindowRect] = None,
+	) -> None:
+		super()._reset_settings_impl(
+				flags=flags,
+				browser_exe=browser_exe,
+				browser_name_in_system=browser_name_in_system,
+				use_browser_exe=use_browser_exe,
+				start_page_url=start_page_url,
+				window_rect=window_rect,
+		)
+	
+	def _update_settings_impl(
+			self,
+			flags: Optional[EdgeFlags] = None,
+			browser_exe: Optional[PATH_TYPEHINT] = None,
+			browser_name_in_system: Optional[str] = None,
+			use_browser_exe: Optional[bool] = None,
+			start_page_url: Optional[str] = None,
+			window_rect: Optional[WindowRect] = None,
+	) -> None:
+		super()._update_settings_impl(
+				flags=flags,
+				browser_exe=browser_exe,
+				browser_name_in_system=browser_name_in_system,
+				use_browser_exe=use_browser_exe,
+				start_page_url=start_page_url,
+				window_rect=window_rect,
+		)

osn_selenium/webdrivers/unified/yandex/__init__.py

@@ -0,0 +1 @@
+

osn_selenium/webdrivers/unified/yandex/base.py

@@ -0,0 +1,58 @@
+from typing import Mapping, Optional, Type
+from osn_selenium.models import WindowRect
+from osn_selenium.flags.models.yandex import YandexFlags
+from osn_selenium.flags.yandex import YandexFlagsManager
+from selenium.webdriver import (
+	Chrome as legacyYandex
+)
+from osn_selenium._typehints import (
+	ARCHITECTURES_TYPEHINT,
+	PATH_TYPEHINT
+)
+from osn_selenium.webdrivers.unified.chrome.base import (
+	UnifiedChromeBaseMixin
+)
+
+
+__all__ = ["UnifiedYandexBaseMixin"]
+
+
+class UnifiedYandexBaseMixin(UnifiedChromeBaseMixin):
+	def __init__(
+			self,
+			webdriver_path: str,
+			architecture: ARCHITECTURES_TYPEHINT,
+			flags_manager_type: Type[YandexFlagsManager] = YandexFlagsManager,
+			use_browser_exe: bool = True,
+			browser_name_in_system: str = "Yandex",
+			browser_exe: Optional[PATH_TYPEHINT] = None,
+			flags: Optional[YandexFlags] = None,
+			start_page_url: str = "about:blank",
+			implicitly_wait: int = 5,
+			page_load_timeout: int = 5,
+			script_timeout: int = 5,
+			window_rect: Optional[WindowRect] = None,
+			cdp_versioned_packages_paths: Optional[Mapping[int, PATH_TYPEHINT]] = None,
+			ignore_cdp_version_package_missing: bool = True,
+	):
+		UnifiedChromeBaseMixin.__init__(
+				self,
+				browser_exe=browser_exe,
+				browser_name_in_system=browser_name_in_system,
+				webdriver_path=webdriver_path,
+				architecture=architecture,
+				use_browser_exe=use_browser_exe,
+				flags_manager_type=flags_manager_type,
+				flags=flags,
+				start_page_url=start_page_url,
+				implicitly_wait=implicitly_wait,
+				page_load_timeout=page_load_timeout,
+				script_timeout=script_timeout,
+				window_rect=window_rect,
+				cdp_versioned_packages_paths=cdp_versioned_packages_paths,
+				ignore_cdp_version_package_missing=ignore_cdp_version_package_missing,
+		)
+	
+	@property
+	def _driver_impl(self) -> Optional[legacyYandex]:
+		return super()._driver_impl

osn_selenium/webdrivers/unified/yandex/lifecycle.py

@@ -0,0 +1,58 @@
+from typing import Optional
+from osn_selenium.models import WindowRect
+from osn_selenium._typehints import PATH_TYPEHINT
+from osn_selenium.flags.models.yandex import YandexFlags
+from osn_selenium.webdrivers.unified.core.window import (
+	UnifiedCoreWindowMixin
+)
+from osn_selenium.webdrivers.unified.core.timeouts import (
+	UnifiedCoreTimeoutsMixin
+)
+from osn_selenium.webdrivers.unified.chrome.lifecycle import (
+	UnifiedChromeLifecycleMixin
+)
+
+
+__all__ = ["UnifiedYandexLifecycleMixin"]
+
+
+class UnifiedYandexLifecycleMixin(
+		UnifiedChromeLifecycleMixin,
+		UnifiedCoreTimeoutsMixin,
+		UnifiedCoreWindowMixin
+):
+	def _restart_webdriver_impl(
+			self,
+			flags: Optional[YandexFlags] = None,
+			browser_exe: Optional[PATH_TYPEHINT] = None,
+			browser_name_in_system: Optional[str] = None,
+			use_browser_exe: Optional[bool] = None,
+			start_page_url: Optional[str] = None,
+			window_rect: Optional[WindowRect] = None,
+	) -> None:
+		super()._restart_webdriver_impl(
+				flags=flags,
+				browser_exe=browser_exe,
+				browser_name_in_system=browser_name_in_system,
+				use_browser_exe=use_browser_exe,
+				start_page_url=start_page_url,
+				window_rect=window_rect,
+		)
+	
+	def _start_webdriver_impl(
+			self,
+			flags: Optional[YandexFlags] = None,
+			browser_exe: Optional[PATH_TYPEHINT] = None,
+			browser_name_in_system: Optional[str] = None,
+			use_browser_exe: Optional[bool] = None,
+			start_page_url: Optional[str] = None,
+			window_rect: Optional[WindowRect] = None,
+	) -> None:
+		super()._start_webdriver_impl(
+				flags=flags,
+				browser_exe=browser_exe,
+				browser_name_in_system=browser_name_in_system,
+				use_browser_exe=use_browser_exe,
+				start_page_url=start_page_url,
+				window_rect=window_rect,
+		)

osn_selenium/webdrivers/unified/yandex/settings.py

@@ -0,0 +1,48 @@
+from typing import Optional
+from osn_selenium.models import WindowRect
+from osn_selenium._typehints import PATH_TYPEHINT
+from osn_selenium.flags.models.yandex import YandexFlags
+from osn_selenium.webdrivers.unified.chrome.settings import (
+	UnifiedChromeSettingsMixin
+)
+
+
+__all__ = ["UnifiedYandexSettingsMixin"]
+
+
+class UnifiedYandexSettingsMixin(UnifiedChromeSettingsMixin):
+	def _reset_settings_impl(
+			self,
+			flags: Optional[YandexFlags] = None,
+			browser_exe: Optional[PATH_TYPEHINT] = None,
+			browser_name_in_system: Optional[str] = None,
+			use_browser_exe: Optional[bool] = None,
+			start_page_url: str = "",
+			window_rect: Optional[WindowRect] = None,
+	) -> None:
+		super()._reset_settings_impl(
+				flags=flags,
+				browser_exe=browser_exe,
+				browser_name_in_system=browser_name_in_system,
+				use_browser_exe=use_browser_exe,
+				start_page_url=start_page_url,
+				window_rect=window_rect,
+		)
+	
+	def _update_settings_impl(
+			self,
+			flags: Optional[YandexFlags] = None,
+			browser_exe: Optional[PATH_TYPEHINT] = None,
+			browser_name_in_system: Optional[str] = None,
+			use_browser_exe: Optional[bool] = None,
+			start_page_url: Optional[str] = None,
+			window_rect: Optional[WindowRect] = None,
+	) -> None:
+		super()._update_settings_impl(
+				flags=flags,
+				browser_exe=browser_exe,
+				browser_name_in_system=browser_name_in_system,
+				use_browser_exe=use_browser_exe,
+				start_page_url=start_page_url,
+				window_rect=window_rect,
+		)