U SZf 8@sddlZddlZddlZddlZddlmZddlmZddl m Z ddZ Gdddej Z Gd d d eZd d Zd dZd#ddZddZddZddZd$ddZd%ddZdd Zd!d"ZdS)&Nreduce) get_module)ImageUriValidatorcCsdd}t||g}|S)z0 Custom cumsum to avoid a numpy import. cSs$t|dkr|gS||d|gSNr)len)axr 7/tmp/pip-unpacked-wheel-5ksk5baj/_plotly_utils/utils.py_reducers zcumsum.._reducerr)r r retr r r cumsum s rcseZdZdZddZfddZddZedd Zed d Z ed d Z eddZ eddZ eddZ eddZeddZeddZZS)PlotlyJSONEncodera9 Meant to be passed as the `cls` kwarg to json.dumps(obj, cls=..) See PlotlyJSONEncoder.default for more implementation information. Additionally, this encoder overrides nan functionality so that 'Inf', 'NaN' and '-Inf' encode to 'null'. Which is stricter JSON than the Python version. cCs|dkr dS|SdS)zU This is used to ultimately *encode* into strict JSON, see `encode` )Infinityz -InfinityNaNNr )selfconstr r r coerce_to_strict%sz"PlotlyJSONEncoder.coerce_to_strictcsxtt||}d|ks$d|ks$|Sztj||jd}Wntk rTtdYn Xtj||j|j |j |j fdSdS)z Load and then dump the result using parse_constant kwarg Note that setting invalid separators will cause a failure at this step. rr)parse_constantzSEncoding into strict JSON failed. Did you set the separators valid JSON separators?) sort_keysindent separatorsN) superrencode_jsonloadsr ValueErrordumpsrritem_separator key_separator)roZ encoded_oZnew_o __class__r r r0s  zPlotlyJSONEncoder.encodec Csd|j|j|j|j|j|j|j|j|jf }|D](}z||WSt k rRYq,Xq,t j ||S)a Accept an object (of unknown type) and try to encode with priority: 1. builtin: user-defined objects 2. sage: sage math cloud 3. pandas: dataframes/series 4. numpy: ndarrays 5. datetime: time/datetime objects Each method throws a NotEncoded exception if it fails. The default method will only get hit if the object is not a type that is naturally encoded by json: Normal objects: dict object list, tuple array str, unicode string int, long, float number True true False false None null Extended objects: float('nan') 'NaN' float('infinity') 'Infinity' float('-infinity') '-Infinity' Therefore, we only anticipate either unknown iterables or values here. ) encode_as_plotlyencode_as_sageencode_as_numpyencode_as_pandasencode_as_datetimeencode_as_dateencode_as_listencode_as_decimal encode_as_pil NotEncodabler JSONEncoderdefault)robjZencoding_methodsZencoding_methodr r r r0Us ! zPlotlyJSONEncoder.defaultcCs(z |WStk r"tYnXdS)z1Attempt to use a builtin `to_plotly_json` method.N)Zto_plotly_jsonAttributeErrorr.r1r r r r%s z"PlotlyJSONEncoder.encode_as_plotlycCst|dr|StdS)z@Attempt to use `tolist` method to convert to normal Python list.tolistN)hasattrr4r.r3r r r r+s z PlotlyJSONEncoder.encode_as_listcCs<td}|st||jkr"t|S||jkr4t|StdS)z@Attempt to convert sage.all.RR to floats and sage.all.ZZ to intszsage.allN)rr.ZRRfloatZZZint)r1Zsage_allr r r r&s  z PlotlyJSONEncoder.encode_as_sagecCsBtddd}|st||jkr"dSt|dr:||jkr:dStdS)z)Attempt to convert pandas.NaT / pandas.NApandasFZ should_loadNNA)rr.ZNaTr5r:)r1r8r r r r(s  z"PlotlyJSONEncoder.encode_as_pandascCsptddd}|st||jjjkr*tdSt||jrh|jj dkrhz| | WSt k rfYnXtdS)z'Attempt to convert numpy.ma.core.maskednumpyFr9nanMN) rr.macoreZmaskedr6 isinstanceZndarrayZdtypekindZdatetime_as_stringr4 TypeError)r1r;r r r r's z!PlotlyJSONEncoder.encode_as_numpycCs(z |WStk r"tYnXdS)z.Convert datetime objects to iso-format stringsN) isoformatr2r.r3r r r r)s z$PlotlyJSONEncoder.encode_as_datetimecCs2z |}Wntk r$tYn Xt|SdS)z=Attempt to convert to utc-iso time string using date methods.N)rCr2r.iso_to_plotly_time_string)r1Z time_stringr r r r*s   z PlotlyJSONEncoder.encode_as_datecCst|tjrt|StdS)z3Attempt to encode decimal by converting it to floatN)r@decimalDecimalr6r.r3r r r r,s z#PlotlyJSONEncoder.encode_as_decimalcCs.td}|dk r&t||jr&t|StdS)z5Attempt to convert PIL.Image.Image to base64 data uriz PIL.ImageN)rr@ZImagerZpil_image_to_urir.)r1imager r r r-s zPlotlyJSONEncoder.encode_as_pil)__name__ __module__ __qualname____doc__rrr0 staticmethodr%r+r&r(r'r)r*r,r- __classcell__r r r#r rs,  %2        rc@s eZdZdS)r.N)rHrIrJr r r r r.sr.cCsj|ddddks(|dddkr0td|dd d d }|d rZ|d d S|d d SdS)z=Remove timezone info and replace 'T' delimeter with ' ' (ws).-Nz00:00+rz]Plotly won't accept timestrings with timezone info. All timestrings are assumed to be in UTC.z-00:00z+00:00z T00:00:00T )split Exceptionreplaceendswith)Z iso_stringr r r rDs(  rDc sfdd}|S)Ncs0tjdddks,|jdk r,|jjf|_|S)N)rOrX)sys version_inforKformat)funcnamesr r _decorators z template_doc.._decoratorr )r^r_r r]r template_docs r`FcCsdd}t|||dS)Nc SsPtd|}tt|D].}zt||||<Wqtk rDYqXqt|S)Nz(\d+))rerTrangerr7rtuple)vZv_partsir r r key s z"_natural_sort_strings..key)rfreversesorted)valsrgrfr r r _natural_sort_strings s rkcCs&tddd}|rt|jf}ntf}|S)Nr;Fr9)rr7integer)npZint_typer r r _get_int_types   rncs>t|dkr|S|tddtfdd|}t||S)ar Split all the strings in ss at any of the characters in chars. Example: >>> ss = ["a.string[0].with_separators"] >>> chars = list(".[]_") >>> split_multichar(ss, chars) ['a', 'string', '0', '', 'with', 'separators'] :param (list) ss: A list of strings. :param (list) chars: Is a list of chars (note: not a string). rcSs||SNr )r yr r r 0z!split_multichar..cs |Sro)rT)r cr r rq0rr)rpoprmapsplit_multichar)sscharsr rsr rw s rwc Cs<ttddttt|tdgttt|ddS)a  Given a list of strings split using split_multichar, return a list of integers representing the indices of the first character of every string in the original string. Example: >>> ss = ["a.string[0].with_separators"] >>> chars = list(".[]_") >>> ss_split = split_multichar(ss, chars) >>> ss_split ['a', 'string', '0', '', 'with', 'separators'] >>> split_string_positions(ss_split) [0, 2, 9, 11, 12, 17] :param (list) ss: A list of strings. cSs|d|dS)Nrr )tr r r rqGrrz(split_string_positions..rNr)listrvziprbrr)rxr r r split_string_positions4s ,r~rz^Tc Csddtt|d||D}d}|dkr\|D]&}t|D]} ||| }|||<q>q2n&t|D]} |||| }|||<qdd|} |r| d|d} | S)a Return a string that is whitespace except at p[i] which is replaced with char. If i is None then all the indices of the string in p are replaced with char. Example: >>> ss = ["a.string[0].with_separators"] >>> chars = list(".[]_") >>> ss_split = split_multichar(ss, chars) >>> ss_split ['a', 'string', '0', '', 'with', 'separators'] >>> ss_pos = split_string_positions(ss_split) >>> ss[0] 'a.string[0].with_separators' >>> display_string_positions(ss_pos,4) ' ^' >>> display_string_positions(ss_pos,4,offset=1,length=3,char="~",trim=False) ' ~~~ ' >>> display_string_positions(ss_pos) '^ ^ ^ ^^ ^' :param (list) p: A list of integers. :param (integer|None) i: Optional index of p to display. :param (integer) offset: Allows adding a number of spaces to the replacement. :param (integer) length: Allows adding a replacement that is the char repeated length times. :param (str) char: allows customizing the replacement character. :param (boolean) trim: trims the remaining whitespace if True. cSsg|]}dqS)rSr ).0_r r r jsz,display_string_positions..rzrNrQ)rbmaxjoin) preoffsetlengthcharZtrimsZmaxaddrZp_lrr r r display_string_positionsMs"     rcCsxdd}|r|t|||St|s*|Sttt|dkrN|t|dgSGddd}tttt|||dgS)a Given a list of strings, some of which are the empty string "", replace the empty strings with c and combine them with the closest non-empty string on the left or "" if it is the first string. Examples: for c="_" ['hey', '', 'why', '', '', 'whoa', '', ''] -> ['hey_', 'why__', 'whoa__'] ['', 'hi', '', "I'm", 'bob', '', ''] -> ['_', 'hi_', "I'm", 'bob__'] ['hi', "i'm", 'a', 'good', 'string'] -> ['hi', "i'm", 'a', 'good', 'string'] Some special cases are: [] -> [] [''] -> [''] ['', ''] -> ['_'] ['', '', '', ''] -> ['___'] If reverse is true, empty strings are combined with closest non-empty string on the right or "" if it is the last string. cSsdd|DdddS)NcSsg|]}|dddqS)Nrr )rrr r r rsz5chomp_empty_strings.._rev..rr )rr r r _revsz!chomp_empty_strings.._revrrzc@seZdZddZddZdS)z%chomp_empty_strings.._ChompercSs ||_dSrors)rrtr r r __init__sz.chomp_empty_strings.._Chomper.__init__cSs6t|dkr(|dd|d|jgS||gSdSr)rrt)rr rpr r r __call__s z.chomp_empty_strings.._Chomper.__call__N)rHrIrJrrr r r r _ChompersrrQ)chomp_empty_stringsrsumrvr|filterr)stringsrtrgrrr r r r{src Cst|t|krt||St|dkr.t|Stt|d}t|D]f\}}|dg}t|D]F\}}||dd}||d} ||||k} |t|| | q`|}qF|dS)Nrrzr)r levenshteinrb enumerateappendmin) s1s2Z previous_rowreZc1Z current_rowjc2Z insertionsZ deletionsZ substitutionsr r r rs    rcsfdd}t||ddS)Ncst||fSro)r)rstringr r _keysz!find_closest_string.._key)rfrrh)rrrr rr find_closest_strings r)F)NrrzrT)F)rEjsonrrYra functoolsrZ_plotly_utils.optional_importsrZ_plotly_utils.basevalidatorsrrr/rrUr.rDr`rkrnrwr~rrrrr r r r s(   R   . 0