gJdZddlmZddlZddlZddlZddlmZddlmZm Z ddl m Z ddl mZmZmZmZmZmZmZddlmZmZdd lmZmZdd lmZdd lmZdd l m!Z!m"Z"dd l#m$Z$ddl%m&Z&ddl'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2ddl3m4Z4edZ5e/dZ6e/dZ7e/dZ8eee9e9fZ:GddeZ;Gdde;Z<Gdde;Z=ddZ>y)zConsole exporter for OpenTelemetry. Inspired by https://opentelemetry-python.readthedocs.io/en/latest/_modules/opentelemetry/sdk/trace/export.html#ConsoleSpanExporter ) annotationsN)Sequence)datetimetimezoneindent)AnyListLiteralMappingTextIOTuplecast)Event ReadableSpan) SpanExporterSpanExportResult)types)Columns)ConsoleGroup)Syntax)Text) ATTRIBUTES_JSON_SCHEMA_KEYATTRIBUTES_LOG_LEVEL_NUM_KEYATTRIBUTES_MESSAGE_KEY'ATTRIBUTES_PENDING_SPAN_REAL_PARENT_KEYATTRIBUTES_SPAN_TYPE_KEYATTRIBUTES_TAGS_KEYDISABLE_CONSOLE_KEY LEVEL_NUMBERSNUMBER_TO_LEVELONE_SECOND_IN_NANOSECONDS LevelName)json_args_value_formatter)autoalwaysneverinfowarnerrorceZdZdZ d ddZddZddZdddZddZddZ dd Z dd Z dd Z dd Z ddd Zy)SimpleConsoleSpanExporterzThe ConsoleSpanExporter prints spans to the console. This simple version does not indent spans based on their parent(s), instead spans are printed as a flat list. NcZ|xstj|_|dk(rd}n|dk(}ttj j drdnd|j|ddd|_|jjsd|_||_ |rdnd |_ ||_ t||_ y) Nr'r(PYTEST_VERSIONstandardFT) color_systemfileforce_terminal highlightmarkup soft_wrap r)sysstdout_outputrosenvironget_console is_terminal_include_timestamp_timestamp_indent_verboser"_min_log_level_num)selfoutputcolorsinclude_timestampverbose min_log_levelr4s X/var/www/openai/venv/lib/python3.12/site-packages/logfire/_internal/exporters/console.py__init__z"SimpleConsoleSpanExporter.__init__8s+ V !N#x/N')zz~~6F'GV)   }}(( DM"3'8a "/ ">c|D]S}|jr4|jjtt}||jkrC|j |Ut jS)z Export the spans to the console.) attributesr>r _INFO_LEVELrD _log_spanrSUCCESS)rEspansspan log_levels rKexportz SimpleConsoleSpanExporter.exportWsVD!%!4!45QS^!_ t666 NN4  '''rMc|jrE|jjtd}|dk(s|jjtry|j |y)zPrint a summary of the span, this method can be overridden to customize how spans are displayed. In this simple case we just print the span if its type is not "span" - e.g. the message at the end of a span. rTN)rOr>rr! _print_span)rErT span_types rKrQz#SimpleConsoleSpanExporter._log_spanbsM ??++,DfMIF"doo&9&9:M&N rMc|j||\}}|j|dzzdz}|j||}|r |dg|zz }|jr,|jj t j |n-t djd|D|j|j||td|jxsgDd}|j||y) zMBuild up a summary of the span, including formatting for rich, then print it.r ) r]c3&K|] \}}| ywN).0text_styles rK z8SimpleConsoleSpanExporter._print_span..{s95<4$5sr3c3@K|]}|jdk(s|yw) exceptionN)name)raevents rKrdz8SimpleConsoleSpanExporter._print_span..s\->E%**P[B[%->sN) _span_text_partsrB_details_partsr?printrassemblejoinr;_print_argumentsnextevents_print_exc_info)rErTr_msgparts indent_str details_parts exc_events rKrXz%SimpleConsoleSpanExporter._print_spanos++D&9 e,,vz9S@ ++D*=  j\M1 1E == MM   u 5 6 "''9599 M dJ/\T[[->B->\^bc  Y 3rMcg}|jrJtj|jxsdtz t j }|ddd}||dfdgz }|r ||dzd fgz }|jrS|jjt}|xs |j}|jjtxsd}n|j}d}|tk\r ||d fgz }n|tk\r ||d fgz }n||d fgz }|jxr|jjtx} r(d jt!d | } |dd| ddfgz }||fS)aReturn the formatted message or span name and parts containing basic span information. The following information is included: * timestamp * message (maybe indented) * tags The log level may be indicated by the color of the message. r)tzz %H:%M:%S.%fNgreenr[r]z r]redyellow,z list[str][]cyan)rAr fromtimestamp start_timer$rutcrOr>rrhr _ERROR_LEVEL _WARN_LEVELr rnr) rErTrrttsts_strformatted_messagemsgleveltagstags_strs rKrjz*SimpleConsoleSpanExporter._span_text_partssk  " "'')=AAZ(Z_g_k_klB;'"-F vw'3 3E  v}b)* *E ??,0OO,?,?@V,W #0tyyC,,-IJOaE))CE L sEl^ #E k ! sHo& &E sBi[ E??Ot':':;N'O O4 Oxx[$ 78H iAhZq/6!:; ;EEzrMcP|jr |jsgS|jjd}|r%|jjd}|r|d|z }|jjt}t j|d}|s|r|dfdd|dfd|dfgSgS) zReturn parts containing details for the span if `self._verbose` is True. The following details are returned: * filename and line number * the log level name z code.filepathz code.lineno:r])u│bluer|rr[)rCrOr>rr#)rErTru file_locationlineno log_level_numrUs rKrkz(SimpleConsoleSpanExporter._details_partss}}DOOI!__00A __((7F1VH- !__001MN #'' r: IR 'YK"%  IrMc |jr |jsyi}tdtj|jj t d}|j dijD]W\}}|jj |}|r#tjtt|}t||}|||<Y|sy|jr|j||y|j||y)zQPretty-print formatted logfire arguments for the span if `self._verbose` is True.Ndict[str, Any]z{} properties)schema) rCrOrjsonloadsr>ritemsstrr&r?_print_arguments_rich_print_arguments_plain)rErTru arguments json_schemakeyrvalues rKroz*SimpleConsoleSpanExporter._print_argumentss}}DOO $& +TZZ8K8KLfhl8m-no &??<<BBDKCOO'',E 4U#34-eFCE"IcN E  ==  & &y* =  ' ' : >rMc |jJg}|jD]z\}}t|dd}t|dd}td|j d d zzdd d}|j t g|gt|z|||d ||jjt|y)zSPrint logfire arguments in color using rich, particularly with syntax highlighting.N=r)stylepythondefaultbackground_coloru│ r\)rr)padding) r?rrrcountappendrboolrlr) rErruchunksk value_coderrbarriers rKrz/SimpleConsoleSpanExporter._print_arguments_richs}}((( "&__.MAz!Awf-C:x)LEH (8(8(>(BCSbIQWXG MM$Z(88    #   /" E6N+rMcg}|jD]N\}}|j}||d|d|dgz }|ddt|zd}|ddD] }|||gz }Ptdj ||j y) zJPrint logfire arguments without color using the built-in `print` function.│ rrr[rNr\re)r splitlineslenrlrnr;) rErruoutrr value_linesprefixlines rKrz0SimpleConsoleSpanExporter._print_arguments_plains&__.MAz$//1K zl$qc;q>*:;< r?r indent_textrrlrrnr;) rErwruexc_typeexc_msgexc_tbr indented_coders rKrrz)SimpleConsoleSpanExporter._print_exc_infos(  I$8$8 Y11556FGHsI00445HIJc9//334JKL ==:.f"EGxjO:FH7mG' V0CDMM8iPF MM  gx A6 J \hZr'; ( 4*)V<?,,0 15*rMr.cNeZdZdZ d dfd ZddZxZS)IndentedConsoleSpanExporteraThe ConsoleSpanExporter exports spans to the console, indented. Spans are intended based simply on how many parents they have. This will work well when spans don't overlap, but will be hard to understand when multiple spans are in progress at the same time. c:t||||||i|_yr_)superrL _indent_levelrErFrGrHrIrJ __class__s rKrLz$IndentedConsoleSpanExporter.__init__'s# ):G]S-/rMc~|jxsi}|jtd}|dk(r=|jr0|jj |jj dy|jtry|dk(rdt|}|r|jj|dnd}|jr|jj ndx}rW|dz|j|<nD|jr|jj nd}|r|jj|dnd}|j||y)zXGet the span indent based on `self._indent_level`, then print the span with that indent.rTN pending_spanrr) rOr>rcontextrpopspan_idr!_pending_span_parentparentrX)rErTrOrY parent_idr block_span_ids rKrQz%IndentedConsoleSpanExporter._log_span3s __* NN#;VD  ||""&&t||';';TB  >>- .   &,Z8I=FT''++Iq9AF8<{{ 3 3L}L4:QJ""=104{{ ++I=FT''++Iq9AF v&rMrrr)rrrrrLrQ __classcell__rs@rKrr s]!%&,"&#) 0 0$ 0 0  0 ! 0  0'rMrcbeZdZdZ d dfd ZddZd fd Zd dZxZS) ShowParentsConsoleSpanExporteraThe ConsoleSpanExporter exports spans to the console, indented with parents displayed where necessary. Spans are intended based on how many parents they have, where multiple concurrent spans overlap and therefore the previously displayed span is not the parent or sibling of a span, parents are printed (with "dim" color) so it's easy (or as easy as possible in a terminal) to understand how nested spans are related. cHt||||||i|_g|_yr_)rrL _span_history _span_stackrs rKrLz'ShowParentsConsoleSpanExporter.__init__Ws. ):G]S?A&(rMc|jxsi}|jtd}|dk(r|jr||jj |jj d|jr@|jd|jj k(r|jj y|jtry|j|y)zbPrint any parent spans which aren't in the current stack of displayed spans, then print this span.rTNr) rOr>rrrrrrr!rX)rErTrOrYs rKrQz(ShowParentsConsoleSpanExporter._log_spanfs__* NN#;VD  ||""&&t||';';TB##(8(8(< @T@T(T$$((*  >>- .  rMc||jxsi}|jtd}g}|dk(rt|}||j |z }t |j }t |!||\}}||z }|jxr|jjx} r1|||xsdf|j| <|j j| ||fS|jr|jjnd}||j |z }t |!|t |j \}}||z }||fS)zjParts for any parent spans which aren't in the current stack of displayed spans, then parts for this span.rTrrNr) rOr>rr_parent_stack_text_partsrrrrjrrrr) rErTrrOrYrtrr span_partsrrs rKrjz/ShowParentsConsoleSpanExporter._span_text_partsws;__* NN#;VD   &,Z8I T229= =E))*F#g6tVDOC Z E $ C 0C0CC}C5;S).q4Q""=1  '' 6Ez 04{{ ++I T229= =E#g6tCHXHXDY6ZOC Z EEzrMcg}d}|rJ |j|\}}} |jj|}|jd|dz|_d} |r|jj g}t|D]F\}} }|j|dzz} |d| z| ddfgz }|s,|jj |H|S#t$r|j |||f|}YnwxYw#t $rYwxYw|r) zParts for "intermediate" parent spans - e.g., spans which are not parents of the currently displayed span. Also build up `self._span_stack` to correctly represent the path to the current span. TNrFrr[r\dim) rrindex ValueErrorrKeyErrorclearreversedrB) rErparents clear_stackrrgrand_parent_id stack_indexrtr total_indents rKrz7ShowParentsConsoleSpanExporter._parent_stack_text_partss7 /1  040B0B90M-o"&"2"2"8"8"CK (,'7'78I+/'JD$"'K     " " $'/w&7 "FC11FQJ>L # ,-cU"5u=> >E  '' 2 '8  )"0NNFD)#<= /I0   s"C*CC'&C'* C65C6rrrr)r int | Nonerr) rrrrrLrQrjrrrs@rKrrOsg!%&,"&#) ) )$ ) )  ) ! )  )"2%rMrcJ|jtx}r t|dSy)aqPending span marks the start of a span. Since they're nested within another span we haven't seen yet, we have to do a trick of getting the 'logfire.pending_parent_id' attribute to get the parent indent. Note that returning `0` is equivalent to returning `None` since top level spans get `ATTRIBUTES_PENDING_SPAN_REAL_PARENT_KEY` encoded from `0`. N)r>rr)rO parent_id_strs rKrrs,#'NOO}O="%%PrM)rOz'Mapping[str, otel_types.AttributeValue]rr)?r __future__rrr<r9collections.abcrrrtextwraprrtypingr r r r r rropentelemetry.sdk.tracerropentelemetry.sdk.trace.exportrropentelemetry.utilr otel_types rich.columnsr rich.consolerr rich.syntaxr rich.textr constantsrrrrrr r!r"r#r$r%json_formatterr&rrPrrrrr.rrrr`rMrKr s # $'*CCC7I2 '    778F# F# W%  sCx ! l l^,'";,'^f%>fR &rM