o
    5@Hh$                     @   sD   d dl mZ d dlmZ d dlmZ d dlmZ G dd deZdS )    )Optional)DesiredCapabilities)ClientConfig)RemoteConnectionc                       sL   e Zd Zejd Z			ddededee dee	 d	df
 fd
dZ
  ZS )FirefoxRemoteConnectionbrowserNameTFNremote_server_addr
keep_aliveignore_proxyclient_configreturnc                    sX   |pt ||dd}t j||d d| jd< d| jd< d| jd	< d
| jd< d| jd< d S )Nx   )r   r	   timeout)r
   r   )GET/session/$sessionId/moz/contextGET_CONTEXT)POSTr   SET_CONTEXT)r   z%/session/$sessionId/moz/addon/installINSTALL_ADDON)r   z'/session/$sessionId/moz/addon/uninstallUNINSTALL_ADDON)r   z'/session/$sessionId/moz/screenshot/fullFULL_PAGE_SCREENSHOT)r   super__init__	_commands)selfr   r	   r
   r   	__class__ l/home/air/sanwanet/gpt-api/venv/lib/python3.10/site-packages/selenium/webdriver/firefox/remote_connection.pyr      s   



z FirefoxRemoteConnection.__init__)TFN)__name__
__module____qualname__r   FIREFOXbrowser_namestrboolr   r   r   __classcell__r   r   r   r   r      s     
r   N)	typingr   .selenium.webdriver.common.desired_capabilitiesr   'selenium.webdriver.remote.client_configr   +selenium.webdriver.remote.remote_connectionr   r   r   r   r   r   <module>   s
   