o
     JAf#,                     @   s   	 d Z ddlZddlZddlZg dZdadadd Zdd ZG dd	 d	e	Z
d
d ZG dd dZdd Zdd ZedkrCe  dS dS )6    N)checkNannyNagprocess_tokensc                  G   s6   d}| D ]}t j|t|  d}qt jd d S )N  
)sysstderrwritestr)argsseparg r   /usr/lib/python3.10/tabnanny.pyerrprint    s
   r   c               
   C   s   dd l } z|  tjdd  d\}}W n | jy+ } zt| W Y d }~d S d }~ww |D ]\}}|dkr:td a|dkrBtd aq.|sPtdtjd d d S |D ]}t| qRd S )Nr      Zqvz-qz-vzUsage:z[-v] file_or_directory ...)getoptr	   argverrorr   filename_onlyverboser   )r   Zoptsr   msgoar   r   r   r   main'   s(   
r   c                   @   s.   e Zd Z	 dd Zdd Zdd Zdd Zd	S )
r   c                 C   s   |||| _ | _| _d S N)linenor   line)selfr   r   r   r   r   r   __init__@   s   zNannyNag.__init__c                 C      | j S r   )r   r    r   r   r   
get_linenoB      zNannyNag.get_linenoc                 C   r"   r   )r   r#   r   r   r   get_msgD   r%   zNannyNag.get_msgc                 C   r"   r   )r   r#   r   r   r   get_lineF   r%   zNannyNag.get_lineN)__name__
__module____qualname__r!   r$   r&   r'   r   r   r   r   r   ;   s    r   c           	   
   C   sP  	 t j| rEt j| sEtrtd| f  t | }|D ]%}t j| |}t j|r2t j|r>t j|dd  dkrBt	| qd S zt
| }W n tyg } ztd| |f  W Y d }~d S d }~ww tdkrrtd|   zz
tt
|j W n t
jy } ztd| |f  W Y d }~W |  d S d }~w ty } ztd| |f  W Y d }~W |  d S d }~w ty } zE| }| }trtd	| |f  td
|f  t|  nd| v rd|  d } trt|  nt| |t| W Y d }~W |  d S d }~ww W |  n|  w tr&td| f  d S d S )Nz%r: listing directoryz.pyz%r: I/O Error: %sr   zchecking %r ...z%r: Token Error: %sz%r: Indentation Error: %sz)%r: *** Line %d: trouble in tab city! ***zoffending line: %rr   "z%r: Clean bill of health.)ospathisdirislinkr   printlistdirjoinnormcaser   tokenizeopenOSErrorr   r   generate_tokensreadline
TokenErrorcloseIndentationErrorr   r$   r'   r&   r   repr)	filenamesnamefullnamefr   ZnagZbadliner   r   r   r   r   I   sl   	

r   c                   @   sL   e Zd Zd\ZZdd Zdd Zdd Zdd	 Zd
d Z	dd Z
dd ZdS )
Whitespacez 	c           	      C   s   || _ tjtj}}g }d } }}| j D ]9}||kr$|d }|d }q||krN|d }|d }|t|krC|dg|t| d   }|| d ||< d}q || _|| _t||f| _t|dk| _	d S )Nr   r   )
rawrC   STlennnttuplenorm	is_simple)	r    wsrE   rF   countbrH   rI   Zchr   r   r   r!      s(   

zWhitespace.__init__c                 C   s   | j \}}tt|d |S Nr   )rK   maxrG   )r    rN   trailingr   r   r   longest_run_of_spaces   s   
z Whitespace.longest_run_of_spacesc                 C   sH   | j \}}d}t|t|D ]}||| ||   }q|||| j   S )Nr   )rK   rangerG   rI   )r    tabsizerN   rR   Zilir   r   r   indent_level   s
   
zWhitespace.indent_levelc                 C   s   | j |j kS r   )rK   )r    otherr   r   r   equal   s   zWhitespace.equalc                 C   sb   t |  | d }g }td|d D ]}| |||kr.||| |||f q|S rP   rQ   rS   rT   rW   appendr    rX   rH   r   tsr   r   r   not_equal_witness      zWhitespace.not_equal_witnessc                 C   sp   | j |j krdS | jr|jr| j|jkS t|  | d }td|d D ]}| |||kr5 dS q&dS )NFr      T)rH   rL   rI   rQ   rS   rT   rW   )r    rX   rH   r]   r   r   r   less   s   zWhitespace.lessc                 C   sb   t |  | d }g }td|d D ]}| |||kr.||| |||f q|S rP   rZ   r\   r   r   r   not_less_witness  r_   zWhitespace.not_less_witnessN)r(   r)   r*   rE   rF   r!   rS   rW   rY   r^   ra   rb   r   r   r   r   rC      s    rC   c                 C   s8   dd | D }d}t | dkr|d }|d d| S )Nc                 s   s    | ]	}t |d  V  qdS )r   N)r   ).0tupr   r   r   	<genexpr>  s    z#format_witnesses.<locals>.<genexpr>zat tab sizer   sr   z, )rG   r3   )wZfirstsprefixr   r   r   format_witnesses  s
   ri   c                 C   s  t j}t j}t j}t jt jf}tdg}d}| D ]l\}}}	}
}||kr&d}q||krRd}t|}|d |sL|d |}dt	| }t
|	d |||| q||kr\d}|d= q|r||vrd}t|}|d |s|d |}dt	| }t
|	d ||qd S )Nr   r   r   zindent not greater e.g. zindent not equal e.g. )r5   INDENTDEDENTNEWLINECOMMENTNLrC   ra   rb   ri   r   r[   rY   r^   )tokensrk   rl   rm   ZJUNKindentsZcheck_equaltypetokenstartendr   ZthisguyZwitnessr   r   r   r   r     s:   

r   __main__)__version__r-   r	   r5   __all__r   r   r   r   	Exceptionr   r   rC   ri   r   r(   r   r   r   r   <module>   s&   = 7
