a $lJhc @sUddlmZddlmZddlZddlZddlmZddlm Z m Z m Z m Z ddl Z ddlmZmZmZmZmZmZddlmZmZmZeZd ed <ed d ed DeedBZd ed<eedZ d ed<eedZ!d ed<e Z"d ed<e!Z#d ed<e Z$d ed<edZ%d ed<e%edBZ&d ed<eej'ej(dZ)d ed<e)edBZ*d ed<eej+Z,d ed<ed ddd!d"d#d$d%Z-d ed&<Gd'd(d(Z.Gd)d*d*e/Z0e1d+d,d-d.d/d0d1Z2e1d+d2d-d.d3d4d5Z3Gd6d7d7Z4Gd8d9d9Z5Gd:d;d;e Z6d2dd?d@Z7d2dd2d#r  ASCII_CTRL ILLEGAL_BASIC_STR_CHARSz !ILLEGAL_MULTILINE_BASIC_STR_CHARSILLEGAL_LITERAL_STR_CHARS#ILLEGAL_MULTILINE_LITERAL_STR_CHARSILLEGAL_COMMENT_CHARSz TOML_WS TOML_WS_AND_NEWLINEz-_BARE_KEY_CHARSz"'KEY_INITIAL_CHARSHEXDIGIT_CHARS  "\)z\bz\tz\nz\fz\rz\"z\\BASIC_STR_ESCAPE_REPLACEMENTSc@seZdZdZdS)DEPRECATED_DEFAULTzjSentinel to be used as default arg during deprecation period of TOMLDecodeError's free-form arguments.N)__name__ __module__ __qualname____doc__rrrrr2Bsr2c@s,eZdZdZeeefdddddddZdS) TOMLDecodeErrora:An error raised if a document is not valid TOML. Adds the following attributes to ValueError: msg: The unformatted error message doc: The TOML document being parsed pos: The index of doc where parsing failed lineno: The line corresponding to pos colno: The column corresponding to pos zstr | type[DEPRECATED_DEFAULT]zPos | type[DEPRECATED_DEFAULT]r)msgdocposargsc Gs|s"t|tr"t|tr"t|tstjdtdd|turF|g|R}|turZ|g|R}|turn|g|R}tj|g|RdS| dd|d}|dkr|d}n|| dd|}|t |krd}nd|d |}|d |d }t||||_ ||_ ||_||_||_dS) Nz{Free-form arguments for TOMLDecodeError are deprecated. Please set 'msg' (str), 'doc' (str) and 'pos' (int) arguments only.) stacklevelr'rr zend of documentzline z , column z (at )) isinstancestrintwarningswarnDeprecationWarningr2 ValueError__init__countrindexlenr8r9r:linenocolno) selfr8r9r:r;rJrKZ coord_reprerrmsgrrrrFRsF      zTOMLDecodeError.__init__N)r3r4r5r6r2rFrrrrr7Gs  r7 parse_floatz IO[bytes]rzdict[str, Any])__fprOreturncCs>|}z |}Wnty0tddYn0t||dS)z%Parse TOML from a binary file object.zEFile must be opened in binary mode, e.g. use `open('foo.toml', 'rb')`NrN)readdecodeAttributeError TypeErrorloads)rPrObsrrrloads  rYr@)__srOrQc Csz|dd}Wn.ttfy>tdt|jddYn0d}ttt}d}t|}t ||t }z ||}Wnt yYqYn0|dkr|d7}q^|t vrt |||||}t ||t }n|d kr>z||d}Wnt yd}Yn0|j|d kr t|||\}}nt|||\}}t ||t }n|d krTtd ||t||}z ||}Wnt yYqYn0|dkrtd |||d7}q^|jjS) zParse TOML from a string.z r'zExpected str object, not ''Nrrr [#zInvalid statementz5Expected newline or end of document after a statement)replacerTrUtyper5Output NestedDictFlagsmake_safe_parse_float skip_charsr& IndexErrorr*key_value_ruleflagsfinalize_pendingcreate_list_rulecreate_dict_ruler7 skip_commentdatadict)rZrOsrcr:outheaderchar second_charrrrrVs\                rVc@seZdZUdZdZded<dZded<ddd d Zd d dd ddZddddZ d ddddZ d d dddddZ d d dd ddZ dS)rbz)Flags that map to parsed keys/namespaces.rrFROZENr EXPLICIT_NESTNonerQcCsi|_t|_dSr)_flagsset_pending_flagsrLrrrrFszFlags.__init__rrA)keyflagrQcCs|j||fdSr)ryaddrLr{r|rrr add_pendingszFlags.add_pendingcCs.|jD]\}}|j||ddq|jdS)NF recursive)ryrxclearr~rrrrhszFlags.finalize_pendingr{rQcCsF|j}|ddD]}||vr$dS||d}q||dddS)Nnested)rwpop)rLr{contkrrr unset_alls zFlags.unset_allbool)r{r|rrQcCs|j}|dd|d}}|D],}||vr@ttid||<||d}q ||vrjttid||<|||rxdnd|dS)Nr)rgrecursive_flagsrrrrg)rwrxr})rLr{r|rr key_parentkey_stemrrrrrxsz Flags.setcCs|sdS|j}|ddD]4}||vr,dS||}||dvrFdS|d}q|d}||vr||}||dvp~||dvSdS)NFrrTrrg)rw)rLr{r|rrZ inner_contrrrris_s  z Flags.is_N) r3r4r5r6rs__annotations__rtrFrrhrrxrrrrrrbs    rbc@sBeZdZddddZddddd d d d Zddd ddZdS)rarurvcCs i|_dSr)rmrzrrrrFszNestedDict.__init__T access_listsrrrm)r{rrQcCsT|j}|D]D}||vri||<||}|rtd|||dYn0||||s|||vrl|d7}qVtd|||||S)Nz Expected r zFound invalid character )indexrErIr7 isdisjoint)rnr:rrrnew_posrrr skip_until?s   r)rnr:rQcCsFz ||}Wnty"d}Yn0|dkrBt||ddtddS|S)Nr]r r'Frr)rerr%)rnr:rqrrrrkUs   rkcCs,|}t||t}t||}||kr|SqdSr)rdr(rk)rnr:Zpos_before_skiprrrskip_comments_and_array_wsas   rztuple[Pos, Key])rnr:rorQcCs|d7}t||t}t||\}}|j|tjsB|j|tjrVtd|d|||jj |tjddz|j |Wn t ytd||dYn0| d|std|||d|fS) Nr zCannot declare z twiceFrCannot overwrite a value]z.Expected ']' at the end of a table declaration)rdr& parse_keyrgrrbrtrsr7rxrlrr startswithrnr:ror{rrrrjjs    rjcCs|d7}t||t}t||\}}|j|tjrDtd||||j||jj |tj ddz|j |Wn t ytd||dYn0|d|std|||d|fS)Nr<"Cannot mutate immutable namespace Frrz]]z0Expected ']]' at the end of an array declaration)rdr&rrgrrbrsr7rrxrtrlrrrrrrrri~s     rir)rnr:rorprOrQc s*t|||dd\}}ddd}}|}fddtdtD} | D]6} |j| tjr|td| |||j| tjqV|j|tj rtd|||z|j |} Wn t ytd ||dYn0|| vrtd ||t |ttfr|jjtj d d || |<|S) Nr)nest_lvlrc3s|]}d|VqdSrrrrpr{rrrrz!key_value_rule..r zCannot redefine namespace rrTr)parse_key_value_pairrangerIrgrrbrtr7rrsrlrrr?rmrrx) rnr:rorprOvaluerrZabs_key_parentZrelative_path_cont_keysZcont_keynestrrrrfs,   rfrAztuple[Pos, Key, Any])rnr:rOrrQcCsvt||\}}z ||}Wnty0d}Yn0|dkrFtd|||d7}t||t}t||||\}}|||fS)N=z,Expected '=' after a key in a key/value pairr )rrer7rdr& parse_value)rnr:rOrr{rqrrrrrs     rcCst||\}}|f}t||t}z ||}WntyBd}Yn0|dkrT||fS|d7}t||t}t||\}}||f7}t||t}q dS)N.r )parse_key_partrdr&re)rnr:Zkey_partr{rqrrrrs      rztuple[Pos, str]cCsz ||}Wnty"d}Yn0|tvrL|}t||t}||||fS|dkr^t||S|dkrpt||Std||dS)Nr[r/z(Invalid initial character for a key part)rer)rdparse_literal_strparse_one_line_basic_strr7)rnr:rq start_posrrrrs      rcCs|d7}t||ddS)Nr F multiline)parse_basic_strrnr:rrrrsrztuple[Pos, list]cCs|d7}g}t||}|d|r.|d|fSt||||\}}||t||}|||d}|dkrx|d|fS|dkrtd|||d7}t||}|d|r.|d|fSq.dS)Nr r,zUnclosed array)rrrrr7)rnr:rOrarrayvalcrrr parse_arrays"         rztuple[Pos, dict]c Cs`|d7}t}t}t||t}|d|r:|d|jfSt||||\}}}|dd|d}} ||tjrt d|||z|j |dd} Wn t yt d||dYn0| | vrt d| |||| | <t||t}|||d} | dkr|d|jfS| d kr$t d ||t |tt frF|j|tjd d |d7}t||t}q:dS) Nr }rrFrrzDuplicate inline table key rzUnclosed inline tableTr)rarbrdr&rrmrrrsr7rrr?rrx) rnr:rOrZ nested_dictrgr{rrrrrrrrparse_inline_tables6       rFr)rnr:rrQcCs|||d}|d7}|r|dvr|dkr|t||t}z ||}Wnty^|dfYS0|dkrttd|||d7}t||t}|dfS|dkrt||d S|d krt||d Sz|t|fWStytd||dYn0dS) Nr<>z\ z\ \ rr'zUnescaped '\' in a stringr z\uz\U)rdr&rer7r(parse_hex_charr1r)rnr:rZ escape_idrqrrrparse_basic_str_escape*s,         rcCst||ddS)NTr)rrrrr parse_basic_str_escape_multilineGsr)rnr:hex_lenrQcCsd||||}t||ks&t|s2td||||7}t|d}t|sXtd|||t|fS)NzInvalid hex valuez/Escaped character is not a Unicode scalar value)rIr+ issupersetr7rAis_unicode_scalar_valuer)rnr:rhex_strZhex_intrrrrKs  rcCs2|d7}|}t||dtdd}|d|||fS)Nr r[Tr)rr#)rnr:rrrrrXs  r)rnr:literalrQcCs|d7}|d|r|d7}|rLd}t||dtdd}|||}|d}nd}t||dd \}}|||sv||fS|d7}|||s|||fS|d7}|||d fS) Nr'r r['''Trr/rr<)rrr$r)rnr:rdelimZend_posresultrrrparse_multiline_stras.      rc Cs|rt}t}nt}t}d}|}z ||}Wn tyJtd||dYn0|dkr|sp|d||||fS|d|r|d||||fS|d7}q|dkr||||7}|||\}}||7}|}q||vrtd||||d7}qdS) NrzUnterminated stringr/r """rr0zIllegal character )r"rr!rrer7r) rnr:rrZ parse_escapesrrrqZ parsed_escaperrrrs6   rztuple[Pos, Any]c Cs|tkrtdtdz ||}Wnty:d}Yn0|dkrh|d|r^t||ddSt||S|dkr|d|rt||d dSt||S|d kr|d |r|d d fS|d kr|d|r|ddfS|dkrt||||dS|dkr t||||dSt ||}|rjz t |}Wn4t y\}zt d|||WYd}~n d}~00||fSt ||}|r|t|fSt ||} | r| t| |fS|||d} | dvr|d|| fS|||d } | dvr|d || fSt d||dS)Nz;TOML inline arrays/tables are nested more than the allowed z levelsr/rF)rr[rTttruerffalser\r {zInvalid date or datetimer>naninf>z-infz-nanz+infz+nanz Invalid value)rRecursionErrorrerrrrrrr matchr rEr7endr rr r) rnr:rOrrqZdatetime_matchZ datetime_objeZlocaltime_matchZ number_matchZ first_threeZ first_fourrrrrs`              $     r) codepointrQcCs0d|kodknp.d|ko*dkSS)Nriiir)rrrrrsr)rOrQcs$tur tSdddfdd }|S)a%A decorator to make `parse_float` safe. `parse_float` must not return dicts or lists, because these types would be mixed with parsed TOML tables and arrays, thus confusing the parser. The returned decorated callable raises `ValueError` instead of returning illegal types. r@r) float_strrQcs"|}t|ttfrtd|S)Nz*parse_float must not return dicts or lists)r?rmrrE)rZ float_valuerNrrsafe_parse_floatsz/make_safe_parse_float..safe_parse_float)float)rOrrrNrrcs rc)M __future__rcollections.abcrstringsystypesrtypingrrrrrB_rer r r r rr_typesrrrgetrecursionlimitrr frozensetrrrr!r"r#r$r%r&r( ascii_lettersdigitsr)r* hexdigitsr+r1r2rEr7rrYrVrbrar`rdrrkrrjrirfrrrrrrrrrrrrrrrcrrrrsx    *    ; G:"   "#   L