o U h@sxddlmZddlmZddlmZddlmZddlmZddlm Z GdddZ Gd d d Z Gd d d e Z d S)) annotations)copy)Any) ParseError)UnexpectedCharErrorTOMLCharc@s.eZdZ  ddd d Zdd d Zd dZdS)_StateFsourceSource save_marker bool | NonerestorereturnNonecCs||_||_||_dSN)_source _save_markerr)selfr r rrb/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/tomlkit/source.py__init__ s z_State.__init__cCs0t|jj|_|jj|_|jj|_|jj|_|Sr)rr_chars_idx_current_markerrrrr __enter__s    z_State.__enter__cCsD|js|r |j|j_|j|j_|j|j_|jr|j|j_dSdSdSr)rrrrrrr)rexception_type exception_valtracerrr__exit__s    z_State.__exit__N)FF)r r r r rr rrrr )__name__ __module__ __qualname__rrr!rrrrr s  r c@s4eZdZdZdddZdd Zdd d Zd dZdS) _StateHandlerz) State preserver for the Parser. r r rrcCs||_g|_dSr)r_states)rr rrrr.s z_StateHandler.__init__cOst|jg|Ri|Sr)r r)rargskwargsrrr__call__2sz_StateHandler.__call__r cCs|}|j||Sr)r'appendr)rstaterrrr5s z_StateHandler.__enter__cCs|j}||||Sr)r'popr!)rrrr r,rrrr!:s z_StateHandler.__exit__N)r r rrr")r#r$r%__doc__rr*rr!rrrrr&)s    r&cseZdZedZd2fdd Zdd Zed3d d Zed4ddZ ed5ddZ ed4ddZ d6ddZ d7d8ddZ d7d9ddZd:d"d#Zd;d$d%Zdd0d1ZZS)?r _strrrcsNttddt|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;markrrrrresetPs z Source.resetr&cC|jSr)r:rrrrr,Wz Source.stateintcCr@r)rrrrridx[rAz Source.idxrcCr@r)rrrrrcurrent_rAzSource.currentcCr@r)rrrrrmarkercrAz Source.markercCs||j|jS)z= Extracts the value between marker and index )rrrrrrextractgszSource.extractN exceptiontype[ParseError] | NoneboolcCsPz t|j\|_|_WdSty't||_|j|_|r$||dYdSw)z Increments the parser if the end of the input has not been reached. Returns whether or not it was able to advance. TNF)nextrrr StopIterationlenEOF parse_error)rrGrrrr;ms   z Source.incncstfddt|DS)zm Increments the parser by n characters if the end of the input has not been reached. c3s|] }jdVqdS))rGN)r;)r2r0rGrrr szSource.inc_n..)allrange)rrOrGrrPrinc_n~sz Source.inc_nrcCsX|j|vr|dkr|d8}|d8}|sn |j|vr|dks |dkr*|t|jdS)zD Consume chars until min/max is satisfied is valid. rN)rDr;rNr)rcharsminmaxrrrconsumeszSource.consumecCs |j|juS)zN Returns True if the parser has reached the end of the input. )rrMrrrrend z Source.endcCs |j|_dS)zA Sets the marker to the index's current position N)rrrrrrr>r\z Source.marktype[ParseError]r(rr)rcOs$|\}}|||g|Ri|S)zJ Creates a generic "parse error" at the current position. ) _to_linecol)rrGr(r)linecolrrrrNs zSource.parse_errortuple[int, int]cCsfd}t|D]"\}}|t|d|jkr"|d|j|fS|t|d7}qt|dfS)NrrV)r9 splitlinesrLrC)rcurr3r_rrrr^s zSource._to_linecol)r0r1rr)rr&)rrB)rr)rr1r)rGrHrrI)rOrBrGrHrrI)rrU)rrI)rr)rGr]r(rr)rrr)rra)r#r$r%rrMrr?propertyr,rCrDrErFr;rTrZr[r>rrNr^ __classcell__rrr<rr ?s*            r N) __future__rrtypingrtomlkit.exceptionsrrtomlkit.toml_charrr r&r1r rrrrs