From 202154dbf79f7c20080a8258c52011d7306ed6c8 Mon Sep 17 00:00:00 2001
From: g1879 <g1879@qq.com>
Date: Wed, 22 Mar 2023 18:18:10 +0800
Subject: [PATCH] =?UTF-8?q?=E5=88=9B=E5=BB=BAChromiumBase=E6=94=AF?=
 =?UTF-8?q?=E6=8C=81=E7=9B=B4=E6=8E=A5=E4=BC=A0=E5=85=A5=E7=AB=AF=E5=8F=A3?=
 =?UTF-8?q?=E5=8F=B7?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 DrissionPage/chromium_base.py  | 3 +++
 DrissionPage/chromium_base.pyi | 2 +-
 DrissionPage/chromium_page.pyi | 2 +-
 3 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/DrissionPage/chromium_base.py b/DrissionPage/chromium_base.py
index cb8c706..caa52c8 100644
--- a/DrissionPage/chromium_base.py
+++ b/DrissionPage/chromium_base.py
@@ -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)
diff --git a/DrissionPage/chromium_base.pyi b/DrissionPage/chromium_base.pyi
index 98e70b4..89fa21b 100644
--- a/DrissionPage/chromium_base.pyi
+++ b/DrissionPage/chromium_base.pyi
@@ -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 = ...
diff --git a/DrissionPage/chromium_page.pyi b/DrissionPage/chromium_page.pyi
index c63ce2e..3444917 100644
--- a/DrissionPage/chromium_page.pyi
+++ b/DrissionPage/chromium_page.pyi
@@ -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] = ...