U Te@sxddlmZddlmZddlmZddlmZddlmZddlm Z GdddZ Gd d d Z Gd d d e Z d S)) annotations)copy)Any) ParseError)UnexpectedCharErrorTOMLCharc@s8eZdZddddddddZddd d Zd d Zd S)_StateFSourcez bool | NoneNone)source save_markerrestorereturncCs||_||_||_dSN)_source _save_markerr)selfr r rrm/mounts/lovelace/software/anaconda3/envs/qiime2-amplicon-2024.2/lib/python3.8/site-packages/tomlkit/source.py__init__ sz_State.__init__rcCs0t|jj|_|jj|_|jj|_|jj|_|Sr)rr_chars_idx_current_markerrrrr __enter__s    z_State.__enter__cCs<|js |r8|j|j_|j|j_|j|j_|jr8|j|j_dSr)rrrrrrr)rexception_type exception_valtracerrr__exit__s     z_State.__exit__N)FF)__name__ __module__ __qualname__rrr!rrrrr s   r c@s>eZdZdZdddddZddZd d d d Zd dZdS) _StateHandlerz) State preserver for the Parser. r r )r rcCs||_g|_dSr)r_states)rr rrrr.sz_StateHandler.__init__cOst|jf||Sr)r r)rargskwargsrrr__call__2sz_StateHandler.__call__r rcCs|}|j||Sr)r&appendr)rstaterrrr5s z_StateHandler.__enter__cCs|j}||||Sr)r&popr!)rrrr r+rrrr!:s z_StateHandler.__exit__N)r"r#r$__doc__rr)rr!rrrrr%)s r%cseZdZedZdddfdd ZddZed d d d Zed d ddZ edd ddZ ed d ddZ dd ddZ d1dddddZ d2d dddddZd3d"d#Zdd d$d%Zdd d&d'Zefd(d)d)d*d+d,d-Zd.d d/d0ZZS)4r strr )_rcsNttddt|D|_d|_d|_td|_t ||_ | dS)NcSsg|]\}}|t|fqSrr).0icrrr Fsz#Source.__init__..r) superriter enumeraterrrrrr%_stateinc)rr0 __class__rrrBs   zSource.__init__cCs||dSr)r:markrrrrresetPsz Source.resetr%rcCs|jSr)r9rrrrr+Wsz Source.stateintcCs|jSr)rrrrridx[sz Source.idxrcCs|jSr)rrrrrcurrent_szSource.currentcCs|jSr)rrrrrmarkercsz Source.markercCs||j|jS)z= Extracts the value between marker and index )rrrrrrextractgszSource.extractNztype[ParseError] | Nonebool) exceptionrcCsTzt|j\|_|_WdStk rNt||_|j|_|rH||YdSXdS)z Increments the parser if the end of the input has not been reached. Returns whether or not it was able to advance. TFN)nextrrr StopIterationlenEOF parse_error)rrErrrr:ms  z Source.inc)nrErcstfddt|DS)zm Increments the parser by n characters if the end of the input has not been reached. c3s|]}jdVqdS))rEN)r:)r1r0rErrr szSource.inc_n..)allrange)rrKrErrLrinc_n~sz Source.inc_nrcCsH|j|kr.|dkr.|d8}|d8}|sq.q|dkrD|t|jdS)zD Consume chars until min/max is satisfied is valid. rN)rAr:rJr)rcharsminmaxrrrconsumeszSource.consumecCs |j|jkS)zN Returns True if the parser has reached the end of the input. )rrIrrrrendsz Source.endcCs |j|_dS)zA Sets the marker to the index's current position N)rrrrrrr=sz Source.markztype[ParseError]rr)rEr'r(rcOs|\}}|||f||S)zJ Creates a generic "parse error" at the current position. ) _to_linecol)rrEr'r(linecolrrrrJs zSource.parse_errorztuple[int, int]cCsfd}t|D]D\}}|t|d|jkrD|d|j|fS|t|d7}qt|dfS)NrrR)r8 splitlinesrHr@)rcurr2rYrrrrXs zSource._to_linecol)N)N)rrQ)r"r#r$rrIrr>propertyr+r@rArBrCr:rPrVrWr=rrJrX __classcell__rrr;rr ?s(  r N) __future__rrtypingrtomlkit.exceptionsrrtomlkit.toml_charrr r%r/r rrrrs