i("dZddlZddlZddlmZddlmZddlmZm Z m Z m Z ddl m Z ddlmZddlmZdd lmZmZeZd ZGd d ZGd deZdZdZddZdZdZddZdZ ddZ!ddZy)zS This module contains a collection of utility classes and functions used in agate. N) OrderedDict)Sequence) ROUND_CEILING ROUND_FLOORDecimal getcontextwraps)slugify)config)warn_duplicate_columnwarn_unnamed_columnc6dtfd}|S)z Dead-simple memoize decorator for instance methods that take no arguments. This is especially useful since so many of our classes are immutable. NcS|SN)selffuncmemos L/var/www/html/strategist-ai/venv/lib/python3.12/site-packages/agate/utils.pywrapperzmemoize..wrappers  KDzr )rrrs` @rmemoizers' D 4[ NrceZdZdZdZdZy) NullOrderzr Dummy object used for sorting in place of None. Sorts as "greater than everything but other nulls." cy)NFrrothers r__lt__zNullOrder.__lt__.src |yy)NFTrrs r__gt__zNullOrder.__gt__1s =rN)__name__ __module__ __qualname____doc__rr!rrrrr(s rrc:eZdZdZdZdZdZdZdZdZ dZ y ) Quantileszn A class representing quantiles (percentiles, quartiles, etc.) for a given column of Number data. c||_yr _quantiles)r quantiless r__init__zQuantiles.__init__=s #rc8|jj|Sr)r* __getitem__)ris rr.zQuantiles.__getitem__@s**1--rc6|jjSr)r*__iter__rs rr1zQuantiles.__iter__Cs''))rc6|jjSr)r*__len__r2s rr4zQuantiles.__len__Fs&&((rc,t|jSr)reprr*r2s r__repr__zQuantiles.__repr__IsDOO$$rc4|j|jk(Srr)rs r__eq__zQuantiles.__eq__Ls%"2"222rcVd}||jdkr td||jdkDr td||jdk(r!tt|jdz S||j|dzk\r|dz }||j|dzk\rt|S)zC Identify which quantile a given value is part of. rz*Value is less than minimum quantile value.z-Value is greater than maximum quantile value.)r* ValueErrorrlen)rvaluer/s rlocatezQuantiles.locateOs  4??1% %IJ J 4??2& &LM M DOOB' '3t/!34 4tq1u-- FAtq1u--qzrN) r"r#r$r%r,r.r1r4r7r9r@rrrr'r'8s*$.*)%3rr'cxt|}|dzdk(r||dzdzdz S|dz}||dz }||}||zdz S)z Finds the median value of a given series of values. :param data_sorted: The values to find the median of. Must be sorted. r<)r>) data_sortedlengthhalfabs rmedianrHdsa F zQVaZA-233 Q;DD1HADA EQ;rc@d}d}tj}|D]s}|*tj|stj|r0|j j \}}}|dz}t||z } | |kDr| }||kDsr|}u||z|kDr||z }|S)z Given a series of values (such as a :class:`.Column`) returns the most significant decimal places present in any value. :param values: The values to analyze. r<rr;)rprecmathisnanisinf normalizeas_tupler>) valuesmax_whole_placesmax_decimal_places precisionr?signdigitsexponentexponent_places whole_placess r max_precisionrYws !!I 1 =DJJu-E1B !&!2!;!;!=fh"R-6{_4 * *+  / /!0  1",,y8&)99 rcld|z}|rtjdnd}djd||d||gS)a Given a number of decimal places creates a formatting string that will display numbers with that precision. :param decimal_places: The number of decimal places :param add_ellipsis: Optionally add an ellipsis symbol at the end of a number 0number_truncation_charsz#,##0.z;-#,##0.)r get_optionjoin)decimal_places add_ellipsisfractionellipsiss rmake_number_formatterrds@^#H?Kv  !:;QSH 77Hh*hQ RRrc|jj}|jj}tt|j|j zt|j|j z}|dkDrMt d|dz z}||z jt}||z jt}||z||zfSt|j |j }t d|dzz} |j| tj|j| tjfS)aD Rounds a pair of minimum and maximum values to form reasonable "round" values suitable for use as axis minimum and maximum values. Values are rounded "out": up for maximum and down for minimum, and "off": to one higher than the first significant digit shared by both. See unit tests for examples. r10r<)rounding) rNrOmaxr>rUrVrto_integral_valuerrquantize) minimummaximummin_bitsmax_bits max_digits multiplier min_fraction max_fraction max_exponentqs r round_limitsrusD  "++-H  "++-H HOOx000 HOOx000J A~T]zA~6 *,???U *,???W  : % : %  x(((*;*;)indexletterscounts r letter_namer|s5 $$G LE 55= !eun%9 ::rc2t|tr|j}n5t|ttfr t |}n|j d|iSt}|D]1\}}t|}|jt|||zdz3|S)z Recursively parse JSON-like Python objects as a dictionary of paths/keys and values. Inspired by JSONPipe (https://github.com/dvxhouse/jsonpipe). /) isinstancedictitemslisttuple enumeratestriprstrupdate parse_object)objpathiteratordkeyr?s rrrs#t99; C$ 'S> 3%% A8 U#h eTCZ#%5678 HrcHt|txrt|t S)zv Returns :code:`True` if the given object is an instance of :class:`.Sequence` that is not also a string. )rrr)rs r issequencers c8 $ AZS-A)AArcNg}t|D]\}}|r8|st|}t||n t|tr|}n t d|}|}d}||vr||zt |dzz}|dz }||vr|r|dkDr t |||j|t|S)a Append a unique identifer to duplicate strings in a given sequence of strings. Identifers are an underscore followed by the occurance number of the specific string. ['abc', 'abc', 'cde', 'abc'] -> ['abc', 'abc_2', 'cde', 'abc_3'] :param column_names: If True, values are treated as column names. Warnings will be thrown if column names are None or duplicates. None values will be replaced with letter indices. z%Column names must be strings or None.rrBr<) rr|rrrr=r appendr) rP column_names separator final_valuesr/r? new_value final_value duplicatess r deduplicatersLf%)5 'N #Ay1E3'!  !HIII  \)#i/#j1n2EEK !OJ\) JN !)[ 9K(-)0  rc ddij||r$tfd|D}t|dStfd|DS)a Given a sequence of strings, returns a standardized version of the sequence. If ``ensure_unique`` is True, any duplicate strings will be appended with a unique identifier. agate uses an underscore as a default separator but this can be changed with kwargs. Any kwargs will be passed to the slugify method in python-slugify. See: https://github.com/un33k/python-slugify r_c36K|]}t|fiywrpslugify.0r? slug_argss r zslugify..7sLE8E7Y7L)rc36K|]}t|fiywrrrs rrzslugify..:sB%%-9-Br)rrr)rP ensure_uniquekwargs new_valuesrs @rr r 'sSc"I VLVLL :;1GHH B6B BBr)F)r])Fr)"r%rKrw collectionsrcollections.abcrdecimalrrrr functoolsr r ragater agate.warnsr robjectdefaultrrr'rHrYrdrur|rrrrrrrs  #$CC'B ($   ))X& F S&R ; .B'TCr