U g<) @s8ddlmZddlmZmZddlmZddlZddlZddl m Z Gddde Z eGd d d Z Gd d d ejZGd ddejZddddZddddZdddddZd0dddddZd1ddd dd!d"d#dd$d%d&Zed'eGd(d)d)Zed*eGd+d,d,Zed-eGd.d/d/ZdS)2) annotations) event_class T_JSON_DICT) dataclassN)iocsBeZdZdZddddZedddddZfd d ZZS) MemoryDumpConfigz[ Configuration for memory dump. Used only when "memory-infra" category is enabled. dictreturncCs|SNselfr r R/tmp/pip-unpacked-wheel-3ysq0mf8/selenium/webdriver/common/devtools/v85/tracing.pyto_jsonszMemoryDumpConfig.to_jsonjsonr cCs||Sr r clsrr r r from_jsonszMemoryDumpConfig.from_jsoncsdtS)NzMemoryDumpConfig({}))formatsuper__repr__r __class__r rrszMemoryDumpConfig.__repr__) __name__ __module__ __qualname____doc__r classmethodrr __classcell__r r rrrs rc@seZdZUdZded<dZded<dZded<dZded<dZded <dZ ded <dZ ded <dZ d ed <ddZ e ddZdS) TraceConfigNtyping.Optional[str] record_modetyping.Optional[bool]enable_samplingenable_systraceenable_argument_filterz!typing.Optional[typing.List[str]]included_categoriesexcluded_categoriessynthetic_delaysz!typing.Optional[MemoryDumpConfig]memory_dump_configcCst}|jdk r|j|d<|jdk r.|j|d<|jdk rB|j|d<|jdk rV|j|d<|jdk rtdd|jD|d<|jdk rdd|jD|d <|jdk rd d|jD|d <|jdk r|j |d <|S) N recordModeenableSamplingenableSystraceenableArgumentFiltercSsg|]}|qSr r .0ir r r Csz'TraceConfig.to_json..includedCategoriescSsg|]}|qSr r r1r r rr4EsexcludedCategoriescSsg|]}|qSr r r1r r rr4GssyntheticDelaysmemoryDumpConfig) r r$r&r'r(r)r*r+r,r)rrr r rr8s$            zTraceConfig.to_jsonc Cs|d|krt|dndd|kr,t|dndd|krBt|dndd|krXt|dndd|krtdd|dDndd|krd d|dDndd |krd d|d Dndd |krt|d ndd S)Nr-r.r/r0r5cSsg|] }t|qSr strr1r r rr4Ssz)TraceConfig.from_json..r6cSsg|] }t|qSr r9r1r r rr4Tsr7cSsg|] }t|qSr r9r1r r rr4Usr8)r$r&r'r(r)r*r+r,)r:boolrrrr r rrLszTraceConfig.from_json)rrrr$__annotations__r&r'r(r)r*r+r,rr rr r r rr"s         r"c@s,eZdZdZdZdZddZeddZdS) 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. rprotocCs|jSr valuerr r rrbszStreamFormat.to_jsoncCs||Sr r rr r rreszStreamFormat.from_jsonN) rrrrZJSONPROTOrr rr r r rr=Zs r=c@s,eZdZdZdZdZddZeddZdS) StreamCompressionzB Compression type to use for traces returned via streams. nonegzipcCs|jSr r?rr r rrqszStreamCompression.to_jsoncCs||Sr r rr r rrtszStreamCompression.from_jsonN) rrrrNONEZGZIPrr rr r r rrBjs rBz0typing.Generator[T_JSON_DICT, T_JSON_DICT, None]r ccsddi}|V}dS)z' Stop trace events collection. methodz Tracing.endNr cmd_dictrr r rendysrIz. categoriesr rGr r rget_categoriess rKr:)sync_idr ccs"t}||d<d|d}|V}dS)zh Record a clock sync marker in the trace. :param sync_id: The ID of this clock sync marker ZsyncIdzTracing.recordClockSyncMarkerrFparamsNr )rLrNrHrr r rrecord_clock_sync_markers rPr%zCtyping.Generator[T_JSON_DICT, T_JSON_DICT, typing.Tuple[str, bool]]) deterministicr ccs>t}|dk r||d<d|d}|V}t|dt|dfS)aI Request a global memory dump. :param deterministic: *(Optional)* Enables more deterministic results by forcing garbage collection :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. NrQzTracing.requestMemoryDumprMZdumpGuidsuccess)r r:r;)rQrNrHrr r rrequest_memory_dumps   rSr#typing.Optional[float]typing.Optional[StreamFormat]"typing.Optional[StreamCompression]ztyping.Optional[TraceConfig])rJoptionsbuffer_usage_reporting_interval transfer_mode stream_formatstream_compression trace_configr c cst}|dk r||d<|dk r&||d<|dk r6||d<|dk rF||d<|dk rZ||d<|dk rn||d<|dk r||d<d |d }|V} dS) a7 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)* NrJrWZbufferUsageReportingIntervalZ transferModeZ streamFormatstreamCompressionZ traceConfigz Tracing.startrM)r r) rJrWrXrYrZr[r\rNrHrr r rstarts&   r^zTracing.bufferUsagec@s:eZdZUded<ded<ded<edddddZd S) BufferUsagerT percent_full event_countr@rrcCsJ|d|krt|dndd|kr,t|dndd|krBt|dnddS)NZ percentFullZ eventCountr@)r`rar@)floatrr r rrs zBufferUsage.from_jsonN)rrrr<r rr r r rr_s r_zTracing.dataCollectedc@s.eZdZUdZded<edddddZdS) DataCollectedz Contains an bucket of collected trace events. When tracing is stopped collected events will be send as a sequence of dataCollected events followed by tracingComplete event. ztyping.List[dict]r@rrcCs|dd|dDdS)NcSsg|] }t|qSr rOr1r r rr4sz+DataCollected.from_json..r@r?r rr r rrszDataCollected.from_jsonNrrrrr<r rr r r rrcs rczTracing.tracingCompletec@sFeZdZUdZded<ded<ded<ded <ed dd d d ZdS)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]streamrU trace_formatrVr[rrcCs\|t|dd|kr$tj|dndd|krs@   ;  *