B
    u9af&                 @   s   d dl Z d dlZd dlmZ dddddgZi ZdZd	d
ddddddgZdd Zd!ddZ	dd Z
dd ZG dd dZd"ddZee	fddZdd Zdd Zed kre  dS )#    N)filterfalseclear_cachecmpdircmpcmpfilesDEFAULT_IGNORESi    ZRCSZCVSZtagsz.gitz.hgz.bzrZ_darcs__pycache__c               C   s   t   d S )N)_cacheclear r   r   /usr/lib/python3.7/filecmp.pyr      s    Tc             C   s   t t| }t t|}|d tjks8|d tjkr<dS |rL||krLdS |d |d kr`dS t| |||f}|d krt| |}ttdkrt  |t| |||f< |S )Nr   FT   d   )	_sigosstatS_IFREGr	   get_do_cmplenr   )f1f2shallows1s2Zoutcomer   r   r   r      s    
c             C   s   t | j| j| jfS )N)r   S_IFMTst_modest_sizest_mtime)str   r   r   r   D   s    
r   c          
   C   sb   t }t| dJ}t|d4}x,||}||}||kr>dS |sdS qW W d Q R X W d Q R X d S )NrbFT)BUFSIZEopenread)r   r   bufsizefp1fp2Zb1Zb2r   r   r   r   I   s    

r   c               @   s   e Zd Zd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eeeeeeeeeeeeedZdd ZdS )r   Nc             C   sD   || _ || _|d kr$tjtjg| _n|| _|d kr:t| _n|| _d S )N)leftrightr   curdirpardirhider   ignore)selfabr,   r+   r   r   r   __init__x   s    zdircmp.__init__c             C   sP   t t| j| j| j | _t t| j| j| j | _| j	  | j	  d S )N)
_filterr   listdirr'   r+   r,   	left_listr(   
right_listsort)r-   r   r   r   phase0   s    
zdircmp.phase0c             C   s   t tttjj| j| j}t tttjj| j| j}tt|j	t
|j|| _tt|j	t|j|| _tt|j	t|j|| _d S )N)dictzipmapr   pathnormcaser3   r4   list__getitem__filter__contains__commonr   	left_only
right_only)r-   r.   r/   r   r   r   phase1   s
    zdircmp.phase1c       
      C   s:  g | _ g | _g | _x | jD ]}tj| j|}tj| j|}d}yt	|}W n& t
k
rz } zd}W d d }~X Y nX yt	|}W n& t
k
r } zd}W d d }~X Y nX |r&t	|j}t	|j}	||	kr| j| n>t	|r| j | n&t	|r| j| n| j| q| j| qW d S )Nr   r   )common_dirscommon_filescommon_funnyr@   r   r:   joinr'   r(   r   OSErrorr   r   appendS_ISDIRS_ISREG)
r-   xZa_pathZb_pathokZa_statZwhyZb_statZa_typeZb_typer   r   r   phase2   s4    
zdircmp.phase2c             C   s&   t | j| j| j}|\| _| _| _d S )N)r   r'   r(   rE   
same_files
diff_filesfunny_files)r-   Zxxr   r   r   phase3   s    zdircmp.phase3c             C   sR   i | _ xF| jD ]<}tj| j|}tj| j|}t||| j| j	| j |< qW d S )N)
subdirsrD   r   r:   rG   r'   r(   r   r,   r+   )r-   rL   Za_xZb_xr   r   r   phase4   s
    zdircmp.phase4c             C   s(   |    x| j D ]}|  qW d S )N)rT   rS   valuesphase4_closure)r-   sdr   r   r   rV      s    zdircmp.phase4_closurec             C   s   t d| j| j | jr2| j  t d| jd| j | jrT| j  t d| jd| j | jrp| j  t d| j | jr| j  t d| j | jr| j  t d| j | j	r| j	  t d| j	 | j
r| j
  t d| j
 d S )	NZdiffzOnly in:zIdentical files :zDiffering files :zTrouble with common files :zCommon subdirectories :zCommon funny cases :)printr'   r(   rA   r5   rB   rO   rP   rQ   rD   rF   )r-   r   r   r   report   s,    






zdircmp.reportc             C   s.   |    x | j D ]}t  |   qW d S )N)rZ   rS   rU   rY   )r-   rW   r   r   r   report_partial_closure   s    zdircmp.report_partial_closurec             C   s.   |    x | j D ]}t  |  qW d S )N)rZ   rS   rU   rY   report_full_closure)r-   rW   r   r   r   r\      s    zdircmp.report_full_closure)rS   rO   rP   rQ   rD   rE   rF   r@   rA   rB   r3   r4   c             C   s*   || j krt|| j | |  t| |S )N)	methodmapAttributeErrorgetattr)r-   attrr   r   r   __getattr__   s    
zdircmp.__getattr__)NN)__name__
__module____qualname__r0   r6   rC   rN   rR   rT   rV   rZ   r[   r\   r7   r]   ra   r   r   r   r   r   V   s    "
#

c             C   sN   g g g f}x>|D ]6}t j| |}t j||}|t||| | qW |S )N)r   r:   rG   _cmprI   )r.   r/   r@   r   ZresrL   ZaxZbxr   r   r   r      s    

c             C   s,   y||| || S  t k
r&   dS X d S )N   )rH   )r.   r/   shabsr   r   r   r   re     s    re   c             C   s   t t|j| S )N)r<   r   r?   )flistskipr   r   r   r1     s    r1   c              C   sr   dd l } dd l}|| jdd  d\}}t|dkrB|dd t|d |d }d|krf|  n|  d S )Nr   r   rrf   zneed exactly two args)z-r )sysgetoptargvr   ZGetoptErrorr   r\   rZ   )rm   rn   optionsargsddr   r   r   demo$  s    
rs   __main__)T)T)r   r   	itertoolsr   __all__r	   r!   r   r   r   r   r   r   r   rh   re   r1   rs   rb   r   r   r   r   <module>   s$   
' %
	