a
    xdV2                     @  sb  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
 G dd deZeG d	d
 d
ZG dd dejZG dd dejZG dd dejZG dd dejZddddZddddZdddddZd6ddd d!d"d#Zd7d$d$d%d$d&d'd(d$d)dd*
d+d,Zed-eG d.d/ d/Zed0eG d1d2 d2Zed3eG d4d5 d5ZdS )8    )annotations   )event_classT_JSON_DICT)	dataclassN)ioc                      sB   e Zd ZdZddddZedd dddZ fd	d
Z  ZS )MemoryDumpConfigz[
    Configuration for memory dump. Used only when "memory-infra" category is enabled.
    dictreturnc                 C  s   | S N selfr   r   g/var/www/html/Ranjet/env/lib/python3.9/site-packages/selenium/webdriver/common/devtools/v112/tracing.pyto_json   s    zMemoryDumpConfig.to_jsonjsonr   c                 C  s   | |S r   r   clsr   r   r   r   	from_json   s    zMemoryDumpConfig.from_jsonc                   s   d t  S )NzMemoryDumpConfig({}))formatsuper__repr__r   	__class__r   r   r      s    zMemoryDumpConfig.__repr__)	__name__
__module____qualname____doc__r   classmethodr   r   __classcell__r   r   r   r   r      s
   r   c                   @  s   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Z
d
ed< dZd
ed< dZded< dd Zedd ZdS )TraceConfigNtyping.Optional[str]record_modetyping.Optional[float]trace_buffer_size_in_kbtyping.Optional[bool]enable_samplingenable_systraceenable_argument_filterz!typing.Optional[typing.List[str]]included_categoriesexcluded_categoriessynthetic_delaysz!typing.Optional[MemoryDumpConfig]memory_dump_configc                 C  s   t  }| jd ur| j|d< | jd ur.| j|d< | jd urB| j|d< | jd urV| j|d< | jd urj| j|d< | jd urdd | jD |d< | jd urd	d | jD |d
< | jd urdd | jD |d< | j	d ur| j	
 |d< |S )N
recordModetraceBufferSizeInKbenableSamplingenableSystraceenableArgumentFilterc                 S  s   g | ]}|qS r   r   .0ir   r   r   
<listcomp>I       z'TraceConfig.to_json.<locals>.<listcomp>includedCategoriesc                 S  s   g | ]}|qS r   r   r4   r   r   r   r7   K   r8   excludedCategoriesc                 S  s   g | ]}|qS r   r   r4   r   r   r   r7   M   r8   syntheticDelaysmemoryDumpConfig)r	   r$   r&   r(   r)   r*   r+   r,   r-   r.   r   )r   r   r   r   r   r   <   s(    













zTraceConfig.to_jsonc                 C  s   | d|v rt |d nd d|v r,t|d nd d|v rBt|d nd d|v rXt|d nd d|v rnt|d nd d|v rdd |d D nd d	|v rd
d |d	 D nd d|v rdd |d D nd d|v rt|d nd d	S )Nr/   r0   r1   r2   r3   r9   c                 S  s   g | ]}t |qS r   strr4   r   r   r   r7   Z   r8   z)TraceConfig.from_json.<locals>.<listcomp>r:   c                 S  s   g | ]}t |qS r   r=   r4   r   r   r   r7   [   r8   r;   c                 S  s   g | ]}t |qS r   r=   r4   r   r   r   r7   \   r8   r<   )	r$   r&   r(   r)   r*   r+   r,   r-   r.   )r>   floatboolr   r   r   r   r   r   r   R   s    zTraceConfig.from_json)r   r   r   r$   __annotations__r&   r(   r)   r*   r+   r,   r-   r.   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 )	StreamFormatz
    Data format of a trace. Can be either the legacy JSON format or the
    protocol buffer format. Note that the JSON format will be deprecated soon.
    r   protoc                 C  s   | j S r   valuer   r   r   r   r   i   s    zStreamFormat.to_jsonc                 C  s   | |S r   r   r   r   r   r   r   l   s    zStreamFormat.from_jsonN)	r   r   r   r   ZJSONPROTOr   r    r   r   r   r   r   rB   a   s   rB   c                   @  s,   e Zd ZdZdZdZdd Zedd ZdS )	StreamCompressionzB
    Compression type to use for traces returned via streams.
    nonegzipc                 C  s   | j S r   rD   r   r   r   r   r   x   s    zStreamCompression.to_jsonc                 C  s   | |S r   r   r   r   r   r   r   {   s    zStreamCompression.from_jsonN)	r   r   r   r   NONEZGZIPr   r    r   r   r   r   r   rG   q   s   rG   c                   @  s0   e Zd ZdZdZdZdZdd Zedd Z	d	S )
MemoryDumpLevelOfDetailz
    Details exposed when memory request explicitly declared.
    Keep consistent with memory_dump_request_args.h and
    memory_instrumentation.mojom
    Z
backgroundlightZdetailedc                 C  s   | j S r   rD   r   r   r   r   r      s    zMemoryDumpLevelOfDetail.to_jsonc                 C  s   | |S r   r   r   r   r   r   r      s    z!MemoryDumpLevelOfDetail.from_jsonN)
r   r   r   r   Z
BACKGROUNDZLIGHTZDETAILEDr   r    r   r   r   r   r   rK      s   rK   c                   @  s0   e Zd ZdZdZdZdZdd Zedd Z	d	S )
TracingBackenda  
    Backend type to use for tracing. ``chrome`` uses the Chrome-integrated
    tracing service and is supported on all platforms. ``system`` is only
    supported on Chrome OS and uses the Perfetto system tracing service.
    ``auto`` chooses ``system`` when the perfettoConfig provided to Tracing.start
    specifies at least one non-Chrome data source; otherwise uses ``chrome``.
    autoZchromesystemc                 C  s   | j S r   rD   r   r   r   r   r      s    zTracingBackend.to_jsonc                 C  s   | |S r   r   r   r   r   r   r      s    zTracingBackend.from_jsonN)
r   r   r   r   ZAUTOZCHROMEZSYSTEMr   r    r   r   r   r   r   rM      s   rM   z0typing.Generator[T_JSON_DICT, T_JSON_DICT, None]r
   c                  c  s   ddi} | V }dS )z'
    Stop trace events collection.
    methodzTracing.endNr   cmd_dictr   r   r   r   end   s    rS   z<typing.Generator[T_JSON_DICT, T_JSON_DICT, typing.List[str]]c                  c  s    ddi} | V }dd |d D S )zc
    Gets supported tracing categories.

    :returns: A list of supported tracing categories.
    rP   zTracing.getCategoriesc                 S  s   g | ]}t |qS r   r=   r4   r   r   r   r7      r8   z"get_categories.<locals>.<listcomp>
categoriesr   rQ   r   r   r   get_categories   s    rU   r>   )sync_idr   c                 c  s"   t  }| |d< d|d}|V }dS )zh
    Record a clock sync marker in the trace.

    :param sync_id: The ID of this clock sync marker
    ZsyncIdzTracing.recordClockSyncMarkerrP   paramsNr	   )rV   rX   rR   r   r   r   r   record_clock_sync_marker   s    rZ   r'   z(typing.Optional[MemoryDumpLevelOfDetail]zCtyping.Generator[T_JSON_DICT, T_JSON_DICT, typing.Tuple[str, bool]])deterministiclevel_of_detailr   c                 c  sR   t  }| dur| |d< |dur*| |d< d|d}|V }t|d t|d fS )a  
    Request a global memory dump.

    :param deterministic: *(Optional)* Enables more deterministic results by forcing garbage collection
    :param level_of_detail: *(Optional)* Specifies level of details in memory dump. Defaults to "detailed".
    :returns: A tuple with the following items:

        0. **dumpGuid** - GUID of the resulting global memory dump.
        1. **success** - True iff the global memory dump succeeded.
    Nr[   ZlevelOfDetailzTracing.requestMemoryDumprW   ZdumpGuidsuccess)r	   r   r>   r@   )r[   r\   rX   rR   r   r   r   r   request_memory_dump   s    

r^   r#   r%   typing.Optional[StreamFormat]"typing.Optional[StreamCompression]ztyping.Optional[TraceConfig]ztyping.Optional[TracingBackend])
rT   optionsbuffer_usage_reporting_intervaltransfer_modestream_formatstream_compressiontrace_configperfetto_configtracing_backendr   c	                 c  s   t  }	| dur| |	d< |dur&||	d< |dur6||	d< |durF||	d< |durZ| |	d< |durn| |	d< |dur| |	d< |dur||	d	< |dur| |	d
< d|	d}
|
V }dS )a]  
    Start trace events collection.

    :param categories: *(Optional)* Category/tag filter
    :param options: *(Optional)* Tracing options
    :param buffer_usage_reporting_interval: *(Optional)* If set, the agent will issue bufferUsage events at this interval, specified in milliseconds
    :param transfer_mode: *(Optional)* Whether to report trace events as series of dataCollected events or to save trace to a stream (defaults to ```ReportEvents````).
    :param stream_format: *(Optional)* Trace data format to use. This only applies when using ````ReturnAsStream```` transfer mode (defaults to ````json````).
    :param stream_compression: *(Optional)* Compression format to use. This only applies when using ````ReturnAsStream```` transfer mode (defaults to ````none````)
    :param trace_config: *(Optional)*
    :param perfetto_config: *(Optional)* Base64-encoded serialized perfetto.protos.TraceConfig protobuf message When specified, the parameters ````categories````, ````options````, ````traceConfig```` are ignored.
    :param tracing_backend: *(Optional)* Backend type (defaults to ````auto```)
    NrT   ra   ZbufferUsageReportingIntervalZtransferModeZstreamFormatstreamCompressionZtraceConfigZperfettoConfigZtracingBackendzTracing.startrW   )r	   r   )rT   ra   rb   rc   rd   re   rf   rg   rh   rX   rR   r   r   r   r   start   s.    rj   zTracing.bufferUsagec                   @  s:   e Zd ZU ded< ded< ded< edd dddZd	S )
BufferUsager%   percent_fullevent_countrE   r   r   c                 C  sJ   | d|v rt |d nd d|v r,t |d nd d|v rBt |d nd dS )NZpercentFullZ
eventCountrE   )rl   rm   rE   )r?   r   r   r   r   r   *  s
    zBufferUsage.from_jsonN)r   r   r   rA   r    r   r   r   r   r   rk     s
   
rk   zTracing.dataCollectedc                   @  s.   e Zd ZU dZded< edd dddZdS )	DataCollectedz
    Contains a bucket of collected trace events. When tracing is stopped collected events will be
    sent as a sequence of dataCollected events followed by tracingComplete event.
    ztyping.List[dict]rE   r   r   c                 C  s   | dd |d D dS )Nc                 S  s   g | ]}t |qS r   rY   r4   r   r   r   r7   ?  r8   z+DataCollected.from_json.<locals>.<listcomp>rE   rD   r   r   r   r   r   r   <  s    zDataCollected.from_jsonNr   r   r   r   rA   r    r   r   r   r   r   rn   3  s   
rn   zTracing.tracingCompletec                   @  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 )TracingCompletez
    Signals that tracing is stopped and there is no trace buffers pending flush, all data were
    delivered via dataCollected events.
    r@   data_loss_occurredz typing.Optional[io.StreamHandle]streamr_   trace_formatr`   re   r   r   c                 C  s\   | t |d d|v r$tj|d nd d|v r<t|d nd d|v rTt|d nd dS )NZdataLossOccurredrr   ZtraceFormatri   )rq   rr   rs   re   )r@   r   ZStreamHandler   rB   rG   r   r   r   r   r   T  s    
zTracingComplete.from_jsonNro   r   r   r   r   rp   C  s   
rp   )NN)	NNNNNNNNN)
__future__r   utilr   r   Zdataclassesr   enumtyping r   r	   r   r"   EnumrB   rG   rK   rM   rS   rU   rZ   r^   rj   rk   rn   rp   r   r   r   r   <module>   sJ   B
           "2