U Tgc @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#sr  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__rrrrr1Bsr1c@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|ts|tjdtdd|tk rD|f|}|tk rV|f|}|tk rh|f|}tj|f|dS| 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 )) isinstancestrintwarningswarnDeprecationWarningr1 ValueError__init__countrindexlenr7r8r9linenocolno) selfr7r8r9r:rIrJZ coord_reprerrmsgrrrrERsF      zTOMLDecodeError.__init__N)r2r3r4r5r1rErrrrr6Gs  r6 parse_floatz IO[bytes]rzdict[str, Any])__fprNreturncCs@|}z |}Wntk r2tddYnXt||dS)z%Parse TOML from a binary file object.zEFile must be opened in binary mode, e.g. use `open('foo.toml', 'rb')`NrM)readdecodeAttributeError TypeErrorloads)rOrNbsrrrloads rXr?)__srNrPc Csz|dd}Wn0ttfk r@tdt|jddYnXd}ttt}d}t|}t ||t }z ||}Wnt k rYqYnX|dkr|d7}q`|t krt |||||}t ||t }n|d krDz||d}Wnt k rd}YnX|j|d kr&t|||\}}nt|||\}}t ||t }n|d krZtd ||t||}z ||}Wnt k rYqYnX|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)replacerSrTtyper4Output NestedDictFlagsmake_safe_parse_float skip_charsr% IndexErrorr)key_value_ruleflagsfinalize_pendingcreate_list_rulecreate_dict_ruler6 skip_commentdatadict)rYrNsrcr9outheaderchar second_charrrrrUs`              rUc@seZdZUdZdZded<dZded<ddd d Zd d dd ddZddddZ d ddddZ d d dddddZ d d dd ddZ dS)raz)Flags that map to parsed keys/namespaces.rrFROZENr EXPLICIT_NESTNonerPcCsi|_t|_dSr)_flagsset_pending_flagsrKrrrrEszFlags.__init__rr@)keyflagrPcCs|j||fdSr)rxaddrKrzr{rrr add_pendingszFlags.add_pendingcCs.|jD]\}}|j||ddq|jdS)NF recursive)rxrwclearr}rrrrgszFlags.finalize_pendingrzrPcCsF|j}|ddD]}||kr$dS||d}q||dddS)Nnested)rvpop)rKrzcontkrrr unset_alls zFlags.unset_allbool)rzr{rrPcCs|j}|dd|d}}|D],}||kr@ttid||<||d}q ||krjttid||<|||rxdnd|dS)Nr)rfrecursive_flagsrrrrf)rvrwr|)rKrzr{rr key_parentkey_stemrrrrrwsz Flags.setcCs|sdS|j}|ddD]4}||kr,dS||}||dkrFdS|d}q|d}||kr||}||dkp~||dkSdS)NFrrTrrf)rv)rKrzr{rrZ inner_contrrrris_s  z Flags.is_N) r2r3r4r5rr__annotations__rsrEr~rgrrwrrrrrras    rac@sBeZdZddddZddddd d d d Zddd ddZdS)r`rtrucCs i|_dSr)rlryrrrrEszNestedDict.__init__T access_listsrrrl)rzrrPcCsT|j}|D]D}||kri||<||}|rlistKeyError)rKrzrrrrrrget_or_create_nests  zNestedDict.get_or_create_nestrcCsV||dd}|d}||krH||}t|tsrrappend)rKrzrZlast_keylist_rrrappend_nest_to_list%s  zNestedDict.append_nest_to_listN)r2r3r4rErrrrrrr`sr`c@seZdZUded<ded<dS)r_r`rkrarfN)r2r3r4rrrrrr_1s r_rz Iterable[str])rmr9charsrPcCs4z|||kr|d7}qWntk r.YnX|S)Nr )rd)rmr9rrrrrc6s  rczfrozenset[str]r)rmr9expecterror_on error_on_eofrPcCsz|||}Wn4tk rDt|}|r@td|||dYnX||||s|||krn|d7}qXtd|||||S)Nz Expected r zFound invalid character )indexrDrHr6 isdisjoint)rmr9rrrnew_posrrr skip_until?s  r)rmr9rPcCsHz ||}Wntk r$d}YnX|dkrDt||ddtddS|S)Nr\r r&Frr)rdrr$)rmr9rprrrrjUs  rjcCs,|}t||t}t||}||kr|SqdSr)rcr'rj)rmr9Zpos_before_skiprrrskip_comments_and_array_wsas   rztuple[Pos, Key])rmr9rnrPcCs|d7}t||t}t||\}}|j|tjsB|j|tjrVtd|d|||jj |tjddz|j |Wn"t k rtd||dYnX| d|std|||d|fS) Nr zCannot declare z twiceFrCannot overwrite a value]z.Expected ']' at the end of a table declaration)rcr% parse_keyrfrrarsrrr6rwrkrr startswithrmr9rnrzrrrrijs"   ricCs|d7}t||t}t||\}}|j|tjrDtd||||j||jj |tj ddz|j |Wn"t k rtd||dYnX|d|std|||d|fS)Nr;"Cannot mutate immutable namespace Frrz]]z0Expected ']]' at the end of an array declaration)rcr%rrfrrarrr6rrwrsrkrrrrrrrrh~s$   rhr)rmr9rnrorNrPc s,t|||dd\}}ddd}}|}fddtdtD} | D]6} |j| tjr|td| |||j| tjqV|j|tj rtd|||z|j |} Wn"t k rtd ||dYnX|| krtd ||t |ttfr |jjtj d d || |<|S) Nr)nest_lvlrc3s|]}d|VqdSrrrrorzrrrsz!key_value_rule..r zCannot redefine namespace rrTr)parse_key_value_pairrangerHrfrrarsr6r~rrrkrrr>rlrrw) rmr9rnrorNvaluerrZabs_key_parentZrelative_path_cont_keysZcont_keynestrrrres0 rer@ztuple[Pos, Key, Any])rmr9rNrrPcCsxt||\}}z ||}Wntk r2d}YnX|dkrHtd|||d7}t||t}t||||\}}|||fS)N=z,Expected '=' after a key in a key/value pairr )rrdr6rcr% parse_value)rmr9rNrrzrprrrrrs    rcCst||\}}|f}t||t}z ||}Wntk rDd}YnX|dkrV||fS|d7}t||t}t||\}}||f7}t||t}q dS)N.r )parse_key_partrcr%rd)rmr9Zkey_partrzrprrrrs     rztuple[Pos, str]cCsz ||}Wntk r$d}YnX|tkrN|}t||t}||||fS|dkr`t||S|dkrrt||Std||dS)NrZr.z(Invalid initial character for a key part)rdr(rcparse_literal_strparse_one_line_basic_strr6)rmr9rp start_posrrrrs     rcCs|d7}t||ddS)Nr F multiline)parse_basic_strrmr9rrrrsrztuple[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)rrrrr6)rmr9rNrarrayvalcrrr parse_arrays"         rztuple[Pos, dict]c Csb|d7}t}t}t||t}|d|r:|d|jfSt||||\}}}|dd|d}} ||tjrt d|||z|j |dd} Wn"t k rt d||dYnX| | krt d| |||| | <t||t}|||d} | dkr|d|jfS| d kr&t d ||t |tt frH|j|tjd d |d7}t||t}q:dS) Nr }rrFrrzDuplicate inline table key rzUnclosed inline tableTr)r`rarcr%rrlrrrrr6rrr>rrw) rmr9rNrZ nested_dictrfrzrrrrrrrrparse_inline_tables6      rFr)rmr9rrPcCs|||d}|d7}|r|dkr|dkr~t||t}z ||}Wntk r`|dfYSX|dkrvtd|||d7}t||t}|dfS|dkrt||d S|d krt||d Sz|t|fWStk rtd||dYnXdS) Nr;>\ \ \ rr&zUnescaped '\' in a stringr z\uz\U)rcr%rdr6r'parse_hex_charr0r)rmr9rZ escape_idrprrrparse_basic_str_escape*s,       rcCst||ddS)NTr)rrrrr parse_basic_str_escape_multilineGsr)rmr9hex_lenrPcCsd||||}t||ks&t|s2td||||7}t|d}t|sXtd|||t|fS)NzInvalid hex valuez/Escaped character is not a Unicode scalar value)rHr* issupersetr6r@is_unicode_scalar_valuer)rmr9rhex_strZhex_intrrrrKs  rcCs2|d7}|}t||dtdd}|d|||fS)Nr rZTr)rr")rmr9rrrrrXsr)rmr9literalrPcCs|d7}|d|r|d7}|rLd}t||dtdd}|||}|d}nd}t||dd \}}|||sv||fS|d7}|||s|||fS|d7}|||d fS) Nr&r rZ'''Trr.rr;)rrr#r)rmr9rdelimZend_posresultrrrparse_multiline_stras.      rc Cs|rt}t}nt}t}d}|}z ||}Wn"tk rLtd||dYnX|dkr|sr|d||||fS|d|r|d||||fS|d7}q|dkr||||7}|||\}}||7}|}q||krtd||||d7}qdS) NrzUnterminated stringr.r """rr/zIllegal character )r!rr rrdr6r) rmr9rrZ parse_escapesrrrpZ parsed_escaperrrrs6  rztuple[Pos, Any]c Cs|tkrtdtdz ||}Wntk r<d}YnX|dkrj|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 |}Wn2t k r\}zt d|||W5d}~XYnX||fSt ||}|r|t|fSt ||} | r| t| |fS|||d} | dkr|d|| fS|||d } | dkr|d || fSt d||dS)Nz;TOML inline arrays/tables are nested more than the allowed z levelsr.rF)rrZrTttruerffalser[r {zInvalid date or datetimer>naninf>-inf+inf+nan-nanz Invalid value)rRecursionErrorrdrrrrrrr matchr rDr6endr rr r) rmr9rNrrpZdatetime_matchZ datetime_objeZlocaltime_matchZ number_matchZ first_threeZ first_fourrrrrs\                    r) codepointrPcCs0d|kodknp.d|ko*dkSS)Nriiir)rrrrrsr)rNrPcs$tkr 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_strrPcs"|}t|ttfrtd|S)Nz*parse_float must not return dicts or lists)r>rlrrD)rZ float_valuerMrrsafe_parse_floatsz/make_safe_parse_float..safe_parse_float)float)rNrrrMrrbs rb)M __future__rcollections.abcrstringsystypesrtypingrrrrrA_rer r r r rr_typesrrrgetrecursionlimitrr frozensetrrrr r!r"r#r$r%r' ascii_lettersdigitsr(r) hexdigitsr*r0r1rDr6rrXrUrar`r_rcrrjrrirhrerrrrrrrrrrrrrrrbrrrrsz     *    ; G:"   "#   L