o ?Di+@srddlZddlZddlZddlZddlZddlmZddlZdZGdddZ GdddZ de d e d e fd d Z d e de d e fddZdefddZde de dedededede d e fddZde ded e fddZd 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 d/e d dfd0d1Zd5d2d3Zed4kredSdS)6N)reducea^ .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@sJeZdZdededefddZdeddfdd Zd ed edefd d ZdS) NodeTimingphasetimereturncCs||_||_d|_dSNr)rr percentage)selfrrr T/var/www/Datamplify/venv/lib/python3.10/site-packages/duckdb/query_graph/__main__.py__init__Qs zNodeTiming.__init__ total_timeNcCs|j||_dSN)rr)r r r r r calculate_percentageWszNodeTiming.calculate_percentagelrcCs|j|j}t|j|Sr)rrr)rrr r r r combine_timingZs  zNodeTiming.combine_timing) __name__ __module__ __qualname__strfloatobjectr rrr r r r rOsrc@sNeZdZddZdefddZdefddZdefd d Zd d Z d dZ dS) AllTimingscCs i|_dSrphase_to_timingsr r r r r b zAllTimings.__init__ node_timingcCs4|j|jvr|j|j|dS|g|j|j<dSr)rrappend)r rr r r add_node_timinges zAllTimings.add_node_timingrcCs |j|Srrr rr r r get_phase_timingskrzAllTimings.get_phase_timingscCsttj|j|Sr)rrrrr!r r r get_summary_phase_timingsnsz$AllTimings.get_summary_phase_timingscs.tj}|jfddd||S)Ncs |jSr)r#r)xrr r ss z'AllTimings.get_phases..)key)listrkeyssortreverse)r phasesr rr get_phasesqszAllTimings.get_phasescCs(d}|jD] }|||j7}q|Sr)rr(r#r)r total_timing_sumrr r r get_sum_of_all_timingswsz!AllTimings.get_sum_of_all_timingsN) rrrr rr rr"r#r,r.r r r r r`s rfpathflagsrcCst||ddS)Nutf8)encoding)open)r/r0r r r open_utf8~sr4top_node query_timingscCs<t|dt|d}|||dD]}t||qdS)N operator_typeoperator_timingchildren)rrr get_child_timings)r5r6rchildr 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)r?)r>#02x)maxminint)r< light_pink dark_pinkrgbr r r get_pink_shade_hexs   rLnameresultcpu_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)rLrreplace) rMrNrOrPrQrRrS 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
    • rrSzEstimated Cardinalityz: z
    operator_cardinalityresult_set_sizer=z__internal_\s*__zcompress_integral\s*compressr7r8z,\s*z, r9z
      z
    )rGrEresubr^lengenerate_tree_recursive) r_rOnode_prefix_htmlnode_suffix_htmlrSestimater&value cardinalityrR node_body children_htmlr;r r r rhs2     rh 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)Nr8latencyz zz
    Phase Time Percentage
    z TOTAL TIMEzExecution Timezzz z dz% )jsonloadsgather_timing_informationrgetr.r,r rr#rrrr) rpr6r_r 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)NrOz&
      z
    r9r)rtrurrh)rpr_rO 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_htmlrY)rr html_outputr r r generate_ipythons  rinclude_meta_infocCs*d}d}|td7}|d7}||dddS)NzP z r`) treeflex_css duckdb_cssrr) qgraph_css)rprrrr r r generate_style_html s rrtcCst|dd|dS)Nr9r)r:)rtr6r r r rvsrv 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})rr4readrrrrYwrite) rrr6ftextr timing_table tree_outputhtmlr r r translate_json_to_html!s        "rcCstjddkrtdtdtjddd}|jdd d |jd d d d |jdd ddd|}|j}|j }|j sOd|vrF| dd}ntdtdnd|j vrX|j }ntdtd|j }t |||ryt j dtj|dddSdS)NrzPlease use python3r=zQuery Graph GeneratorzbGiven 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 jsonz/please provide valid .html file for output namezfile://rB)new)sys version_infoprintexitargparseArgumentParser add_argument parse_argsroutrYr3r webbrowserospathabspath)parserargsinputoutput open_outputr r r mainGs6   r__main__)rN)rtrrrer functoolsrrrrrrrr4r:rrLrGr^rhrrrboolrrvrrrr r r r s2 F*"&   &#