o i?Q@sdZddlmZddlZddlZddlZddlmZmZddl m Z ddl m Z ddl m ZddlmZmZmZmZdd lmZmZdd lmZmZdd lmZmZdd lmZmZdd l m!Z!ddl"m#Z#m$Z$ddl%m&Z&ddl'm(Z(ddl)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4ddl5m6Z6ddl7m8Z8edZ9e1dZ:e1dZ;e1dZe?e?fZ@e GdddZAGdddeZBGdddeBZCGdd d eBZDd)d%d&ZEe Gd'd(d(eZFdS)*zConsole exporter for OpenTelemetry. Inspired by https://opentelemetry-python.readthedocs.io/en/latest/_modules/opentelemetry/sdk/trace/export.html#ConsoleSpanExporter ) annotationsN)MappingSequence) dataclass)datetimeindent)AnyLiteralTextIOcast)LogData LogRecord) LogExporterLogExportResult)Event ReadableSpan) SpanExporterSpanExportResult)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)truncate_string)autoalwaysneverinfowarnerrorc@sjeZdZUded<ded<ded<ded<d ed <d ed <ded <ded <edddZedddZdS)RecordMapping[str, object] attributesint timestampstrmessagezSequence[Event]events int | Nonespan_idparent_span_idkindlevelspanrreturnc CsZ|jpi}|||jp d|tp|j|j|jo|jj|jo |jj|t d|t t dS)Nrr;r0r2r4r5r7r8r9r:) r0 start_timegetrnamer5contextr7parentrr _INFO_LEVEL)clsr;r0rEi/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/logfire/_internal/exporters/console.py from_span?s     zRecord.from_spanlogrc Cs|jpi}|t}|sLg}t|ddp|d}r"|t||j}r3|tt|ddndd|D}|tt|ddd |}|||j pU|j pUd|gd|j d |j rd|j jd Std S) N event_name event.named) max_lengthcSsi|] \}}|dkr||qS)rJrE).0kvrErErF Zsz#Record.from_log..: rrHr=)r0r?rgetattrappendr3bodyr'itemsjoinr2observed_timestampr7severity_numbervaluerC)rDrHr0r4partsrIrTother_attributesrErErFfrom_logMs.     zRecord.from_logN)r;rr<r.)rHrr<r.)__name__ __module__ __qualname____annotations__ classmethodrGr\rErErErFr.4s   r.c@seZdZdZ      d:d;ddZdd?d!d"Zd@d$d%ZdAd)d*Z dBd+d,Z dCd/d0Z dCd1d2Z dDd5d6Z d>dEd8d9ZdS)FSimpleConsoleSpanExporterzThe 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. Nr(TFr+output TextIO | NonecolorsConsoleColorsValuesinclude_timestampbool include_tagsverbose min_log_levelr%r<NonecCs|ptj|_|dkr d}n|dk}ttjdrdnd|j|dddd|_|jjs,d|_||_ ||_ |r6dnd |_ ||_ t ||_dS) Nr(r)PYTEST_VERSIONstandardFT) color_systemfileforce_terminal highlightmarkup soft_wrap r)sysstdout_outputrosenvironr?_console is_terminal_include_timestamp _include_tags_timestamp_indent_verboser"_min_log_level_num)selfrcrergrirjrkrqrErErF__init__ps& z"SimpleConsoleSpanExporter.__init__spansSequence[ReadableSpan]rcCs |D] }|t|qtjS)z Export the spans to the console.) export_recordr.rGrSUCCESS)rrr;rErErFexportsz SimpleConsoleSpanExporter.exportr;r.cCs||dS)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. N) _print_spanrr;rErErFrsz'SimpleConsoleSpanExporter.export_recordrrr1cCs|||\}}|jdks|jtrdS|j|jkrdS|j|dd}|||}|r5|dg|7}|j rB|j t j |nt d dd|D|jd |||td d|jp_gDd}|||dS) zMBuild up a summary of the span, including formatting for rich, then print it.r;Nr ) rcss|]\}}|VqdSNrE)rMtext_stylerErErF sz8SimpleConsoleSpanExporter._print_span..rpcss|] }|jdkr|VqdS) exceptionN)r@)rMeventrErErFrs)_span_text_partsr9r0r?r!r:rr_details_partsr{printrassemblerVrx_print_argumentsnextr5_print_exc_info)rr;r_msgrZ indent_str details_parts exc_eventrErErFrs   z%SimpleConsoleSpanExporter._print_spantuple[str, TextParts]c Csg}|jrt|jt}|ddd}||dfdg7}|r(||ddfg7}|j}|j}|tkr:||dfg7}n|tkrF||d fg7}n||dfg7}|j rl|j t }rld td |} |dd | d dfg7}||fS)a<Return the formatted message or span name and parts containing basic span information. The following information is included: * timestamp * message (maybe indented) * tags (if `self._include_tags` is True) The log level may be indicated by the color of the message. z %H:%M:%S.%fNgreen)rrz rredyellow,z list[str][]cyan)r}r fromtimestampr2r$r4r: _ERROR_LEVEL _WARN_LEVELr~r0r?r rVr ) rr;rrZtsts_strmsgr:tagstags_strrErErFrs& z*SimpleConsoleSpanExporter._span_text_partsrr3 TextPartsc Cs|jr|jsgS|jd}|dvrdnt|}|r+|jd}|dvr+|d|7}|jt}t|}|s:|rZ|dfdg}|rL|d|d f|rX|d|df|SgS) 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.filepath)NnullNz code.lineno:r)u│bluerr)rr0r?r3rr#rS) rr;rfile_location_raw file_locationlineno log_level_num log_levelrZrErErFrs$      z(SimpleConsoleSpanExporter._details_partsc Cs|jr|jsdSi}tdt|jtd}|diD]/\}}|j|}|rDt|t rDzt|}Wn tj yCd}Ynwt ||d}|||<q|sSdS|j r^| ||dS|||dS)zQPretty-print formatted logfire arguments for the span if `self._verbose` is True.Ndict[str, Any]z{} properties)schema)rr0r jsonloadsr?rrU isinstancer3JSONDecodeErrorr&r{_print_arguments_rich_print_arguments_plain)rr;r arguments json_schemakeyrrYrErErFrs&    z*SimpleConsoleSpanExporter._print_argumentsrrc Cs|jdusJg}|D]<\}}t|ddd}t|ddd}td|d d dd dd}|tg|gt||||Rd d q |jt |dS)zSPrint logfire arguments in color using rich, particularly with syntax highlighting.N=r)stylepythondefaultbackground_coloru│ r)rr)padding) r{rUrrcountrSrrhrr) rrrchunksrN value_coderrYbarrierrErErFrs*"  z/SimpleConsoleSpanExporter._print_arguments_richc Csg}|D]5\}}|}||d|d|dg7}|ddt|d}|ddD] }|||g7}q/qtd||jddS) zJPrint logfire arguments without color using the built-in `print` function.│ rrrrNrr)rU splitlineslenrrVrx) rrroutrNr value_linesprefixlinerErErFr/sz0SimpleConsoleSpanExporter._print_arguments_plainr Event | Nonec Cs|dus|js dStt|jd}tt|jd}tt|jd}|jrYt|dddd}t|d dd d }t|}t||d}t|d d d}|jt ||||dS|d|d |g}|t||dg7}td ||j ddS)z=Print exception information if an exception event is present.Nzexception.typezexception.messagezexception.stacktracerrr)rendrQzbold red)rrrrrrr) r0r r3r?r{r indent_textrrrrVrx) rrrexc_typeexc_msgexc_tbr indented_coderrErErFr:sz)SimpleConsoleSpanExporter._print_exc_infotimeout_milliscCsdS)z6Force flush all spans, does nothing for this exporter.TrE)rrrErErF force_flushOsz%SimpleConsoleSpanExporter.force_flushNr(TTFr+rcrdrerfrgrhrirhrjrhrkr%r<rl)rrr<rr;r.r<rl)r)r;r.rr1r;r.rr1r<r)r;r.rr3r<r)r;r.rr3)rrrr3r<rl)rrrr3r<rl)rr1r<rh)r]r^r___doc__rrrrrrrrrrrrErErErFrbis& !    &    rbcs:eZdZdZ      ddfdd ZdddZZS)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. Nr(TFr+rcrdrerfrgrhrirjrkr%r<rlcs t||||||i|_dSr)superr _indent_levelrrcrergrirjrk __class__rErFr[s z$IndentedConsoleSpanExporter.__init__r;r.cCs|jdkr|jdur|j|jddS|j}|jdkr8t|j}|r*|j|dnd}|dur7|d|j|<n |rA|j|dnd}|||dS)zXGet the span indent based on `self._indent_level`, then print the span with that indent.r;N pending_spanrr) r9r7rpopr8_pending_span_parentr0r?r)rr; block_span_id parent_idrrErErFrhs    z)IndentedConsoleSpanExporter.export_recordrrr)r]r^r_rrr __classcell__rErErrFrTs rcsReZdZdZ      d#d$fdd Zd%ddZd&fdd Zd'd!d"ZZS)(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. Nr(TFr+rcrdrerfrgrhrirjrkr%r<rlcs&t||||||i|_g|_dSr)rr _span_history _span_stackrrrErFrs  z'ShowParentsConsoleSpanExporter.__init__r;r.cCsV|jdkr$|jdur"|j|jd|jr"|jd|jkr"|jdS||dS)zbPrint any parent spans which aren't in the current stack of displayed spans, then print this span.r;Nr)r9r7rrrrrrErErFrs   z,ShowParentsConsoleSpanExporter.export_recordrr1rcsg}|j}|jdkr@t|j}|||7}t|j}t||\}}||7}|dur<|||p1df|j |<|j |||fS|||7}tj|t|jd\}}||7}||fS)zjParts for any parent spans which aren't in the current stack of displayed spans, then parts for this span.rNrr) r8r9rr0_parent_stack_text_partsrrrrrrS)rr;rrZrrr span_partsrrErFrs     z/ShowParentsConsoleSpanExporter._span_text_partsrr6rc Csg}d}|rEz |j|\}}}Wn tyYn,wz|j|}Wnty5||||f|}Ynw|jd|d|_d}n|s|rL|jg}t|D]"\}} }|j|d} |d| | ddfg7}|rt|j|qR|S) 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. TNrFrrrdim) rKeyErrorrindex ValueErrorrSclearreversedr) rrparents clear_stackrrgrand_parent_id stack_indexrZr total_indentrErErFrs8    z7ShowParentsConsoleSpanExporter._parent_stack_text_partsrrrr)rr6r<r) r]r^r_rrrrrrrErErrFrs   rr0r/r<r6cCs|t}r t|dSdS)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?rr1)r0 parent_id_strrErErFrs rc@s(eZdZUded<d ddZd d Zd S) ConsoleLogExporterrb span_exporterbatchSequence[LogData]r<rcCs$|D] }|jt|jqtjSr)rrr.r\ log_recordrr)rrlog_datarErErFrszConsoleLogExporter.exportcCs|jdSr)rshutdown)rrErErFr szConsoleLogExporter.shutdownN)rr r<r)r]r^r_r`rr rErErErFrs   r)r0r/r<r6)Gr __future__rrryrvcollections.abcrr dataclassesrrtextwraprrtypingr r r r opentelemetry.sdk._logsr ropentelemetry.sdk._logs.exportrropentelemetry.sdk.tracerropentelemetry.sdk.trace.exportrr rich.columnsr rich.consolerr rich.syntaxr rich.textr constantsrrrrrr r!r"r#r$r%json_formatterr&utilsr'rfrCrrlisttupler3rr.rbrrrrrErErErFsF       4 4l+ b