ޚjyWddlZddlZddlZddlZddlmZddlmZddlm Z dZ GddZ Gdd Z d e d e d efd ZGddZddZedk(reyy)N)reduce)Path)DuckDBPyConnectiona :root { --text-primary-color: #0d0d0d; --text-secondary-color: #444; --doc-codebox-border-color: #e6e6e6; --doc-codebox-background-color: #f7f7f7; --doc-scrollbar-bg: #e6e6e6; --doc-scrollbar-slider: #ccc; --duckdb-accent: #009982; --duckdb-accent-light: #00b89a; --card-bg: #fff; --border-radius: 8px; --shadow: 0 4px 14px rgba(0,0,0,0.05); } html, body { margin: 0; padding: 0; font-family: Inter, system-ui, -apple-system, "Segoe UI", Roboto, sans-serif; color: var(--text-primary-color); background: #fafafa; line-height: 1.55; } .container { max-width: 1000px; margin: 40px auto; padding: 0 20px; } header { display: flex; align-items: center; gap: 10px; margin-bottom: 5px; } header img { width: 100px; height: 100px; } header h1 { font-size: 1.5rem; font-weight: 600; margin: 0; color: var(--text-primary-color); } /* === Table Styling (DuckDB documentation style, flat header) === */ table { border-collapse: collapse; width: 100%; margin-bottom: 20px; text-align: left; font-variant-numeric: tabular-nums; border: 1px solid var(--doc-codebox-border-color); border-radius: var(--border-radius); overflow: hidden; box-shadow: var(--shadow); background: var(--card-bg); } thead { background-color: var(--duckdb-accent); color: white; } th, td { padding: 10px 12px; font-size: 14px; vertical-align: top; } th { font-weight: 700; } tbody tr { border-bottom: 1px solid var(--doc-codebox-border-color); } tbody tr:last-child td { border-bottom: none; } tbody tr:hover { background: var(--doc-codebox-border-color); } tbody tr.phase-details-row { border-bottom: none; } tbody tr.phase-details-row:hover { background: transparent; } tbody tr.phase-details-row details summary { font-size: 12px; padding: 4px 0; } tbody tr.phase-details-row details[open] summary { margin-bottom: 4px; } /* === Chart/Card Section === */ .chart { padding: 20px; border: 1px solid var(--doc-codebox-border-color); border-radius: var(--border-radius); background: var(--card-bg); box-shadow: var(--shadow); overflow: visible; } /* === Tree Layout Styling === */ .tf-tree { overflow-x: visible; overflow-y: visible; padding-top: 20px; } .tf-nc { background: var(--card-bg); border: 1px solid var(--doc-codebox-border-color); border-radius: var(--border-radius); padding: 6px; display: inline-block; } .node-body { font-size: 13px; text-align: left; padding: 10px; white-space: nowrap; } .node-body p { margin: 2px 0; } .node-details { white-space: nowrap; overflow: visible; display: inline-block; } /* === Metric Boxes === */ .chart .metrics-grid { display: grid; grid-template-columns: repeat(auto-fit, minmax(150px, 1fr)); gap: 16px; margin-bottom: 20px; } .chart .metric-box { background: var(--card-bg); border: 1px solid var(--doc-codebox-border-color); border-radius: var(--border-radius); box-shadow: var(--shadow); padding: 12px 16px; text-align: center; transition: transform 0.2s ease, box-shadow 0.2s ease; } .chart .metric-box:hover { transform: translateY(-2px); box-shadow: 0 6px 18px rgba(0, 0, 0, 0.08); } .chart .metric-title { font-size: 13px; color: var(--text-secondary-color); margin-bottom: 4px; text-transform: uppercase; letter-spacing: 0.5px; } .chart .metric-value { font-size: 18px; font-weight: 600; color: var(--duckdb-accent); } /* === SQL Query Block === */ .chart.sql-block { background: var(--doc-codebox-background-color); border: 1px solid var(--doc-codebox-border-color); border-radius: var(--border-radius); box-shadow: var(--shadow); padding: 16px; overflow-x: auto; margin-top: 20px; } .chart.sql-block pre { margin: 0; font-family: "JetBrains Mono", "Fira Code", Consolas, monospace; font-size: 13.5px; line-height: 1.5; color: var(--text-primary-color); white-space: pre; } .chart.sql-block code { color: var(--duckdb-accent); font-weight: 500; } /* === Links, Typography, and Consistency === */ a { color: var(--duckdb-accent); text-decoration: underline; transition: color 0.3s; } a:hover { color: black; } strong { font-weight: 600; } /* === Dark Mode Support === */ @media (prefers-color-scheme: dark) { :root { --text-primary-color: #e6e6e6; --text-secondary-color: #b3b3b3; --doc-codebox-border-color: #2a2a2a; --doc-codebox-background-color: #1e1e1e; --card-bg: #111; } body { background: #0b0b0b; } thead { background-color: var(--duckdb-accent); } tbody tr:hover { background: #222; } /* Fix tree node text visibility in dark mode */ .tf-nc .node-body, .tf-nc .node-body p, .tf-nc .node-details { color: #1a1a1a !important; } /* Fix metric title visibility in dark mode */ .chart .metric-title { color: #b3b3b3; } } c<eZdZdedededdfdZdeddfdZd d Zy) NodeTimingphasetimedepthreturnNc<||_||_||_d|_yNr)rr r percentage)selfrr r s ^C:\Users\Luka\AppData\Local\Temp\pip-target-o84fipzb\lib\python\duckdb/query_graph/__main__.py__init__zNodeTiming.__init__s    total_timec,|j|z |_yN)r r)rrs rcalculate_percentagezNodeTiming.calculate_percentages))j0rcv|j|jz}t|j||jSr)r rrr )rrrs rcombine_timingzNodeTiming.combine_timings+YY' $**j$**==r)rrr r) __name__ __module__ __qualname__strfloatintrrrrrrrs;cst1u11>rrcheZdZd dZdeddfdZdedeefdZdedefdZ deefd Z de fd Z y) AllTimingsr Nci|_yrphase_to_timingsrs rrzAllTimings.__init__"s "r node_timingc|j|jvr)|j|jj|y|g|j|j<yr)rr%append)rr's radd_node_timingzAllTimings.add_node_timing%sM    5 5 5  ! !+"3"3 4 ; ;K H8C}D ! !+"3"3 4rrc |j|Srr$rrs rget_phase_timingszAllTimings.get_phase_timings+s$$U++rcPttj|j|Sr)rrrr%r,s rget_summary_phase_timingsz$AllTimings.get_summary_phase_timings.s j//1F1Fu1MNNrctjj}|jfd|j |S)Nc:j|jSr)r/r )xrs rz'AllTimings.get_phases..3s4#A#A!#D"J"Jr)key)listr%keyssortreverse)rphasess` r get_phaseszAllTimings.get_phases1s:d++0023 J K rchd}|jD] }||j|jz }"|Sr )r%r/r )rtotal_timing_sumrs rget_sum_of_all_timingsz!AllTimings.get_sum_of_all_timings7s:**E  > >u E J J J +rr N) rrrrrr*rr5r-r/r:rr=r rrr"r"!sh#E:E$E ,s,tJ/?,OsOzOD,   rr"fpathflagsr c:t|j|dS)Nutf8)modeencoding)ropen)r?r@s r open_utf8rF>s ;    88rceZdZd(dedzdedzddfdZdefdZdefdZd)dedefd Z d*d e d e d e defd Z e dedefdZdededede de de dedefdZde dedefdZde d e de fdZe dedefdZe dedefdZde defdZdedefd Ze ded!eddfd"Zd#ed e ddfd$Z d+d%edzd&edzdeddfd'Zy), ProfilingInfoNconn from_filer c ||_||_yr)rIrJ)rrIrJs rrzProfilingInfo.__init__Cs "rc|j0t|jd5}|jcdddS|jj dS#1swY%xYw)Nrjson)format)rJrFreadrIget_profiling_information)rfs rto_jsonzProfilingInfo.to_jsonGsP >> %4>>3/1vvx0/yy22&2AA0/s AA"cHtj|jSr)rMloadsrRr&s r to_pydictzProfilingInfo.to_pydictNszz$,,.))r output_filecL|j}|j||}|S)N) input_textrV)rR_translate_json_to_html)rrVprofiling_info_text html_outputs rto_htmlzProfilingInfo.to_htmlQs+"lln22>Q_j2k rtop_node query_timingsr ct|dt|d|}|j||dD]}|j|||dzy)N operator_typeoperator_timingchildren)rrr*_get_child_timings)rr]r^r r'childs rrdz ProfilingInfo._get_child_timingsVsT /!:E(K\B]<^`ef %%k2j)E  # #E=%!) D*rfractionctdtd|}d}d}t|d|d|dz |zz}t|d|d|dz |zz}t|d|d|dz |zz}d|d|d|dS)zReturns a shade between very light (#f7fff0) and a slightly darker green-yellow, depending on the fraction (0..1). rrc))ri#02x)maxminr)rf light_color dark_colorrgbs r_get_f7fff0_shade_hexz#ProfilingInfo._get_f7fff0_shade_hex\s q#a*+& $   A*Q-+a."@H!LL M  A*Q-+a."@H!LL M  A*Q-+a."@H!LL M1S'!C3((rnameresultcpu_timecardest result_size extra_infocfd|jt||z d}|dk(rdn|jdd} t|d} d|d } | d z } | d | d z } |d kDr$| d| dz } | d|dz } | d|dz } | d|dz } | dz } | dz } | dz } | d|dz } | dz } | dz } | dz } | dz } | S)z5Generate the HTML body for a single node in the tree.zbackground-color: ;INVALIDBRIDGE_ .4fzz
z

z

rz

time: zs

z

cardinality: z

z

estimate: z

result size: z bytes

z
zExtra infoz
z



z
z)rvrreplace) rrwrxryrzr{r|r} node_stylenew_name formatted_numbodys r_get_node_bodyzProfilingInfo._get_node_bodyns1*$*D*DU6]U]E]*^)__`a $ 18 S#8N =- ,ZL; )) & (++ ? i e4 4D &tfD1 1D mC5- -D &{m:> >D  // ,, #j\&&     r json_graphc d}d}d}d}|dD]&}|d|}|dk(r t|}||d|dz }(tjd d |}tjd d |}|j|d |d||d||dtjdd|} d} t |ddk\r)| dz } |dD]} | |j | |z } | dz } || z| z|zS)Nz
  • z
    • rr}zEstimated Cardinalityz: z
    z__internal_\s*__zcompress_integral\s*compressr`raoperator_cardinalityresult_set_sizez,\s*z, rbrcz
      z
    )rresubrlen_generate_tree_recursive) rrrynode_prefix_htmlnode_suffix_htmlr}estimater4value node_body children_htmlres rrz&ProfilingInfo._generate_tree_recursives9!" l+C|,S1E--u:Rwe44 ,VV-tZ@ VV3ZL ''  ' ( )  - .  ( ) FF7D* -  z*% &! + V #M#J/!>!>uh!OO 0 W $M)+m;>NNNr graph_jsoncTtj|}|j||d}d}d}|j}|j }|j t d|ddg|}|D]+} |j| } | j|| dk(rd| dn| } |d| d t| jd d t| jd zdd d z }| dk7su|j| } t| dkDs|dt| dz }t| ddD]n} | j|d| j dzz}|d|d| j dt| jd dt| jd zdd d z }p|dz }.||z }||zS)z3Generates timing HTML table with expandable phases.z zz
    Phase Time (s) Percentage
    zExecution Time (CPU)Nzzz z dz% rcz
    Show z nodes c|jSr)r )r2s rr3z5ProfilingInfo._generate_timing_html..s166rT)r4r8z z z~
    u ↳ Depth zS z-%
    )rMrT_gather_timing_informationr=r:r*rr/rroundr rrr-rsortedr )rrr^r table_head table_body table_endexecution_time all_phasesrsummarized_phase phase_column phase_timingsr' depth_indents r_generate_timing_htmlz#ProfilingInfo._generate_timing_htmls:ZZ +  '' MB  & &==?"--/ %%j1GY]&^_,:z: E,FFuM   1 1. A059O0OSt,UZL n+00!456)44s:;BQ?@A  J.. - ? ? F }%)' "-012 # J(.mAQ[_'` #88H'/;3D3Dq3H'I "+NNZ^[efqfwfwexyNNST_TdTdfgNhMijNNQR]RhRhknRnNoprqrNsMtu ' (a#JM \ i J&&rc @tj|}t|jddd|jdddk7r t|jdddz dnd|jdddk7r t|jdddz dnd|jdddk7r t|jdddz dnd|jdddk7r|jddd ndd }d }|D]}|d |d ||dz }|dz }|S)NlatencyN/Artotal_bytes_readi@total_bytes_writtensystem_peak_buffer_memorycumulative_rows_scanned,)zExecution Time (s)z Total GB ReadzTotal GB WrittenzPeak Memory (GB)z Rows Scannedz
    zP
    z1
    z&
    r)rMrTrget)rrmetricsmetric_grid_htmlr4s r_generate_metric_grid_htmlz(ProfilingInfo._generate_metric_grid_htmls[ZZ + %*:>>)U+K%LS$Q~~0%8EA!&jnn5G&O PT[ \]`a~~3U;uD$)8Mu)U#VZa#bcf"g~~95AUJ$)8SUZ)[#\`g#hil"m~~7?5H *~~.GOPQR <C %++.%0++23<.9 !   H$rc`tj|}|jdd}d|d}|S)N query_namerzx
    SQL Query 
    
    zI

    )rMrTr)rr sql_querysql_htmls r_generate_sql_query_htmlz&ProfilingInfo._generate_sql_query_htmls?ZZ + NN<7 K  rctj|}t|d}d}d}|j|dd|}||z|zS)Nryz&
      z
    rbr)rMrTrr)rrrry tree_prefix tree_suffix tree_bodys r_generate_tree_htmlz!ProfilingInfo._generate_tree_html!sWZZ + J/0? $ 11*Z2H2KXV Y&44r json_inputcddlm}|j|d}|djd|djd|djd |d S) Nr)HTMLFzW ${CSS} ${LIBRARIES}
    ${CHART_SCRIPT} z${CSS}cssz${CHART_SCRIPT} chart_scriptz ${LIBRARIES} libraries)IPython.core.displayr_generate_htmlr)rrrr[s r_generate_ipythonzProfilingInfo._generate_ipython+s[-))*e< n WX{51 2 W& N(C D W^[%= >   rinclude_meta_infoc d}d}|t|ddS)NzP zp r) treeflex_css duckdb_cssrr) qgraph_css)rrrrs r_generate_style_htmlz"ProfilingInfo._generate_style_html9s j H ,JU^prssrrMc4|j|dd|y)Nrbr)rd)rrMr^s rrz(ProfilingInfo._gather_timing_information?s Z 0 3]Cr input_filerXct}||}n>|&t|d5}|j}dddntdt d|j d}|j |}|j||} |j|} |j|} t|d5}d} | jd|d} | jd |d } | jd |} | jd | } | jd | } | jd| } |j| dddy#1swYxYw#1swYyxYw)Nrz.please provide either input file or input textrcTzw+a  Query Profile Graph for Query ${TREEFLEX_CSS}
    DuckDB Logo

    Query Profile Graph

    ${METRIC_GRID}
    ${SQL_QUERY} ${TIMING_TABLE}
    ${TREE} z${TREEFLEX_CSS}rz ${DUCKDB_CSS}rz${METRIC_GRID}z ${SQL_QUERY}z${TIMING_TABLE}z${TREE}) r"rFrOprintexitrrrrrrwrite) rrrXrVr^textrQr[highlight_metric_grid timing_table tree_outputsql_query_htmlhtmls rrYz%ProfilingInfo._translate_json_to_htmlEsK#  !D  #:s+qvvx,+ B C G//d; $ ? ? E11$ F ..t4 66t< {D )QD8<< 1;~3NOD<<\1JKD<< 02GHD<<?D<< 1<@D<< ;7D GGDMG* ),+* )sE1BE E  E)NN) profile.html)r)NNr)rrrrrrrRdictrUr\objectrrd staticmethodrrvrrrrrrrboolrrrYr rrrHrHBs #/$6##PT*#`d#BB*4*3C E6E&EQTE]`E ))#))"!$05=@GJY\jm 6!O6!OU!Os!OHF'F'vF'RXF'P s s  8 S S  5f55  C  C  ttt$tt DsD6DdD`n5*59)rrMrr functoolsrpathlibrduckdbrrrr"rrrFrHrrr rrrsy %C L>>"  :9S999xxv CD zFr