o 'i,@srddlZddlZddlZddlZddlmZddlmZdZGdddZ GdddZ d e d e d e fd d Z de de d e fddZded e fddZde de dedededede d e fddZde ded e fdd Zd!e de d e fd"d#Zd!e d e fd$d%Zd&e d e fd'd(Zd!e d)ed dfd*d+Zd,e de d dfd-d.Zd/e d0e d dfd1d2Zd6d3d4Zed5kredSdS)7N)reduce)PathaZ .styled-table { border-collapse: collapse; margin: 25px 0; font-size: 0.9em; font-family: sans-serif; min-width: 400px; box-shadow: 0 0 20px rgba(0, 0, 0, 0.15); } .styled-table thead tr { background-color: #009879; color: #ffffff; text-align: left; } .styled-table th, .styled-table td { padding: 12px 15px; } .styled-table tbody tr { border-bottom: 1px solid #dddddd; } .styled-table tbody tr:nth-of-type(even) { background-color: #f3f3f3; } .styled-table tbody tr:last-of-type { border-bottom: 2px solid #009879; } .node-body { font-size:15px; } .tf-nc { position: relative; width: 180px; text-align: center; background-color: #fff100; } .custom-tooltip { position: relative; display: inline-block; } .tooltip-text { visibility: hidden; background-color: #333; color: #fff; text-align: center; padding: 0px; border-radius: 1px; /* Positioning */ position: absolute; z-index: 1; bottom: 100%; left: 50%; transform: translateX(-50%); margin-bottom: 8px; /* Tooltip Arrow */ width: 400px; } .custom-tooltip:hover .tooltip-text { visibility: visible; } c@s>eZdZdededdfddZdeddfdd Zd d d ZdS) NodeTimingphasetimereturnNcCs||_||_d|_dSNr)rr percentage)selfrrr Y/home/ubuntu/transcripts/venv/lib/python3.10/site-packages/duckdb/query_graph/__main__.py__init__Os zNodeTiming.__init__ total_timecCs|j||_dSN)rr )r rr r r calculate_percentageUszNodeTiming.calculate_percentagercCs|j|j}t|j|Sr)rrr)r rrr r r combine_timingXs  zNodeTiming.combine_timing)rrrr)__name__ __module__ __qualname__strfloatr rrr r r r rNsrc@speZdZdddZdeddfddZdedeefd d Zdedefd d Z deefd dZ de fddZ dS) AllTimingsrNcCs i|_dSrphase_to_timingsr r r r r _ zAllTimings.__init__ node_timingcCs4|j|jvr|j|j|dS|g|j|j<dSr)rrappend)r rr r r add_node_timingbs zAllTimings.add_node_timingrcCs |j|Srrr rr r r get_phase_timingshrzAllTimings.get_phase_timingscCsttj|j|Sr)rrrrr r r r get_summary_phase_timingskz$AllTimings.get_summary_phase_timingscs.tj}|jfddd||S)Ncs |jSr)r"r)xrr r ps z'AllTimings.get_phases..)key)listrkeyssortreverse)r phasesr rr get_phasesnszAllTimings.get_phasescCs$d}|jD] }|||j7}q|Sr)rr"r)r total_timing_sumrr r r get_sum_of_all_timingsts z!AllTimings.get_sum_of_all_timingsrN) rrrr rrrr'r!r"r,rr.r r r r r^s rfpathflagsrcCst|j|ddS)Nutf8)modeencoding)ropen)r0r1r r r open_utf8{r#r6top_node query_timingscCs<t|dt|d}|||dD]}t||qdS)N operator_typeoperator_timingchildren)rrrget_child_timings)r7r8rchildr r r r<s    r<fractioncCstdtd|}d}d}t|d|d|d|}t|d|d|d|}t|d|d|d|}d|d|d|dS)Nr)rA)r@#02x)maxminint)r> light_pink dark_pinkrgbr r r get_pink_shade_hexs   rNnameresultcpu_timecardestwidth extra_infoc Csdtt||d}d|d}|d7}|dkrdn|dd } t|d } |d | d | d 7}|d|d7}|dkrW|d|d7}|d|d7}|d|d7}|d7}|d7}|S)Nzbackground-color: ;z*z
INVALIDBRIDGE_ z.4fz

z

time: z seconds

z z rz

cardinality: z

z

estimate: z

width: z bytes

z
z)rNrreplace) rOrPrQrRrSrTrU node_stylebodynew_name formatted_numr r r get_node_bodys  r` json_graphc Csd}d}d}d}|dD]}|d|}|dkrt|}q ||d|d7}q |d }t|d td |} td d |}tdd|}t|d|d|||| tdd|} d} t|dd krx| d7} |dD] } | t| |7} qj| d7} || | |S)Nz
  • z
    • rrUzEstimated Cardinalityz: z
    operator_cardinalityresult_set_sizer?z__internal_\s*__zcompress_integral\s*compressr9r:z,\s*z, r;z
      z
    )rIrGresubr`lengenerate_tree_recursive) rarQnode_prefix_htmlnode_suffix_htmlrUestimater&value cardinalityrT node_body children_htmlr=r r r rjs:      rj graph_jsonc Cst|}t||t|dp|d}d}d}d}|}|}|td||td|ddg|}|D]4} | | } | || dksP| dkrVd| d n| } |d | d | j d t | j d dd d7}q<||7}||S)Nr:latencyz zz
    Phase Time Percentage
    z TOTAL TIMEzExecution Timezzz z dz% )jsonloadsgather_timing_informationrgetr.r,rrr"rrrr ) rrr8rar table_head table_body table_endexecution_time all_phasesrsummarized_phase phase_columnr r r generate_timing_htmls0        rcCs<t|}t|d}d}d}t|dd|}|||S)NrQz&
      z
    r;r)rvrwrrj)rrrarQ tree_prefix tree_suffix tree_bodyr r r generate_tree_htmls   r json_inputcCsBddlm}t|d}|dd|dd|dd |d S) Nr)HTMLFzW ${CSS} ${LIBRARIES}
    ${CHART_SCRIPT} z${CSS}cssz${CHART_SCRIPT} chart_scriptz ${LIBRARIES} libraries)IPython.core.displayr generate_htmlr[)rr html_outputr r r generate_ipythons     rinclude_meta_infocCs*d}d}|td7}|d7}||dddS)NzP z rb) treeflex_css duckdb_cssrr) qgraph_css)rrrrrr r r generate_style_htmls  rrvcCst|dd|dS)Nr;r)r<)rvr8r r r rxsrx input_file output_filec Cst}t|d }|}Wdn1swYt|d}t||}t|}t|d,}d}|d|d}|d|d}|d |}|d |}||WddS1s^wYdS) NrTzw+aT Query Profile Graph for Query ${TREEFLEX_CSS}
    ${TIMING_TABLE}
    ${TREE} z${TREEFLEX_CSS}rz ${DUCKDB_CSS}rz${TIMING_TABLE}z${TREE})rr6readrrrr[write) rrr8ftextr timing_table tree_outputhtmlr r r translate_json_to_htmls        "rcCstjddd}|jddd|jdddd |jd dd d d |}|j}|j}|js@d|vr7|dd}ntdtdnd|jvrI|j}ntdtd|j }t |||rkt j dt | dddSdS)NzQuery Graph GeneratorzjGiven a json profile output, generate a html file showing the query graph and timings of operators)prog description profile_inputzprofile input in json)helpz--outF)requireddefaultz--open store_trueT)ractionrz.jsonz.htmlz%please provide profile output in jsonr?z/please provide valid .html file for output namezfile://rD)new)argparseArgumentParser add_argument parse_argsroutr[printexitr5r webbrowserrresolve)parserargsinputoutput open_outputr r r mainCs0    r__main__r/)rrvrgr functoolsrpathlibrrrrrobjectr6r<rrNrIr`rjrrrboolrrxrrrr r r r s0  F*'&   &"