o h.@sddlmZddlZddlZddlZddlZddlZddlmZm Z ddl m Z er0ddl m Z edZedZe dejdd dZdddZdddZdS)) annotationsN) TYPE_CHECKINGAny)gather_metrics)Iterablez\s*z\s* echoabovec #s>ddlm}m}m}m}|dkr|}|}n |}|j}|j}zftd}|j|jp,d} } | |  } | } Wdn1sCwYt d | } idfd d | | jdjd }| j}| ||}td |}dV||dWdSty}z |d|WYd}~dSd}~ww)aUse in a `with` block to draw some code on the app, then execute it. Parameters ---------- code_location : "above" or "below" Whether to show the echoed code before or after the results of the executed code block. Example ------- >>> import streamlit as st >>> >>> with st.echo(): >>> st.write('This code will be printed') r)codeempty source_utilwarningbelowNnodeast.ASTreturnNonecs>t|dsdSt|D]}t|dr||j<|q dS)Nbodylineno)hasattrastiter_child_nodesr)rchildcollect_body_statementsline_to_node_mapK/var/www/vscode/kcb/lib/python3.10/site-packages/streamlit/commands/echo.pyrLs   z%echo..collect_body_statementspythonzUnable to display code. %s)rrrr) streamlitr r r r traceback extract_stackfilenameropen_python_file readlinesrparsejoinr end_linenotextwrapdedentFileNotFoundError) code_locationr r r r show_code show_warning placeholderframer$ start_line source_file source_lines root_nodeecho_block_start_lineecho_block_end_linelines_to_display code_stringerrrrrr!s8      lines Iterable[str]rintcCs&|D]}t|}|dur|SqdS)zeReturn the indent of the first non-empty line in the list. If all lines are empty, return 0. Nr) _get_indent)r;lineindentrrr_get_initial_indentis rAr?str int | NonecCs0t|dur dSt|}|dur|SdS)zGet the number of whitespaces at the beginning of the given line. If the line is empty, or if it contains just whitespace and a newline, return None. Nr)_EMPTY_LINE_REmatch _SPACES_REend)r?rErrrr>us r>)r)r;r<rr=)r?rBrrC) __future__rr contextlibrer*r"typingrrstreamlit.runtime.metrics_utilrcollections.abcrcompilerFrDcontextmanagerrrAr>rrrrs"      F