创建ChromiumBase支持直接传入端口号

This commit is contained in:
g1879 2023-03-22 18:18:10 +08:00
parent 8d72e275a0
commit 202154dbf7
3 changed files with 5 additions and 2 deletions

View File

@ -42,6 +42,9 @@ class ChromiumBase(BasePage):
self._set = None
self._screencast = None
if isinstance(address, int) or address.isdigit():
address = f'localhost:{address}'
self._set_start_options(address, None)
self._set_runtime_settings()
self._connect_browser(tab_id)

View File

@ -21,7 +21,7 @@ from .session_element import SessionElement
class ChromiumBase(BasePage):
def __init__(self,
address: str,
address: Union[str, int],
tab_id: str = None,
timeout: float = None):
self._control_session: Session = ...

View File

@ -22,7 +22,7 @@ from .session_page import DownloadSetter
class ChromiumPage(ChromiumBase):
def __init__(self,
addr_driver_opts: Union[str, ChromiumOptions, ChromiumDriver, DriverOptions] = None,
addr_driver_opts: Union[str, int, ChromiumOptions, ChromiumDriver, DriverOptions] = None,
tab_id: str = None,
timeout: float = None):
self._driver_options: [ChromiumDriver, DriverOptions] = ...