创建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._set = None
self._screencast = None self._screencast = None
if isinstance(address, int) or address.isdigit():
address = f'localhost:{address}'
self._set_start_options(address, None) self._set_start_options(address, None)
self._set_runtime_settings() self._set_runtime_settings()
self._connect_browser(tab_id) self._connect_browser(tab_id)

View File

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

View File

@ -22,7 +22,7 @@ from .session_page import DownloadSetter
class ChromiumPage(ChromiumBase): class ChromiumPage(ChromiumBase):
def __init__(self, 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, tab_id: str = None,
timeout: float = None): timeout: float = None):
self._driver_options: [ChromiumDriver, DriverOptions] = ... self._driver_options: [ChromiumDriver, DriverOptions] = ...