B eJd:8@sddlZddlZddlZddlZddlmZddlmZddl m Z ej dkZ ddZ Gdd d ejZGd d d eZd d ZddZd$ddZddZddZddZd%ddZd&ddZd d!Zd"d#ZdS)'N)reduce) get_module)ImageUriValidator)cCsdd}t||g}|S)z0 Custom cumsum to avoid a numpy import. cSs$t|dkr|gS||d|gS)Nr)len)axr E/var/www/html/venv/lib/python3.7/site-packages/_plotly_utils/utils.py_reducers zcumsum.._reducer)r)r r retr r r cumsums 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$|Sytj||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 r3s zPlotlyJSONEncoder.encodec Csb|j|j|j|j|j|j|j|j|jf }x*|D]"}y||St k rNYq.Xq.Wt 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 r/Xs!  zPlotlyJSONEncoder.defaultcCs&y|Stk r tYnXdS)z1Attempt to use a builtin `to_plotly_json` method.N)Zto_plotly_jsonAttributeErrorr-)r0r r r r$sz"PlotlyJSONEncoder.encode_as_plotlycCst|dr|StdS)z@Attempt to use `tolist` method to convert to normal Python list.tolistN)hasattrr2r-)r0r 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)r0Zsage_allr r r r%s  z PlotlyJSONEncoder.encode_as_sagecCsBtddd}|st||jkr"dSt|dr:||jkr:dStdS)z)Attempt to convert pandas.NaT / pandas.NApandasF) should_loadNNA)rr-ZNaTr3r8)r0r6r r r r's  z"PlotlyJSONEncoder.encode_as_pandascCsntddd}|st||jjjkr*tdSt||jrf|jj dkrfy| | St k rdYnXtdS)z'Attempt to convert numpy.ma.core.maskednumpyF)r7nanMN) rr-macoreZmaskedr4 isinstanceZndarrayZdtypekindZdatetime_as_stringr2 TypeError)r0r9r r r r&s z!PlotlyJSONEncoder.encode_as_numpycCs&y|Stk r tYnXdS)z.Convert datetime objects to iso-format stringsN) isoformatr1r-)r0r r r r(sz$PlotlyJSONEncoder.encode_as_datetimecCs2y |}Wntk r$tYn Xt|SdS)z=Attempt to convert to utc-iso time string using date methods.N)rAr1r-iso_to_plotly_time_string)r0Z time_stringr r r r)s   z PlotlyJSONEncoder.encode_as_datecCst|tjrt|StdS)z3Attempt to encode decimal by converting it to floatN)r>decimalDecimalr4r-)r0r 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-)r0imager r r r,s zPlotlyJSONEncoder.encode_as_pil)__name__ __module__ __qualname____doc__rrr/ staticmethodr$r*r%r'r&r(r)r+r, __classcell__r r )r#r rs  %2       rc@s eZdZdS)r-N)rFrGrHr 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).-Nrz00: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 rBs(  rBc sfdd}|S)Ncs0tjdddks,|jdk r,|jjf|_|S)N)rrU)sys version_inforIformat)func)namesr r _decorators z template_doc.._decoratorr )rZr[r )rZr template_docs r\FcCsdd}t|||dS)Nc SsTtd|}x>tt|D].}yt||||<Wqtk rFYqXqWt|S)Nz(\d+))rerQrangerr5rtuple)vZv_partsir r r key s  z"_natural_sort_strings..key)rbreverse)sorted)valsrcrbr r r _natural_sort_strings s rfcCs&tddd}|rt|jf}ntf}|S)Nr9F)r7)rr5integer)npZint_typer r r _get_int_types   rics>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||S)Nr )r yr r r 3z!split_multichar..cs |S)N)rQ)r )cr r rk3rl)rpoprmapsplit_multichar)sscharsr )rmr rp#s rpc 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 rkJrlz(split_string_positions..rNr)listrozipr^rr)rqr r r split_string_positions7srwrs^Tc Csddtt|d||D}d}|dkrdx^|D]*}x$t|D]} ||| }|||<qBWq4Wn*x(t|D]} |||| }|||<qnWd|} |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)rPr ).0_r r r msz,display_string_positions..rsrNrN)r^maxjoin) praoffsetlengthcharZtrimsZmaxaddrZp_lrr r r display_string_positionsPs"    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 )ryrr r r r{sz5chomp_empty_strings.._rev..rr )rr r r _revsz!chomp_empty_strings.._revrrsc@seZdZddZddZdS)z%chomp_empty_strings.._ChompercSs ||_dS)N)rm)rrmr r r __init__sz.chomp_empty_strings.._Chomper.__init__cSs6t|dkr(|dd|d|jgS||gSdS)Nrr)rrm)rr rjr r r __call__s z.chomp_empty_strings.._Chomper.__call__N)rFrGrHrrr r r r _ChompersrrN)chomp_empty_stringsrsumrorufilterr)stringsrmrcrrr r r r~src Cst|t|krt||St|dkr.t|Stt|d}xvt|D]j\}}|dg}xRt|D]F\}}||dd}||d} ||||k} |t|| | qdW|}qHW|dS)Nrrsr)r levenshteinr^ enumerateappendmin) s1s2Z previous_rowrac1Z current_rowjc2 insertionsZ deletionsZ substitutionsr r r rs    rcsfdd}t||ddS)Ncst||fS)N)r)r)stringr r _keysz!find_closest_string.._key)rbr)rd)rrrr )rr find_closest_strings r)F)NrrsrxT)F)rCjsonrrVr] functoolsrZ_plotly_utils.optional_importsrZ_plotly_utils.basevalidatorsrrWZ PY36_OR_LATERrr.rrRr-rBr\rfrirprwrrrrr r r r s*    R   . 0