a
    IfEm                     @   s$  d Z ddlZddlZddlT ejeks.J deehZee	e
hZeehZeehZeeehB ZdZdd eD Zejfdd	Zd
d Zdd Zd-ddZejd Zde> d ZdZee fddZ!dd Z"dd Z#dd Z$dd Z%dd Z&dd  Z'd!d" Z(d#d$ Z)d%d& Z*d'd( Z+d)d* Z,d.d+d,Z-dS )/zInternal support module for sre    N)*zSRE module mismatch))i   i1  )s   i  )   i  )iE  i  i  )i  i  )i  i  )i  i  )i  i  )i  i  )i  i  )i  i  )i  i  )i  i  )i  i  )i2  i  )i4  i  )i>  i  )iA  i  )iB  i  i  )iJ  i  )ic  i  )i  iK  )ia  i  )i  i  c                    s.   i | ]&}|D ]  t  fd d|D qqS )c                 3   s   | ]} |kr|V  qd S N ).0jir   lib/python3.9/sre_compile.py	<genexpr>N       z<dictcomp>.<genexpr>)tuple)r   tr   r
   r   
<dictcomp>N   s   
r   c                 C   s    ||@ r| | M } | |B | @ S r   r   )flags	add_flags	del_flags
TYPE_FLAGSr   r   r   _combine_flagsQ   s    
r   c                 C   s  | j }t}t}t}t}t}d }	d }
d }|t@ r\|t@ s\|t@ rPt	j
}	t	j}
t}nt	j}	t	j}
|D ]|\}}||v rr|t@ s|| || n|t@ r|t|  || n|	|s|| || n|
|}|s|t|  || n||vr|t|  || nh|t || }|d |tu r2|t |f||  D ]}|t || q@|t || | | |< q`|tu rt||	|
|\}}|t@ r|t@ r|t n(|s|t n|s|t n|t || }|d t|||  || | | |< q`|tu r*|t@ r |t n|t q`||v r6|t@ rLt d|f t!|d r|t"u rn|t# n|t$ || }|d ||d  ||d  t%| |d | |t& || | | |< nl|t' || }|d ||d  ||d  t%| |d | || | | |< |t"u r,|t( n|t) q`|t*u r|\}}}}|rj|t+ ||d d  t%| |t,||| |r|t+ ||d d d  q`||v r|| q`||v rD|| || }|d |d dkr|d n*|d - \}}||krt d|| t%| |d | |t& || | | |< q`|t.u r|| || }|d t%| || |t& || | | |< q`|t/u r|| |t0@ rt12||}|t@ rt32||}n|t@ rt42||}|| q`|t5u r|| g }|j }|d D ]N}|| }|d t%| || |t6 |||  |d || | | |< q|t |D ]}|| | | |< qlq`|t7u r|| |t@ rt8| }n|t@ rt9| }|| q`|t:u r |t@ s|| n,|t@ r|t; n|s
|t< n|t= ||d  q`|t>u r|| ||d d  || }|d t%| |d | |d r|t6 || }|d || | d | |< t%| |d | || | | |< n|| | d | |< q`t d|f q`d S )Nr   z*internal: unsupported template operator %r      z(look-behind requires fixed-width patternz%internal: unsupported operand type %r)?appendlen_LITERAL_CODES_REPEATING_CODES_SUCCESS_CODES_ASSERT_CODESSRE_FLAG_IGNORECASESRE_FLAG_LOCALESRE_FLAG_UNICODE_sreunicode_iscasedunicode_tolower_ignorecase_fixesascii_iscasedascii_tolowerOP_LOCALE_IGNORE	OP_IGNOREOP_UNICODE_IGNOREIN_UNI_IGNORENOT_LITERALNEGATELITERALFAILUREIN_optimize_charsetIN_LOC_IGNORE	IN_IGNORE_compile_charsetANYSRE_FLAG_DOTALLANY_ALLSRE_FLAG_TEMPLATEerror_simple
MAX_REPEAT
REPEAT_ONEMIN_REPEAT_ONE_compileSUCCESSREPEAT	MAX_UNTIL	MIN_UNTIL
SUBPATTERNMARKr   getwidthCALLATSRE_FLAG_MULTILINEAT_MULTILINEget	AT_LOCALE
AT_UNICODEBRANCHJUMPCATEGORY	CH_LOCALE
CH_UNICODEGROUPREFGROUPREF_LOC_IGNOREGROUPREF_IGNOREGROUPREF_UNI_IGNOREGROUPREF_EXISTS)codepatternr   emit_lenLITERAL_CODESREPEATING_CODESSUCCESS_CODESASSERT_CODESiscasedtolowerfixesopavloskipkcharsethascasedgroupr   r   phitail
tailappendskipyesskipnor   r   r   r>   W   sB   
















































r>   c                 C   s   |j }| D ]\}}|| |tu r$q
|tu r6|| q
|tu sF|tu r`||d  ||d  q
|tu rt|| q
|tu r|| q
|tu r|t	@ r|t
|  q|t@ r|t|  q|| q
td|f q
|t d S )Nr   r   z%internal: unsupported set operator %r)r   r-   r.   RANGERANGE_UNI_IGNORECHARSETextend
BIGCHARSETrO   r    rP   r!   rQ   r9   r/   )rg   r   rW   rY   rb   rc   r   r   r   r4   	  s,    

r4   c              	   C   s  g }g }t d}d}| D ]\}}	z,|tu r|rv||	}
d||
< |rd|
|v rd||
 D ]}d||< qV|s~||	r~d}nd||	< n|tu r&t|	d |	d d }|r|rt||D ]*}d||< ||v r|| D ]}d||< qqnt||D ]}d||< q|s$tt||}n|D ]}d||< qn(|tu r@|||	f n|||	f W nX ty   t	|dkr~|dd 7 }Y q"|rd}|tu rt
}|||	f Y n0 qq"qg }d}|d|}|dk rҐq&t	|dkrd }q&|d|}|dk r||t	|f q&|||f q|d ur|D ]>\}}|| dkrZ|t|f n|t||d ff q4||7 }|st	|t	| k r||fS | |fS t	|dkrt|}|t|f ||7 }||fS t|}i }t d}d}t  }tdd	dD ]V}|||d  }||v r2|| ||d < n$| ||d < ||< |d7 }||7 }qt|}|gt| |dd< |t|f ||7 }||fS )
N   Fr   Tr       i   r   i   )	bytearrayr.   rp   rangemapanyr-   r   
IndexErrorr   rq   find
_mk_bitmaprr   bytes_bytes_to_codesrt   )rg   r_   fixupra   outrl   charmaprh   rb   rc   rd   rf   rr   runsqrj   datacompsmappingblockchunkr   r   r   r1   $  s    










r1      r   s   0111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111c                    s8   |  td d d  fddttd  D S )Nc                    s"   g | ]}|  | d qS )r   r   )r   r   	_CODEBITS_intsr   r   
<listcomp>  s   z_mk_bitmap.<locals>.<listcomp>r   )	translate_BITS_TRANSrx   r   )bitsr   r   r   r   r   r}     s    r}   c                 C   s@   t | d}|jtjksJ t||j t| ks8J | S )NI)
memoryviewcastitemsizer"   CODESIZEr   tolist)bar   r   r   r     s    r   c                 C   sD   t | dkrdS | d \}}|tu r<|d d u o:t|d S |tv S )Nr   Fr   r   )r   rC   r:   _UNIT_CODES)rj   rb   rc   r   r   r   r:     s    r:   c                 C   sn   dgt |  }tdt | D ]L}||d  }| | | | kr\|dkrNd||< q||d  }q,|d ||< q|S )aj  
    Generate an overlap table for the following prefix.
    An overlap table is a table of the same size as the prefix which
    informs about the potential self-overlap for each index in the prefix:
    - if overlap[i] == 0, prefix[i:] can't overlap prefix[0:...]
    - if overlap[i] == k with 0 < k <= i, prefix[i-k+1:i+1] overlaps with
      prefix[0:k]
    r   r   )r   rx   )prefixtabler   idxr   r   r   _generate_overlap_table  s    	r   c                 C   s$   | t @ sd S | t@ rtjS tjS d S r   )r   r!   r"   r#   r&   )r   r   r   r   _get_iscased  s
    r   c                 C   s   g }|j }d }t|}| jD ]\}}|tu rF|r<||r< q|| q|tu r|\}}	}
}t||	|
}|t@ rz|t@ rz qt||\}}}|d u r|d urt	|}n|d urt	|| }|
| |s qq qq||dfS ||dfS )NTF)r   r   r   r.   rC   r   r   r    _get_literal_prefixr   rs   )rX   r   r   prefixappendprefix_skipr_   rb   rc   ri   r   r   rj   flags1prefix1prefix_skip1got_allr   r   r   r     s4    



r   c                 C   sd  | j s
d S | j d \}}|tur"qP|\}}}} t|||}|t@ r |t@ r d S q t|}|tu rz|rp||rpd S ||fgS |tu rg }|j}	|d D ]B}
|
s d S |
d \}}|tu r|r||s|	||f q d S q|S |t	u r`|}|r\|D ]f\}}|tu r||rZ d S q|t
u r|d dkr4 d S tt|t|d |d d r d S q|S d S )Nr   r   i  )r   rC   r   r   r    r   r.   rM   r   r0   rp   rz   ry   rx   )rX   r   rb   rc   ri   r   r   r_   rg   charsetappendrj   r   r   r   _get_charset_prefix  sN    



 r   c                 C   s|  |  \}}|tkrt}|dkr8| tdd||g d S g }d}g }|t@ rT|t@ srt||\}}}|srt||}| j}	|	t t	| }
|	d d}|rt
}|d u r|r|tB }n|r|tB }|	| |tk r|	| n|	t |d t }|	t|t |r@|	t	| |d u rt	|}|	| | | | t| n(|rht|\}}|r\J t|||  t	| |
 | |
< d S )Nr      )rE   MAXCODErs   INFOr   r    r   r   r   r   SRE_INFO_PREFIXSRE_INFO_LITERALSRE_INFO_CHARSETminr   r1   r4   )rW   rX   r   rd   rk   r   r   rg   r   rY   re   maskrh   r   r   r   _compile_info0  sR    





r   c                 C   s   t | ttfS r   )
isinstancestrr~   )objr   r   r   isstringk  s    r   c                 C   s8   | j j|B }g }t|| | t|| j| |t |S r   )stater   r   r>   r   r   r?   )rj   r   rW   r   r   r   _coden  s    
r   c                 C   s   dd dd | D  S )N[%s], c                 s   s$   | ]}d t jd d |f V  qdS )z%#0*xr   N)r"   r   r   xr   r   r   r   ~  r   z_hex_code.<locals>.<genexpr>)joinrW   r   r   r   	_hex_code}  s    r   c                    sN   dd l t dttt d  fdddt  d S )Nr   r   c              	      s  d d fdd
}fdd}d7  }||k r| | }|d7 }t | }|tttttttfv rx|| q2|tt	t
tttttfv r| }|d7 }||d|t|f  q2|tu r| }|d7 }tt| }|d d d	ksJ |||dd   q2|tu rV| }|d7 }tt| }|d d
 dksBJ |||d
d   q2|ttttfv r| }||||| d |d ||  ||7 }q2|ttfv r||d  \}}	|d7 }||d||	t|t|	f  q2|tu r||t||dt    |dt 7 }q2|tu rʈ| }|d7 }t d!fdd||dt"j#   D }
||||
 |dt"j# 7 }d7 t$|D ].}|t||dt    |dt 7 }qd8 q2|t%t&t't(t)fv r| }|d7 }||| q2|t*u r(| }||||| d |d7 }q2|t+u r| }||||| d |r|d ||  ||7 }| | }|r|d||| d n|t qL|d7 }q2|t,t-t.fv r||d  \}}}|t/krd}||||||| d |d ||  ||7 }q2|t0u rJ||d  \}}|||||| d |d7 }q2|t1t2fv r||d  \}}|||||| d |d ||  ||7 }q2|t3u rĈ||d  \}}}}|t/krd}|||t4||||| d |d  |t5@ r|d |d  \}}|d| |d    |  }|ddd!dd |D  dd!t6t|   |7  |d  |    |7  |t7@ rd7 |d  ||  d8 ||7 }q2t8|q2d8 d S )N)toc                    sX   | d ur"  |  |d| f f7 }td v r6dndf dd  d t|  d S )Nz(to %d)z%*d%s :.z  r   end)addprint)r   args)labelsleveloffset_widthstartr   r   print_  s    

z!dis.<locals>.dis_.<locals>.print_c                     s"   t dd    d t |   d S )N r   r   )r   )r   )r   r   r   r   print_2  s    z"dis.<locals>.dis_.<locals>.print_2r   z
%#02x (%r)   AT_	   	CATEGORY_r   z%#02x %#02x (%r-%r)ru   r   c                 3   s   | ]}| tj jV  qd S r   )to_bytesr"   r   	byteorderr   )sysr   r   r     s   z$dis.<locals>.dis_.<locals>.<genexpr>branch	MAXREPEATr      z  prefix_skipz  prefixr   r   c                 s   s   | ]}d | V  qdS )z%#02xNr   r   r   r   r   r     r   z(%r) z	  overlapin)9OPCODESr?   r/   r5   r7   rA   rB   r-   r.   r,   LITERAL_IGNORENOT_LITERAL_IGNORELITERAL_UNI_IGNORENOT_LITERAL_UNI_IGNORELITERAL_LOC_IGNORENOT_LITERAL_LOC_IGNOREchrrG   r   ATCODESrO   CHCODESr0   r3   r+   r2   rp   rq   rr   r   r   rt   listr   r"   r   rx   rD   rR   rT   rU   rS   rN   rM   r@   r<   r=   r   rV   ASSERT
ASSERT_NOTr   binr   ry   r   
ValueError)r   r   r   r   r   rb   argre   rd   rk   r   r	   r   maxr   
prefix_lenr   r   rW   dis_r   r   r   r   )r   r   r     s    





 




















zdis.<locals>.dis_)r   setr   r   r   r   r   r   dis  s     r   c              	   C   s   t | r| }t| |} nd }t| |}|t@ r>t  t| | jj}d g| jj	 }|
 D ]\}}|||< q\t||| jjB || jj	d |t|S )Nr   )r   	sre_parseparser   SRE_FLAG_DEBUGr   r   r   	groupdictgroupsitemsr"   compiler   r   )rj   r   rX   rW   
groupindex
indexgrouprf   r   r   r   r   r     s"    


r   )NNN)r   ).__doc__r"   r   sre_constantsMAGICr.   r,   r   r@   
MIN_REPEATr;   r   r?   r/   r   r   r   r   r5   r0   r   _equivalencesr%   r   r   r>   r4   r1   r   r   r   r   intr}   r   r:   r   r   r   r   r   r   r   r   r   r   r   r   r   r   <module>   sJ   
4
 3
 
	,; 