U
    Pf#*                     @   sn   d dl mZ 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	G dd	 d	Z
G d
d deZdS )    )Enum)Any)Dict)DesiredCapabilities)
ArgOptionsc                   @   s   e Zd ZdZdZdS )ElementScrollBehaviorr      N)__name__
__module____qualname__TOPBOTTOM r   r   A/tmp/pip-unpacked-wheel-ds5t_0qj/selenium/webdriver/ie/options.pyr      s   r   c                   @   s.   e Zd ZdZdd Zdd Zdddd	ZdS )
_IeOptionsDescriptora  _IeOptionsDescriptor is an implementation of Descriptor Protocol:

    : Any look-up or assignment to the below attributes in `Options` class will be intercepted
    by `__get__` and `__set__` method respectively.

    - `browser_attach_timeout`
    - `element_scroll_behavior`
    - `ensure_clean_session`
    - `file_upload_dialog_timeout`
    - `force_create_process_api`
    - `force_shell_windows_api`
    - `full_page_screenshot`
    - `ignore_protected_mode_settings`
    - `ignore_zoom_level`
    - `initial_browser_url`
    - `native_events`
    - `persistent_hover`
    - `require_window_focus`
    - `use_per_process_proxy`
    - `use_legacy_file_upload_dialog_handling`
    - `attach_to_edge_chrome`
    - `edge_executable_path`


    : When an attribute lookup happens,
    Example:
        `self. browser_attach_timeout`
        `__get__` method does a dictionary look up in the dictionary `_options` in `Options` class
        and returns the value of key `browserAttachTimeout`
    : When an attribute assignment happens,
    Example:
        `self.browser_attach_timeout` = 30
        `__set__` method sets/updates the value of the key `browserAttachTimeout` in `_options`
        dictionary in `Options` class.
    c                 C   s   || _ || _d S N)nameexpected_type)selfr   r   r   r   r   __init__C   s    z_IeOptionsDescriptor.__init__c                 C   s   |j | jS r   )_optionsgetr   )r   objclsr   r   r   __get__G   s    z_IeOptionsDescriptor.__get__Nreturnc                 C   sV   t || js$t| j d| jj | jdkrF|tjtjfkrFtd||j| j< d S )Nz should be of type elementScrollBehaviorz%Element Scroll Behavior out of range.)	
isinstancer   
ValueErrorr   r	   r   r   r   r   )r   r   valuer   r   r   __set__J   s    z_IeOptionsDescriptor.__set__)r	   r
   r   __doc__r   r   r!   r   r   r   r   r      s   $r   c                       sp  e Zd ZdZdZdZdZdZdZdZ	dZ
d	Zd
ZdZdZdZdZdZdZdZdZdZdZeeeZeeeZeeeZeeeZee	eZee
eZ eeeZ!eeeZ"eeeZ#eee$Z%eeeZ&eeeZ'eeeZ(eeeZ)eeeZ*eeeZ+eee$Z,dd fddZ-e.e/dddZ0e.e/dddZ1e$ddddZ2e/dd d!Z3e.e/dd"d#Z4  Z5S )$Optionszse:ieOptionszie.browserCommandLineSwitchesZbrowserAttachTimeoutr   zie.ensureCleanSessionzie.fileUploadDialogTimeoutzie.forceCreateProcessApizie.forceShellWindowsApizie.enableFullPageScreenshotZignoreProtectedModeSettingsZignoreZoomSettingZinitialBrowserUrlZnativeEventsZenablePersistentHoverZrequireWindowFocuszie.usePerProcessProxyz$ie.useLegacyFileUploadDialogHandlingzie.edgechromiumzie.edgepathzie.ignoreprocessmatchNr   c                    s   t    i | _i | _d S r   )superr   r   _additionalr   	__class__r   r   r   m  s    
zOptions.__init__c                 C   s   | j S )z*:Returns: A dictionary of browser options.)r   r&   r   r   r   optionsr  s    zOptions.optionsc                 C   s   | j S )z!:Returns: The additional options.r%   r&   r   r   r   additional_optionsw  s    zOptions.additional_options)r   r   c                 C   s   || j |< dS )zAdds an additional option not yet added as a safe option for IE.

        :Args:
         - name: name of the option to add
         - value: value of the option to add
        Nr*   )r   r   r    r   r   r   add_additional_option|  s    zOptions.add_additional_optionc                 C   sL   | j }| j }| jr(d| j|| j< | jr:|| j |rH||tj	< |S )z.Marshals the IE options to the correct object. )
Z_capsr   copyZ
_argumentsjoinSWITCHESr%   updater#   KEY)r   Zcapsoptsr   r   r   to_capabilities  s    

zOptions.to_capabilitiesc                 C   s
   t j S r   )r   ZINTERNETEXPLORERr.   r&   r   r   r   default_capabilities  s    zOptions.default_capabilities)6r	   r
   r   r2   r0   ZBROWSER_ATTACH_TIMEOUTZELEMENT_SCROLL_BEHAVIORZENSURE_CLEAN_SESSIONZFILE_UPLOAD_DIALOG_TIMEOUTZFORCE_CREATE_PROCESS_APIZFORCE_SHELL_WINDOWS_APIZFULL_PAGE_SCREENSHOTZIGNORE_PROTECTED_MODE_SETTINGSZIGNORE_ZOOM_LEVELZINITIAL_BROWSER_URLZNATIVE_EVENTSZPERSISTENT_HOVERZREQUIRE_WINDOW_FOCUSZUSE_PER_PROCESS_PROXYZ&USE_LEGACY_FILE_UPLOAD_DIALOG_HANDLINGZATTACH_TO_EDGE_CHROMEZEDGE_EXECUTABLE_PATHZIGNORE_PROCESS_MATCHr   intZbrowser_attach_timeoutr   Zelement_scroll_behaviorboolZensure_clean_sessionZfile_upload_dialog_timeoutZforce_create_process_apiZforce_shell_windows_apiZfull_page_screenshotZignore_protected_mode_settingsZignore_zoom_levelstrZinitial_browser_urlZnative_eventsZpersistent_hoverZrequire_window_focusZuse_per_process_proxyZ&use_legacy_file_upload_dialog_handlingZattach_to_edge_chromeZedge_executable_pathr   propertydictr)   r+   r,   r4   r5   __classcell__r   r   r'   r   r#   V   s\   
















	r#   N)enumr   typingr   r   Z.selenium.webdriver.common.desired_capabilitiesr   Z!selenium.webdriver.common.optionsr   r   r   r#   r   r   r   r   <module>   s   8