U
    gfU                     @  sz  d dl mZ ddlmZmZ d dlmZ d dlZd dlZddl	m
Z
 ddl	mZ G dd	 d	eZG d
d deZG dd dejZeG dd dZG dd dejZG dd dejZeG dd dZG dd dejZeG dd dZeG dd dZdaddddddd d!Zdbd"dddd#d$d%Zdcddd&d'd(Zddd)ddd*dd+d,d-Zded)ddd.d/d0Zdd1d2d3Zdd1d4d5Zdd1d6d7Z d8d1d9d:Z!d;d1d<d=Z"dfdd*d>d?d@dAZ#dgd)d*dBdCdDdEZ$ddFdGdHdIZ%dhdJdKdLdMdNZ&ddddOdPdQZ'diddddRdSdTZ(dddUdVdWZ)d)ddXdYdZZ*ed[eG d\d] d]Z+ed^eG d_d` d`Z,dS )j    )annotations   )event_classT_JSON_DICT)	dataclassN)page)targetc                      s>   e Zd ZddddZedd dddZ fdd	Z  ZS )
BrowserContextIDstrreturnc                 C  s   | S N selfr   r   S/tmp/pip-unpacked-wheel-3ysq0mf8/selenium/webdriver/common/devtools/v130/browser.pyto_json   s    zBrowserContextID.to_jsonjsonr   c                 C  s   | |S r   r   clsr   r   r   r   	from_json   s    zBrowserContextID.from_jsonc                   s   d t  S )NzBrowserContextID({})formatsuper__repr__r   	__class__r   r   r      s    zBrowserContextID.__repr____name__
__module____qualname__r   classmethodr   r   __classcell__r   r   r   r   r	      s   r	   c                      s>   e Zd ZddddZedd dddZ fdd	Z  ZS )
WindowIDintr   c                 C  s   | S r   r   r   r   r   r   r      s    zWindowID.to_jsonr   c                 C  s   | |S r   r   r   r   r   r   r       s    zWindowID.from_jsonc                   s   d t  S )NzWindowID({})r   r   r   r   r   r   $   s    zWindowID.__repr__r   r   r   r   r   r$      s   r$   c                   @  s4   e Zd ZdZdZdZdZdZdd Ze	dd	 Z
d
S )WindowStatez*
    The state of the browser window.
    normalZ	minimizedZ	maximizedZ
fullscreenc                 C  s   | j S r   valuer   r   r   r   r   1   s    zWindowState.to_jsonc                 C  s   | |S r   r   r   r   r   r   r   4   s    zWindowState.from_jsonN)r   r    r!   __doc__NORMALZ	MINIMIZEDZ	MAXIMIZEDZ
FULLSCREENr   r"   r   r   r   r   r   r&   (   s   r&   c                   @  sb   e Zd ZU dZdZded< dZded< dZded< dZded< dZ	ded	< d
d Z
edd ZdS )Boundsz+
    Browser window bounds information
    Nztyping.Optional[int]lefttopwidthheightztyping.Optional[WindowState]window_statec                 C  sr   t  }| jd k	r| j|d< | jd k	r.| j|d< | jd k	rB| j|d< | jd k	rV| j|d< | jd k	rn| j |d< |S )Nr-   r.   r/   r0   windowState)dictr-   r.   r/   r0   r1   r   r   r   r   r   r   r   M   s    








zBounds.to_jsonc              	   C  sx   | d|krt |d nd d|kr,t |d nd d|krBt |d nd d|krXt |d nd d|krpt|d nd dS )Nr-   r.   r/   r0   r2   )r-   r.   r/   r0   r1   )r%   r&   r   r   r   r   r   r   [   s    zBounds.from_json)r   r    r!   r*   r-   __annotations__r.   r/   r0   r1   r   r"   r   r   r   r   r   r,   9   s   
r,   c                   @  s   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dZdZdZdZdZdZdZdZdZdZ dd  Z!e"d!d" Z#d#S )$PermissionTypeZaccessibilityEventsZaudioCaptureZbackgroundSyncZbackgroundFetchZcapturedSurfaceControlZclipboardReadWriteZclipboardSanitizedWriteZdisplayCaptureZdurableStorageZflashZgeolocationZidleDetectionZ
localFontsZmidiZ	midiSysexZnfcZnotificationsZpaymentHandlerZperiodicBackgroundSyncZprotectedMediaIdentifierZsensorsZstorageAccessZspeakerSelectionZtopLevelStorageAccessZvideoCaptureZvideoCapturePanTiltZoomZwakeLockScreenZwakeLockSystemZwebAppInstallationZwindowManagementc                 C  s   | j S r   r(   r   r   r   r   r      s    zPermissionType.to_jsonc                 C  s   | |S r   r   r   r   r   r   r      s    zPermissionType.from_jsonN)$r   r    r!   ZACCESSIBILITY_EVENTSZAUDIO_CAPTUREZBACKGROUND_SYNCZBACKGROUND_FETCHZCAPTURED_SURFACE_CONTROLZCLIPBOARD_READ_WRITEZCLIPBOARD_SANITIZED_WRITEZDISPLAY_CAPTUREZDURABLE_STORAGEZFLASHZGEOLOCATIONZIDLE_DETECTIONZLOCAL_FONTSZMIDIZ
MIDI_SYSEXNFCZNOTIFICATIONSZPAYMENT_HANDLERZPERIODIC_BACKGROUND_SYNCZPROTECTED_MEDIA_IDENTIFIERZSENSORSZSTORAGE_ACCESSZSPEAKER_SELECTIONZTOP_LEVEL_STORAGE_ACCESSZVIDEO_CAPTUREZVIDEO_CAPTURE_PAN_TILT_ZOOMZWAKE_LOCK_SCREENZWAKE_LOCK_SYSTEMZWEB_APP_INSTALLATIONZWINDOW_MANAGEMENTr   r"   r   r   r   r   r   r6   f   sB   r6   c                   @  s,   e Zd ZdZdZdZdd Zedd ZdS )	PermissionSettingZgrantedZdeniedpromptc                 C  s   | j S r   r(   r   r   r   r   r      s    zPermissionSetting.to_jsonc                 C  s   | |S r   r   r   r   r   r   r      s    zPermissionSetting.from_jsonN)	r   r    r!   ZGRANTEDZDENIEDZPROMPTr   r"   r   r   r   r   r   r8      s   r8   c                   @  sj   e Zd ZU dZded< dZded< dZded< dZded< dZded	< dZ	ded
< dd Z
edd ZdS )PermissionDescriptorz
    Definition of PermissionDescriptor defined in the Permissions API:
    https://w3c.github.io/permissions/#dom-permissiondescriptor.
    r
   nameNtyping.Optional[bool]sysexuser_visible_onlyallow_without_sanitizationallow_without_gesturepan_tilt_zoomc                 C  sx   t  }| j|d< | jd k	r$| j|d< | jd k	r8| j|d< | jd k	rL| j|d< | jd k	r`| j|d< | jd k	rt| j|d< |S )Nr;   r=   userVisibleOnlyallowWithoutSanitizationallowWithoutGesturepanTiltZoom)r3   r;   r=   r>   r?   r@   rA   r4   r   r   r   r      s    










zPermissionDescriptor.to_jsonc              	   C  s   | t |d d|kr t|d nd d|kr6t|d nd d|krLt|d nd d|krbt|d nd d|krxt|d nd dS )Nr;   r=   rB   rC   rD   rE   )r;   r=   r>   r?   r@   rA   )r
   boolr   r   r   r   r      s    
zPermissionDescriptor.from_json)r   r    r!   r*   r5   r=   r>   r?   r@   rA   r   r"   r   r   r   r   r   r:      s   
r:   c                   @  s,   e Zd ZdZdZdZdd Zedd ZdS )	BrowserCommandIdz<
    Browser command ids used by executeBrowserCommand.
    ZopenTabSearchZcloseTabSearchc                 C  s   | j S r   r(   r   r   r   r   r      s    zBrowserCommandId.to_jsonc                 C  s   | |S r   r   r   r   r   r   r      s    zBrowserCommandId.from_jsonN)	r   r    r!   r*   ZOPEN_TAB_SEARCHZCLOSE_TAB_SEARCHr   r"   r   r   r   r   r   rG      s   rG   c                   @  s>   e Zd ZU dZded< ded< ded< dd Zedd	 Zd
S )Bucketz"
    Chrome histogram bucket.
    r%   lowhighcountc                 C  s(   t  }| j|d< | j|d< | j|d< |S )NrI   rJ   rK   )r3   rI   rJ   rK   r4   r   r   r   r      s
    


zBucket.to_jsonc                 C  s&   | t |d t |d t |d dS )NrI   rJ   rK   )rI   rJ   rK   )r%   r   r   r   r   r      s
    


zBucket.from_jsonNr   r    r!   r*   r5   r   r"   r   r   r   r   r   rH      s   
rH   c                   @  sF   e Zd ZU dZded< ded< ded< ded< d	d
 Zedd ZdS )	Histogramz
    Chrome histogram.
    r
   r;   r%   sum_rK   ztyping.List[Bucket]bucketsc                 C  s<   t  }| j|d< | j|d< | j|d< dd | jD |d< |S )Nr;   sumrK   c                 S  s   g | ]}|  qS r   r   .0ir   r   r   
<listcomp>  s     z%Histogram.to_json.<locals>.<listcomp>rO   )r3   r;   rN   rK   rO   r4   r   r   r   r     s    


zHistogram.to_jsonc                 C  s6   | t |d t|d t|d dd |d D dS )Nr;   rP   rK   c                 S  s   g | ]}t |qS r   )rH   r   rR   r   r   r   rU     s     z'Histogram.from_json.<locals>.<listcomp>rO   )r;   rN   rK   rO   )r
   r%   r   r   r   r   r     s    


zHistogram.from_jsonNrL   r   r   r   r   rM      s   
rM   ztyping.Optional[str]z!typing.Optional[BrowserContextID]z0typing.Generator[T_JSON_DICT, T_JSON_DICT, None])
permissionsettingoriginbrowser_context_idr   c                 c  sV   t  }|  |d< | |d< |dk	r.||d< |dk	rB| |d< d|d}|V }dS )a  
    Set permission settings for given origin.

    **EXPERIMENTAL**

    :param permission: Descriptor of permission to override.
    :param setting: Setting of the permission.
    :param origin: *(Optional)* Origin the permission applies to, all origins if not specified.
    :param browser_context_id: *(Optional)* Context to override. When omitted, default browser context is used.
    rV   rW   NrX   browserContextIdzBrowser.setPermissionmethodparamsr3   r   )rV   rW   rX   rY   r]   cmd_dictr   r   r   r   set_permission   s    r`   ztyping.List[PermissionType])permissionsrX   rY   r   c                 c  sP   t  }dd | D |d< |dk	r(||d< |dk	r<| |d< d|d}|V }dS )	aa  
    Grant specific permissions to the given origin and reject all others.

    **EXPERIMENTAL**

    :param permissions:
    :param origin: *(Optional)* Origin the permission applies to, all origins if not specified.
    :param browser_context_id: *(Optional)* BrowserContext to override permissions. When omitted, default browser context is used.
    c                 S  s   g | ]}|  qS r   rQ   rR   r   r   r   rU   M  s     z%grant_permissions.<locals>.<listcomp>ra   NrX   rZ   zBrowser.grantPermissionsr[   r^   )ra   rX   rY   r]   r_   r   r   r   r   grant_permissions>  s    rb   )rY   r   c                 c  s.   t  }| dk	r|  |d< d|d}|V }dS )z
    Reset all permission management for all origins.

    :param browser_context_id: *(Optional)* BrowserContext to reset permissions. When omitted, default browser context is used.
    NrZ   zBrowser.resetPermissionsr[   r^   )rY   r]   r_   r   r   r   r   reset_permissionsY  s    rc   r
   r<   )behaviorrY   download_pathevents_enabledr   c                 c  sV   t  }| |d< |dk	r"| |d< |dk	r2||d< |dk	rB||d< d|d}|V }dS )a  
    Set the behavior when downloading a file.

    **EXPERIMENTAL**

    :param behavior: Whether to allow all or deny all download requests, or use default Chrome behavior if available (otherwise deny). ``allowAndName`` allows download and names files according to their download guids.
    :param browser_context_id: *(Optional)* BrowserContext to set download behavior. When omitted, default browser context is used.
    :param download_path: *(Optional)* The default path to save downloaded files to. This is required if behavior is set to 'allow' or 'allowAndName'.
    :param events_enabled: *(Optional)* Whether to emit download events (defaults to false).
    rd   NrZ   ZdownloadPathZeventsEnabledzBrowser.setDownloadBehaviorr[   r^   )rd   rY   re   rf   r]   r_   r   r   r   r   set_download_behaviork  s    rg   )guidrY   r   c                 c  s6   t  }| |d< |dk	r"| |d< d|d}|V }dS )a   
    Cancel a download if in progress

    **EXPERIMENTAL**

    :param guid: Global unique identifier of the download.
    :param browser_context_id: *(Optional)* BrowserContext to perform the action in. When omitted, default browser context is used.
    rh   NrZ   zBrowser.cancelDownloadr[   r^   )rh   rY   r]   r_   r   r   r   r   cancel_download  s    ri   r   c                  c  s   ddi} | V }dS )z#
    Close browser gracefully.
    r\   zBrowser.closeNr   r_   r   r   r   r   close  s     rk   c                  c  s   ddi} | V }dS )zC
    Crashes browser on the main thread.

    **EXPERIMENTAL**
    r\   zBrowser.crashNr   rj   r   r   r   crash  s     rl   c                  c  s   ddi} | V }dS )z4
    Crashes GPU process.

    **EXPERIMENTAL**
    r\   zBrowser.crashGpuProcessNr   rj   r   r   r   crash_gpu_process  s     rm   zQtyping.Generator[T_JSON_DICT, T_JSON_DICT, typing.Tuple[str, str, str, str, str]]c                  c  sD   ddi} | V }t |d t |d t |d t |d t |d fS )a,  
    Returns version information.

    :returns: A tuple with the following items:

        0. **protocolVersion** - Protocol version.
        1. **product** - Product name.
        2. **revision** - Product revision.
        3. **userAgent** - User-Agent.
        4. **jsVersion** - V8 version.
    r\   zBrowser.getVersionZprotocolVersionproductrevisionZ	userAgentZ	jsVersionr
   rj   r   r   r   get_version  s     




rq   z<typing.Generator[T_JSON_DICT, T_JSON_DICT, typing.List[str]]c                  c  s    ddi} | V }dd |d D S )z
    Returns the command line switches for the browser process if, and only if
    --enable-automation is on the commandline.

    **EXPERIMENTAL**

    :returns: Commandline parameters
    r\   zBrowser.getBrowserCommandLinec                 S  s   g | ]}t |qS r   rp   rR   r   r   r   rU     s     z,get_browser_command_line.<locals>.<listcomp>	argumentsr   rj   r   r   r   get_browser_command_line  s
    
 rs   zBtyping.Generator[T_JSON_DICT, T_JSON_DICT, typing.List[Histogram]])querydeltar   c                 c  sH   t  }| dk	r| |d< |dk	r&||d< d|d}|V }dd |d D S )	aY  
    Get Chrome histograms.

    **EXPERIMENTAL**

    :param query: *(Optional)* Requested substring in name. Only histograms which have query as a substring in their name are extracted. An empty or absent query returns all histograms.
    :param delta: *(Optional)* If true, retrieve delta since last delta call.
    :returns: Histograms.
    Nrt   ru   zBrowser.getHistogramsr[   c                 S  s   g | ]}t |qS r   )rM   r   rR   r   r   r   rU     s     z"get_histograms.<locals>.<listcomp>Z
histogramsr3   )rt   ru   r]   r_   r   r   r   r   get_histograms  s    rw   z5typing.Generator[T_JSON_DICT, T_JSON_DICT, Histogram])r;   ru   r   c                 c  s<   t  }| |d< |dk	r||d< d|d}|V }t|d S )z
    Get a Chrome histogram by name.

    **EXPERIMENTAL**

    :param name: Requested histogram name.
    :param delta: *(Optional)* If true, retrieve delta since last delta call.
    :returns: Histogram.
    r;   Nru   zBrowser.getHistogramr[   Z	histogram)r3   rM   r   )r;   ru   r]   r_   r   r   r   r   get_histogram  s    rx   z2typing.Generator[T_JSON_DICT, T_JSON_DICT, Bounds])	window_idr   c                 c  s0   t  }|  |d< d|d}|V }t|d S )z
    Get position and size of the browser window.

    **EXPERIMENTAL**

    :param window_id: Browser window id.
    :returns: Bounds information of the window. When window state is 'minimized', the restored window position and size are returned.
    windowIdzBrowser.getWindowBoundsr[   bounds)r3   r   r,   r   )ry   r]   r_   r   r   r   r   get_window_bounds  s    r|   z typing.Optional[target.TargetID]zJtyping.Generator[T_JSON_DICT, T_JSON_DICT, typing.Tuple[WindowID, Bounds]])	target_idr   c                 c  sF   t  }| dk	r|  |d< d|d}|V }t|d t|d fS )a  
    Get the browser window that contains the devtools target.

    **EXPERIMENTAL**

    :param target_id: *(Optional)* Devtools agent host id. If called as a part of the session, associated targetId is used.
    :returns: A tuple with the following items:

        0. **windowId** - Browser window id.
        1. **bounds** - Bounds information of the window. When window state is 'minimized', the restored window position and size are returned.
    NZtargetIdzBrowser.getWindowForTargetr[   rz   r{   )r3   r   r$   r   r,   )r}   r]   r_   r   r   r   r   get_window_for_target4  s    r~   )ry   r{   r   c                 c  s2   t  }|  |d< | |d< d|d}|V }dS )a5  
    Set position and/or size of the browser window.

    **EXPERIMENTAL**

    :param window_id: Browser window id.
    :param bounds: New window bounds. The 'minimized', 'maximized' and 'fullscreen' states cannot be combined with 'left', 'top', 'width' or 'height'. Leaves unspecified fields unchanged.
    rz   r{   zBrowser.setWindowBoundsr[   Nr^   )ry   r{   r]   r_   r   r   r   r   set_window_boundsP  s    r   )badge_labelimager   c                 c  s:   t  }| dk	r| |d< |dk	r&||d< d|d}|V }dS )z
    Set dock tile details, platform-specific.

    **EXPERIMENTAL**

    :param badge_label: *(Optional)*
    :param image: *(Optional)* Png encoded image.
    NZ
badgeLabelr   zBrowser.setDockTiler[   rv   )r   r   r]   r_   r   r   r   r   set_dock_tilef  s    r   )
command_idr   c                 c  s&   t  }|  |d< d|d}|V }dS )zi
    Invoke custom browser commands used by telemetry.

    **EXPERIMENTAL**

    :param command_id:
    Z	commandIdzBrowser.executeBrowserCommandr[   Nr^   )r   r]   r_   r   r   r   r   execute_browser_command~  s    
r   )urlr   c                 c  s"   t  }| |d< d|d}|V }dS )z
    Allows a site to use privacy sandbox features that require enrollment
    without the site actually being enrolled. Only supported on page targets.

    :param url:
    r   z+Browser.addPrivacySandboxEnrollmentOverrider[   Nrv   )r   r]   r_   r   r   r   r   'add_privacy_sandbox_enrollment_override  s    	r   zBrowser.downloadWillBeginc                   @  sF   e Zd ZU dZded< ded< ded< ded< edd d	d
dZdS )DownloadWillBeginzM
    **EXPERIMENTAL**

    Fired when page is about to start a download.
    zpage.FrameIdframe_idr
   rh   r   suggested_filenamer   r   c                 C  s4   | t j|d t|d t|d t|d dS )NZframeIdrh   r   ZsuggestedFilename)r   rh   r   r   )r   ZFrameIdr   r
   r   r   r   r   r     s    


zDownloadWillBegin.from_jsonNr   r    r!   r*   r5   r"   r   r   r   r   r   r     s   
r   zBrowser.downloadProgressc                   @  sF   e Zd ZU dZded< ded< ded< ded< edd d	d
dZdS )DownloadProgresszc
    **EXPERIMENTAL**

    Fired when download makes progress. Last call has ``done`` == true.
    r
   rh   floattotal_bytesreceived_bytesstater   r   c                 C  s0   | t |d t|d t|d t |d dS )Nrh   Z
totalBytesZreceivedBytesr   )rh   r   r   r   )r
   r   r   r   r   r   r     s    



zDownloadProgress.from_jsonNr   r   r   r   r   r     s   
r   )NN)NN)N)NNN)N)NN)N)N)NN)-
__future__r   utilr   r   Zdataclassesr   enumtyping r   r   r
   r	   r%   r$   Enumr&   r,   r6   r8   r:   rG   rH   rM   r`   rb   rc   rg   ri   rk   rl   rm   rq   rs   rw   rx   r|   r~   r   r   r   r   r   r   r   r   r   r   <module>   sv   ,(4%         ! 
      