U g83 @sbddlmZddlmZmZddlmZddlZddlZddl m Z Gddde Z eGd d d Z Gd d d ejZGd ddejZGdddejZGdddejZddddZddddZdddddZd6ddd d!d"d#Zd7d$d$d%d$d&d'd(d$d)dd* d+d,Zed-eGd.d/d/Zed0eGd1d2d2Zed3eGd4d5d5ZdS)8) 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 S/tmp/pip-unpacked-wheel-3ysq0mf8/selenium/webdriver/common/devtools/v130/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 d ed <dZ d ed <dZ d ed <dZ ded<ddZ eddZdS) TraceConfigNtyping.Optional[str] record_modetyping.Optional[float]trace_buffer_size_in_kbtyping.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 rj|j|d<|jdk rdd|jD|d<|jdk rd d|jD|d <|jdk rd d|jD|d <|j dk r|j |d <|S)N recordModetraceBufferSizeInKbenableSamplingenableSystraceenableArgumentFiltercSsg|]}|qSr r .0ir r r Isz'TraceConfig.to_json..includedCategoriescSsg|]}|qSr r r4r r rr7KsexcludedCategoriescSsg|]}|qSr r r4r r rr7MssyntheticDelaysmemoryDumpConfig) r r$r&r(r)r*r+r,r-r.r)rrr r rr<s(              zTraceConfig.to_jsonc Cs|d|krt|dndd|kr,t|dndd|krBt|dndd|krXt|dndd|krnt|dndd|krdd|dDndd |krd d|d Dndd |krd d|d Dndd |krt|d ndd S)Nr/r0r1r2r3r8cSsg|] }t|qSr strr4r r rr7Zsz)TraceConfig.from_json..r9cSsg|] }t|qSr r<r4r r rr7[sr:cSsg|] }t|qSr r<r4r r rr7\sr;) r$r&r(r)r*r+r,r-r.)r=floatboolrrrr r rrRszTraceConfig.from_json)rrrr$__annotations__r&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 rriszStreamFormat.to_jsoncCs||Sr r rr r rrlszStreamFormat.from_jsonN) rrrrZJSONPROTOrr rr r r rrAas rAc@s,eZdZdZdZdZddZeddZdS) StreamCompressionzB Compression type to use for traces returned via streams. nonegzipcCs|jSr rCrr r rrxszStreamCompression.to_jsoncCs||Sr r rr r rr{szStreamCompression.from_jsonN) rrrrNONEZGZIPrr rr r r rrFqs rFc@s0eZdZdZdZdZdZddZeddZ d S) MemoryDumpLevelOfDetailz Details exposed when memory request explicitly declared. Keep consistent with memory_dump_request_args.h and memory_instrumentation.mojom Z backgroundlightZdetailedcCs|jSr rCrr r rrszMemoryDumpLevelOfDetail.to_jsoncCs||Sr r rr r rrsz!MemoryDumpLevelOfDetail.from_jsonN) rrrrZ BACKGROUNDZLIGHTZDETAILEDrr rr r r rrJsrJc@s0eZdZdZdZdZdZddZeddZ 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``. autoZchromesystemcCs|jSr rCrr r rrszTracingBackend.to_jsoncCs||Sr r rr r rrszTracingBackend.from_jsonN) rrrrZAUTOZCHROMEZSYSTEMrr rr r r rrLsrLz0typing.Generator[T_JSON_DICT, T_JSON_DICT, None]r ccsddi}|V}dS)z' Stop trace events collection. methodz Tracing.endNr cmd_dictrr r rendsrRz. categoriesr rPr r rget_categoriess rTr=)sync_idr ccs"t}||d<d|d}|V}dS)z~ Record a clock sync marker in the trace. **EXPERIMENTAL** :param sync_id: The ID of this clock sync marker ZsyncIdzTracing.recordClockSyncMarkerrOparamsNr )rUrWrQrr r rrecord_clock_sync_markers rYr'z(typing.Optional[MemoryDumpLevelOfDetail]zCtyping.Generator[T_JSON_DICT, T_JSON_DICT, typing.Tuple[str, bool]]) deterministiclevel_of_detailr ccsRt}|dk r||d<|dk r*||d<d|d}|V}t|dt|dfS)a Request a global memory dump. **EXPERIMENTAL** :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. NrZZ levelOfDetailzTracing.requestMemoryDumprVZdumpGuidsuccess)r rr=r?)rZr[rWrQrr r rrequest_memory_dumps   r]r#r%typing.Optional[StreamFormat]"typing.Optional[StreamCompression]ztyping.Optional[TraceConfig]ztyping.Optional[TracingBackend]) rSoptionsbuffer_usage_reporting_interval transfer_mode stream_formatstream_compression trace_configperfetto_configtracing_backendr c cst} |dk r|| d<|dk r&|| d<|dk r6|| d<|dk rF|| d<|dk rZ|| d<|dk rn|| d<|dk r|| d<|dk r|| d <|dk r|| d <d | d } | V} dS) a Start trace events collection. :param categories: **(EXPERIMENTAL)** *(Optional)* Category/tag filter :param options: **(EXPERIMENTAL)** *(Optional)* Tracing options :param buffer_usage_reporting_interval: **(EXPERIMENTAL)** *(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: **(EXPERIMENTAL)** *(Optional)* Compression format to use. This only applies when using ````ReturnAsStream```` transfer mode (defaults to ````none````) :param trace_config: *(Optional)* :param perfetto_config: **(EXPERIMENTAL)** *(Optional)* Base64-encoded serialized perfetto.protos.TraceConfig protobuf message When specified, the parameters ````categories````, ````options````, ````traceConfig```` are ignored. :param tracing_backend: **(EXPERIMENTAL)** *(Optional)* Backend type (defaults to ````auto```) NrSr`ZbufferUsageReportingIntervalZ transferModeZ streamFormatstreamCompressionZ traceConfigZperfettoConfigZtracingBackendz Tracing.startrV)r r) rSr`rarbrcrdrerfrgrWrQrr r rstarts.    rizTracing.bufferUsagec@s>eZdZUdZded<ded<ded<eddddd Zd S) BufferUsagez **EXPERIMENTAL** r% percent_full event_countrDrrcCsJ|d|krt|dndd|kr,t|dndd|krBt|dnddS)NZ percentFullZ eventCountrD)rkrlrD)r>rr r rr5s zBufferUsage.from_jsonNrrrrr@r rr r r rrj$s rjzTracing.dataCollectedc@s.eZdZUdZded<edddddZdS) DataCollectedz **EXPERIMENTAL** 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. ztyping.List[dict]rDrrcCs|dd|dDdS)NcSsg|] }t|qSr rXr4r r rr7Lsz+DataCollected.from_json..rDrCr rr r rrIszDataCollected.from_jsonNrmr r r rrn>s rnzTracing.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]streamr^ trace_formatr_rdrrcCs\|t|dd|kr$tj|dndd|krsJ   B !"2