o i6@sddlmZddlZddlmZmZmZmZddlmZddl m Z m Z m Z ddl mZmZmZddlmZGd d d Zed d Zedd ZdS) ) annotationsN)datedatetimetime timedelta)partial)AnyCallablecast) ArraySchemaDataType JSONSchema) safe_reprc@seZdZdZd?ddZdddd@ddZdAddZdBddZdCddZdDd!d"Z dDd#d$Z dEd'd(Z dFd)d*Z dFd+d,Z dFd-d.ZdGd5d6ZdHd7d8ZdFd9d:ZdFd;d<ZdFd=d>ZdS)IJsonArgsValueFormatterzFormat values recursively based on the information provided in value dict. When a custom format is identified, the `$__datatype__` key is always present. indentintc Csp||_|dk|_id|jd|jd|jdt|jdddd d t|jdddd d t|jd d dddt|jdddt|jdddt|jdddt|jdddt|jdddt|jddd|jdt|j d ddd!|j d"|j d#|j id$|j d%t|j dddd&t|j d'ddd(t|j d)ddd*t|j d+ddd,t|j d-ddd.t|j d/ddd0t|j d1ddd2t|j d3ddd4t|j d5ddd6t|j d7ddd8t|j d9ddd:t|j d;dd dt|j d?d?d d@t|j dAdBd dCt|j dddt|jdDd|jt|jdddd t|jdddd t|j d?d?d dE|_dS)FNrstrrfloat PydanticModel(=)F dataclassMappingz({: z})TtupleSequencez([z])set{} frozensetz frozenset({dequezdeque([ generatorz generator((z))bytesDecimalzDecimal(rrrrEnum IPv4Addressz IPv4Address(AnyUrlzAnyUrl(UrlzUrl( IPv4InterfacezIPv4Interface( IPv4Networkz IPv4Network( IPv6Addressz IPv6Address( IPv6InterfacezIPv6Interface( IPv6Networkz IPv6Network( PosixPathz PosixPath(Patternz re.compile( SecretBytesz SecretBytes( SecretStrz SecretStr( NameEmailUUIDzUUID('z') Exceptionzarray([)ndarray DataFrameattrs sqlalchemyunknown) _indent_step _newlines_format_string_format_numberr _format_items_format_list_like_format_sequence _format_bytes_write _format_date_format_datetime _format_time_format_timedelta_format_data_frame_data_type_map)selfrrLf/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/logfire/_internal/json_formatter.py__init__s       !" #zJsonArgsValueFormatter.__init__Nr)schemaindent_currentvaluerrOJSONSchema | NonerPcCs$t|_||d|||jS)NT)ioStringIO_stream_formatgetvalue)rKrQrOrPrLrLrM__call__?s  zJsonArgsValueFormatter.__call__use_reprbool dict[str, Any] | list[Any] | AnyreturnNonec Cs|durrd|vrd|d}durK|ddkr)t|tr)|dddd||ddS|ddkr?t|tr?|d d |||dS|d d d d |ddS|j|}|dus\Jd|||||dS|d d d d t|ddS|rxt|}|d d d d |ddS)Ntypezx-python-datatypeobjectrrr Tarray[]r4FrzUnknown data type ) get isinstancedictr@listrArDrJr)rKrPrYrQrO data_typefuncrLrLrMrVDs zJsonArgsValueFormatter._formatopen_rafter__indent_currentcCsJ|dur|d}r||}|rt|}|j|||dS)Ntitle)rcrrUwrite)rKrirjrYrkrQrOclsrLrLrMrD]s  zJsonArgsValueFormatter._write_schemac Cs|ddddt|dddS)Nr4Tr)seconds)rDr)rKrkrQrorLrLrMrHnsz(JsonArgsValueFormatter._format_timedeltaclose_c Csf|pi}|d}dkr$|dddd|dd|dd |dS||||||ddS) NrlrangerrFr, r )rcrDrA)rKrirqrPrQrOrnrLrLrMrBqs.z'JsonArgsValueFormatter._format_sequencecCs*tt|pi}||j}|d}|jrdnd}|j|d} |p"id} |p)id} |p0id} rEt| dkrE| ddd d } t|D]4\} }| r[d } |jrZ|jd n|j||j|| rst| | krs| | n| }| |d||qI|jr| s|j||j|d|dS) N , rsTitems prefixItemsx-shaper r`)ryr^F ) r r r<r=rUrmrclen enumeraterV)rKrirqrPrQrO indent_newbeforecommafirst items_schema prefix_itemsshapeiv item_schemarLrLrMrAzs0        z(JsonArgsValueFormatter._format_list_likesplit_repr_keyc Cs||j}|d} |jrdnd} |pi}|d} r!| |}|j|d} |di} |D]8\}}| rE|jrB|jdd} n|j| |j| ||||d|j|||d|| |dq3|jrw| sw|jd|j|d|dS) NrurvrsrlT propertiesrzF)r<r=rcrUrmrwrV)rKrirrqrrPrQrOr}before_rrnrrkrrLrLrMr@s,          z$JsonArgsValueFormatter._format_itemscCs6|o|d}|r|d|dn|}|j|dS)aFormat bytes value. Examples: >>> value = b'hello' >>> schema = {'type': 'string', 'x-python-datatype': 'bytes'} >>> _format_bytes(0, value, schema) "hello" >>> schema = {'type': 'string', 'x-python-datatype': 'bytes', 'title': 'MyBytes'} >>> _format_bytes(0, value, schema) MyBytes("hello") rlrrN)rcrUrmrKrkrQrOrnoutputrLrLrMrCs z$JsonArgsValueFormatter._format_bytescCs>|o|d}|r|dt|dnt|}|j|dS)aFormat string value. Examples: >>> value = 'hello' >>> schema = {'type': 'string', 'x-python-datatype': 'str'} >>> _format_string(0, value, schema) "hello" >>> schema = {'type': 'string', 'x-python-datatype': 'str', 'title': 'MyString'} >>> _format_string(0, value, schema) MyString("hello") rlrrN)rcreprrUrmrrLrLrMr>s z%JsonArgsValueFormatter._format_stringcCs:|o|d}|r|d|dnt|}|j|dS)aFormat number value. Supports both integer and float types. Examples: >>> value = 42 >>> schema = {'type': 'integer', 'x-python-datatype': 'int'} >>> _format_number(0, value, schema) 42 >>> schema = {'type': 'number', 'x-python-datatype': 'float', 'title': 'MyNumber'} >>> _format_number(0, value, schema) MyNumber(42) rlrrN)rcrrUrmrrLrLrMr?s z%JsonArgsValueFormatter._format_numbercolumns list[Any]indicesrowsreal_column_countreal_row_countc CsNddd}t|}||kr||d}d gd d |D}g}t|D]\} } ||kr/|| d} || gd d | D} || q"t||krQ||dgt|}d d t|g|D} |jdddt| |Dd|jddd| Dd|D]} |jdddt| | Ddq|jd|d|ddS)aInspired by https://gist.github.com/lonetwin/4721748. >>> columns = ['col1', 'col2', 'col4', 'col5'] >>> indices = ['a', 'b', 'd', 'e'] >>> rows = [[1, 2, 4, 5], [2, 4, 8, 10], [4, 8, 16, 20], [5, 10, 20, 25]] >>> real_column_count = 5 >>> real_rows_count = 5 >>> _format_table(columns, indices, rows, real_column_count, real_rows_count) | col1 | col2 | ... | col4 | col5 ----+------+------+-----+------+----- a | 1 | 2 | ... | 4 | 5 b | 2 | 4 | ... | 8 | 10 ... | ... | ... | ... | ... | ... d | 4 | 8 | ... | 16 | 20 e | 5 | 10 | ... | 20 | 25 [5 rows x 5 columns] rwritemstr | list[str]r\cSs*t|d}|d||g||dS)Nr)r{)rwrmidpointrLrLrMinsert_into_list_middle s zEJsonArgsValueFormatter._format_table..insert_into_list_middlez...r4cSg|]}t|qSrLr.0xrLrLrM z8JsonArgsValueFormatter._format_table..cSrrLrrrLrLrMrrcSsg|] }tt|tdqS))key)r{max)rcolsrLrLrMrsz | cs"|] \}}||dVqdSsNrL)rwidthrlrLrLrM " z7JsonArgsValueFormatter._format_table..rzz-+-css|]}d|VqdS)-NrL)rrrLrLrMr%scsrrrL)rrcdatarLrLrMr)rz [z rows x z columns]N)rwrrrr\r)r{r|appendziprUrmjoin) rKrrrrrr column_countconverted_rowsrrowwidthsrLrLrM _format_tables&     & (z$JsonArgsValueFormatter._format_tablec Cs@|pi}|j|dg|dg||dd|ddddS)Nz x-columnsz x-indiceszx-column-countrz x-row-count)rrrrr)rrcrKrkrQrOrLrLrMrI.s     z)JsonArgsValueFormatter._format_data_framec C,|ddddt|trt|n||dSNr4Tr)rDrdrr fromisoformatrrLrLrMrE8,z#JsonArgsValueFormatter._format_datec Crr)rDrdrrrrrLrLrMrF;rz'JsonArgsValueFormatter._format_datetimec Crr)rDrdrrrrrLrLrMrG>rz#JsonArgsValueFormatter._format_time)rr)rQrrOrRrPr) rPrrYrZrQr[rOrRr\r])rirrjrrYrZrkrrQrrOrRr\r])rkrrQrrorRr\r]) rirrqrrPrrQrrOrRr\r])rirrrrqrrrZrPrrQrrOrRr\r])rkrrQrrOrRr\r]) rrrrrrrrrrr\r])rkrrQrrOrRr\r])__name__ __module__ __qualname____doc__rNrXrVrDrHrBrAr@rCr>r?rrIrErFrGrLrLrLrMr s$ -     # %    < r)r) __future__rrSrrrr functoolsrtypingrr r json_typesr r rutilsrrjson_args_value_formatter!json_args_value_formatter_compactrLrLrLrMs    8