U
    g                     @  s   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 eG dd	 d	ZeG d
d dZddddZddddZddddZdddddZddddZedeG dd dZdS )    )annotations   )event_classT_JSON_DICT)	dataclassN)network)runtimec                   @  s   e Zd ZU dZded< ded< ded< 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Z
ded< dd Zedd ZdS )LogEntryz
    Log entry.
    strsourceleveltextzruntime.Timestamp	timestampNztyping.Optional[str]urlztyping.Optional[int]line_numberz#typing.Optional[runtime.StackTrace]stack_tracez"typing.Optional[network.RequestId]network_request_id	worker_idz2typing.Optional[typing.List[runtime.RemoteObject]]argsc                 C  s   t  }| j|d< | j|d< | j|d< | j |d< | jd k	rF| j|d< | jd k	rZ| j|d< | jd k	rr| j |d< | j	d k	r| j	 |d< | j
d k	r| j
|d	< | jd k	rd
d | jD |d< |S )Nr   r   r   r   r   
lineNumber
stackTracenetworkRequestIdworkerIdc                 S  s   g | ]}|  qS  to_json.0ir   r   N/tmp/pip-unpacked-wheel-3ysq0mf8/selenium/webdriver/common/devtools/v85/log.py
<listcomp>D   s     z$LogEntry.to_json.<locals>.<listcomp>r   )dictr   r   r   r   r   r   r   r   r   r   r   selfjsonr   r   r   r   3   s$    











zLogEntry.to_jsonc                 C  s   | t |d t |d t |d tj|d d|krBt |d nd d|krXt|d nd d|krrtj|d nd d|krtj|d nd d	|krt |d	 nd d
|krdd |d
 D nd d
S )Nr   r   r   r   r   r   r   r   r   r   c                 S  s   g | ]}t j|qS r   )r   ZRemoteObject	from_jsonr   r   r   r   r    S   s     z&LogEntry.from_json.<locals>.<listcomp>)
r   r   r   r   r   r   r   r   r   r   )r
   r   Z	Timestampr%   intZ
StackTracer   Z	RequestIdclsr$   r   r   r   r%   G   s    


zLogEntry.from_json)__name__
__module____qualname____doc____annotations__r   r   r   r   r   r   r   classmethodr%   r   r   r   r   r	      s   
r	   c                   @  s6   e Zd ZU dZded< ded< dd Zedd	 Zd
S )ViolationSettingz*
    Violation configuration setting.
    r
   namefloat	thresholdc                 C  s   t  }| j|d< | j|d< |S )Nr0   r2   )r!   r0   r2   r"   r   r   r   r   b   s    

zViolationSetting.to_jsonc                 C  s   | t |d t|d dS )Nr0   r2   )r0   r2   )r
   r1   r'   r   r   r   r%   h   s    

zViolationSetting.from_jsonN)r)   r*   r+   r,   r-   r   r.   r%   r   r   r   r   r/   W   s   
r/   z0typing.Generator[T_JSON_DICT, T_JSON_DICT, None])returnc                  c  s   ddi} | V }dS )z
    Clears the log.
    methodz	Log.clearNr   cmd_dictr$   r   r   r   clearp   s     r7   c                  c  s   ddi} | V }dS )z^
    Disables log domain, prevents further log entries from being reported to the client.
    r4   zLog.disableNr   r5   r   r   r   disablez   s     r8   c                  c  s   ddi} | V }dS )z
    Enables log domain, sends the entries collected so far to the client by means of the
    ``entryAdded`` notification.
    r4   z
Log.enableNr   r5   r   r   r   enable   s     r9   ztyping.List[ViolationSetting])configr3   c                 c  s,   t  }dd | D |d< d|d}|V }dS )zV
    start violation reporting.

    :param config: Configuration for violations.
    c                 S  s   g | ]}|  qS r   r   r   r   r   r   r       s     z+start_violations_report.<locals>.<listcomp>r:   zLog.startViolationsReport)r4   paramsN)r!   )r:   r;   r6   r$   r   r   r   start_violations_report   s    r<   c                  c  s   ddi} | V }dS )z#
    Stop violation reporting.
    r4   zLog.stopViolationsReportNr   r5   r   r   r   stop_violations_report   s     r=   zLog.entryAddedc                   @  s.   e Zd ZU dZded< edd dddZdS )	
EntryAddedz-
    Issued when new message was logged.
    r	   entryr   )r$   r3   c                 C  s   | t |d dS )Nr?   )r?   )r	   r%   r'   r   r   r   r%      s    zEntryAdded.from_jsonN)r)   r*   r+   r,   r-   r.   r%   r   r   r   r   r>      s   
r>   )
__future__r   utilr   r   Zdataclassesr   enumtyping r   r   r	   r/   r7   r8   r9   r<   r=   r>   r   r   r   r   <module>   s$   F


