U
    g$                     @   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                       s>   e Zd Zejd Zdeeee ee	 dd fddZ
  ZS )	FirefoxRemoteConnectionZ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/contextZGET_CONTEXT)POSTr   ZSET_CONTEXT)r   z%/session/$sessionId/moz/addon/installZINSTALL_ADDON)r   z'/session/$sessionId/moz/addon/uninstallZUNINSTALL_ADDON)r   z'/session/$sessionId/moz/screenshot/fullZFULL_PAGE_SCREENSHOT)r   super__init__Z	_commands)selfr   r   r	   r
   	__class__ P/tmp/pip-unpacked-wheel-3ysq0mf8/selenium/webdriver/firefox/remote_connection.pyr      s      



z FirefoxRemoteConnection.__init__)TFN)__name__
__module____qualname__r   ZFIREFOXZbrowser_namestrboolr   r   r   __classcell__r   r   r   r   r      s   
   r   N)	typingr   Z.selenium.webdriver.common.desired_capabilitiesr   Z'selenium.webdriver.remote.client_configr   Z+selenium.webdriver.remote.remote_connectionr   r   r   r   r   r   <module>   s   