diff --git a/requests_html.py b/requests_html.py
index 48e7fb2..2774284 100644
--- a/requests_html.py
+++ b/requests_html.py
@@ -756,18 +756,27 @@ class BaseSession(requests.Session):
amongst other things.
"""
- def __init__(self, mock_browser : bool = True, verify : bool = True,
- browser_args : list = ['--no-sandbox']):
+ def __init__(
+ self,
+ mock_browser : bool = True,
+ verify : bool = True,
+ browser_args : list = ['--no-sandbox'],
+ headers : dict = None,
+ port : str = None,
+ ):
super().__init__()
# Mock a web browser's user agent.
- if mock_browser:
+ if mock_browser and not headers:
self.headers['User-Agent'] = user_agent()
+ elif headers:
+ self.headers = headers
self.hooks['response'].append(self.response_hook)
self.verify = verify
self.__browser_args = browser_args
+ self.port = port
def response_hook(self, response, **kwargs) -> HTMLResponse: