bDg(ddlmZddlmZmZddlmZddlZddlZddl m Z Gdde Z eGd d Z Gd d ejZGd dejZddZddZ ddZd dZ d! d"dZedeGddZedeGddZedeGddZy)#) annotations) event_class T_JSON_DICT) dataclassN)ioc<eZdZdZddZeddZfdZxZS)MemoryDumpConfigz[ Configuration for memory dump. Used only when "memory-infra" category is enabled. c|SNselfs K/var/www/highfloat_scraper/venv/lib/python3.12/site-packages/cdp/tracing.pyto_jsonzMemoryDumpConfig.to_jsons c||Sr r clsjsons r from_jsonzMemoryDumpConfig.from_json 4yrc>djt| S)NzMemoryDumpConfig({}))formatsuper__repr__)r __class__s rrzMemoryDumpConfig.__repr__s%,,UW-=-?@@r)returndict)rrrr ) __name__ __module__ __qualname____doc__r classmethodrr __classcell__)rs@rr r s-AArr ceZdZUdZded<dZded<dZded<dZded<dZded <dZ ded <dZ ded <dZ d ed <ddZ e ddZy) TraceConfigNtyping.Optional[str] record_modeztyping.Optional[bool]enable_samplingenable_systraceenable_argument_filterz!typing.Optional[typing.List[str]]included_categoriesexcluded_categoriessynthetic_delaysz!typing.Optional[MemoryDumpConfig]memory_dump_configcXt}|j|j|d<|j|j|d<|j|j|d<|j|j|d<|j |j Dcgc]}|c}|d<|j |j Dcgc]}|c}|d<|j|jDcgc]}|c}|d<|j|jj|d<|Scc}wcc}wcc}w) N recordModeenableSamplingenableSystraceenableArgumentFilterincludedCategoriesexcludedCategoriessyntheticDelaysmemoryDumpConfig) rr)r*r+r,r-r.r/r0r)rris rrzTraceConfig.to_json:s+ F    '!%!1!1D     +%)%9%9D! "    +%)%9%9D! "  & & 2+/+F+FD' (  # # /595M5M)N!)ND% &  # # /595M5M)N!)ND% &  ,262G2G&HQq&HD" #  " " .'+'>'>'F'F'HD# $ *O)N&Hs D: D"# D'c|d|vrt|dndd|vrt|dndd|vrt|dndd|vrt|dndd|vr|dDcgc] }t|c}ndd|vr|dDcgc] }t|c}ndd|vr|dDcgc] }t|c}ndd|vrtj|d Sd Scc}wcc}wcc}w) Nr2r3r4r5r6r7r8r9)r)r*r+r,r-r.r/r0)strboolr rrrr:s rrzTraceConfig.from_jsonNs3?43GD./TB:A>B:A;?7>=A9@(    rr'c2eZdZdZdZdZddZeddZy) 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. rprotoc|jSr valuers rrzStreamFormat.to_jsond zzrc||Sr r rs rrzStreamFormat.from_jsongrrNrr<)rr<rrA) r r!r"r#JSONPROTOrr$rr rrrArA\s, D ErrAc2eZdZdZdZdZddZeddZy) StreamCompressionzB Compression type to use for traces returned via streams. nonegzipc|jSr rDrs rrzStreamCompression.to_jsonsrFrc||Sr r rs rrzStreamCompression.from_jsonvrrNrH)rr<rrL) r r!r"r#NONEGZIPrr$rr rrrLrLls, D DrrLc#Kddi}|}yw)z' Stop trace events collection. methodz Tracing.endNr cmd_dictrs rendrW{s -H>Ds c#\Kddi}|}|dDcgc] }t|c}Scc}ww)zc Gets supported tracing categories. :returns: A list of supported tracing categories. rTzTracing.getCategories categories)r<)rVrr:s rget_categoriesrZs; )H>D . /qCF // /s ,',c#<Kt}||d<d|d}|}yw)zh Record a clock sync marker in the trace. :param sync_id: The ID of this clock sync marker syncIdzTracing.recordClockSyncMarkerrTparamsNr)sync_idr^rVrs rrecord_clock_sync_markerras/&FF81H>Dsc#RKddi}|}t|dt|dfSw)u Request a global memory dump. :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. rTzTracing.requestMemoryDumpdumpGuidsuccess)r<r=rUs rrequest_memory_dumpresA -H>D D  T)_ s%'c#Kt}|||d<|||d<|||d<|||d<||j|d<||j|d<||j|d<d |d }|} yw) aY Start trace events collection. :param categories: **(DEPRECATED)** *(Optional)* Category/tag filter :param options: **(DEPRECATED)** *(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)* NrYoptionsbufferUsageReportingInterval transferMode streamFormatstreamCompression traceConfigz Tracing.startr])rr) rYrgbuffer_usage_reporting_interval transfer_mode stream_formatstream_compression trace_configr^rVrs rstartrrs(&F)|#y&21P-. !.~ !.!6!6!8~%&8&@&@&B"# , 4 4 6}!H>DsA0A2zTracing.bufferUsagec>eZdZUded<ded<ded<eddZy) BufferUsagetyping.Optional[float] percent_full event_countrEc|d|vrt|dndd|vrt|dndd|vrt|dSdS)N percentFull eventCountrE)rvrwrE)floatrs rrzBufferUsage.from_jsons_7D7LtM23RV5AT5Id<01t*1T/%W &  @D  rN)rrrrt)r r!r"r?r$rr rrrtrts* )('' "!  rrtzTracing.dataCollectedc.eZdZUdZded<eddZy) 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]rEcR||dDcgc] }t|c}Scc}w)NrErDr_r>s rrzDataCollected.from_jsons($(M2q472  2s$N)rrrr}r r!r"r#r?r$rr rrr}r}s!   rr}zTracing.tracingCompletecLeZdZUdZded<ded<ded<ded <ed d Zy ) 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]streamtyping.Optional[StreamFormat] trace_format"typing.Optional[StreamCompression]rpc |t|dd|vr"tjj|dndd|vrtj|dndd|vrt j|dSdS)NdataLossOccurredr traceFormatrk)rrrrp)r=r StreamHandlerrArLrs rrzTracingComplete.from_jsons#D);$<=@HD@P2??,,T(^rsd#-!  At A 8 8  8 v 499    0 7"(,0(,BF.27;AE59'('%'*@', ' 5 ' ? '3' 8'T "#    $ & $%     &   &'    ( r