a
    ¹ßúe('  ã                   @   s®   d Z ddlZddlZddlmZ ddlmZ g d¢Zi ZdZ	g d¢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rªeƒ  dS )z³Utilities for comparing files and directories.

Classes:
    dircmp

Functions:
    cmp(f1, f2, shallow=True) -> int
    cmpfiles(a, b, common) -> ([], [], [])
    clear_cache()

é    N)Úfilterfalse)ÚGenericAlias)Úclear_cacheÚcmpÚdircmpÚcmpfilesÚDEFAULT_IGNORESi    )ZRCSZCVSZtagsz.gitz.hgz.bzrZ_darcsÚ__pycache__c                   C   s   t  ¡  dS )zClear the filecmp cache.N)Ú_cacheÚclear© r   r   úlib/python3.9/filecmp.pyr      s    r   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u r¦t| |ƒ}ttƒdkr–tƒ  |t| |||f< |S )aM  Compare two files.

    Arguments:

    f1 -- First file name

    f2 -- Second file name

    shallow -- treat files as identical if their stat signatures (type, size,
               mtime) are identical. Otherwise, files are considered different
               if their sizes or contents differ.  [default: True]

    Return value:

    True if the files are the same, False otherwise.

    This function uses a cache for past comparisons and the results,
    with cache entries invalidated if their stat information
    changes.  The cache may be cleared by calling clear_cache().

    r   FTé   Néd   )	Ú_sigÚosÚstatÚS_IFREGr
   ÚgetÚ_do_cmpÚlenr   )Úf1Úf2ÚshallowÚs1Ús2Zoutcomer   r   r   r      s    
r   c                 C   s   t  | j¡| j| jfS ©N)r   ÚS_IFMTÚst_modeÚst_sizeÚst_mtime)Ústr   r   r   r   F   s    
þr   c              	   C   s¶   t }t| dƒ”}t|dƒj}| |¡}| |¡}||krTW d   ƒ W d   ƒ dS |sW d   ƒ W d   ƒ dS qW d   ƒ n1 sŠ0    Y  W d   ƒ n1 s¨0    Y  d S )NÚrbFT)ÚBUFSIZEÚopenÚread)r   r   ÚbufsizeÚfp1Úfp2Zb1Zb2r   r   r   r   K   s    

r   c                   @   s’   e Zd 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eeƒZdS )r   aM  A class that manages the comparison of 2 directories.

    dircmp(a, b, ignore=None, hide=None)
      A and B are directories.
      IGNORE is a list of names to ignore,
        defaults to DEFAULT_IGNORES.
      HIDE is a list of names to hide,
        defaults to [os.curdir, os.pardir].

    High level usage:
      x = dircmp(dir1, dir2)
      x.report() -> prints a report on the differences between dir1 and dir2
       or
      x.report_partial_closure() -> prints report on differences between dir1
            and dir2, and reports on common immediate subdirectories.
      x.report_full_closure() -> like report_partial_closure,
            but fully recursive.

    Attributes:
     left_list, right_list: The files in dir1 and dir2,
        filtered by hide and ignore.
     common: a list of names in both dir1 and dir2.
     left_only, right_only: names only in dir1, dir2.
     common_dirs: subdirectories in both dir1 and dir2.
     common_files: files in both dir1 and dir2.
     common_funny: names in both dir1 and dir2 where the type differs between
        dir1 and dir2, or the name is not stat-able.
     same_files: list of identical files.
     diff_files: list of filenames which differ.
     funny_files: list of files which could not be compared.
     subdirs: a dictionary of dircmp objects, keyed by names in common_dirs.
     Nc                 C   sD   || _ || _|d u r$tjtjg| _n|| _|d u r:t| _n|| _d S r   )ÚleftÚrightr   ÚcurdirÚpardirÚhider   Úignore)ÚselfÚaÚbr.   r-   r   r   r   Ú__init__z   s    zdircmp.__init__c                 C   sP   t t | j¡| j| j ƒ| _t t | j¡| j| j ƒ| _| j 	¡  | j 	¡  d S r   )
Ú_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 r   )ÚdictÚzipÚmapr   ÚpathÚnormcaser5   r6   ÚlistÚ__getitem__ÚfilterÚ__contains__Úcommonr   Ú	left_onlyÚ
right_only)r/   r0   r1   r   r   r   Úphase1Ž   s
    zdircmp.phase1c           	   	   C   s  g | _ g | _g | _| jD ]ð}tj | j|¡}tj | j|¡}d}zt 	|¡}W n t
yd   d}Y n0 zt 	|¡}W n t
yŠ   d}Y n0 |rüt	 |j¡}t	 |j¡}||kr¾| j |¡ n<t	 |¡rÖ| j  |¡ n$t	 |¡rî| j |¡ n| j |¡ q| j |¡ qd S )Nr   r   )Úcommon_dirsÚcommon_filesÚcommon_funnyrC   r   r=   Újoinr)   r*   r   ÚOSErrorr   r   ÚappendÚS_ISDIRÚS_ISREG)	r/   ÚxZa_pathZb_pathÚokZa_statZb_statZa_typeZb_typer   r   r   Úphase2•   s4    




zdircmp.phase2c                 C   s&   t | j| j| jƒ}|\| _| _| _d S r   )r   r)   r*   rH   Ú
same_filesÚ
diff_filesÚfunny_files)r/   Zxxr   r   r   Úphase3¸   s    zdircmp.phase3c                 C   sN   i | _ | jD ]<}tj | j|¡}tj | j|¡}t||| j| j	ƒ| j |< qd S r   )
ÚsubdirsrG   r   r=   rJ   r)   r*   r   r.   r-   )r/   rO   Za_xZb_xr   r   r   Úphase4¼   s
    
zdircmp.phase4c                 C   s$   |   ¡  | j ¡ D ]}| ¡  qd S r   )rW   rV   ÚvaluesÚphase4_closure©r/   Zsdr   r   r   rY   Æ   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*   rD   r7   rE   rR   rS   rT   rG   rI   r8   r   r   r   ÚreportË   s,    






zdircmp.reportc                 C   s*   |   ¡  | j ¡ D ]}tƒ  |  ¡  qd S r   )r]   rV   rX   r\   rZ   r   r   r   Úreport_partial_closureä   s    zdircmp.report_partial_closurec                 C   s*   |   ¡  | j ¡ D ]}tƒ  | ¡  qd S r   )r]   rV   rX   r\   Úreport_full_closurerZ   r   r   r   r_   ê   s    zdircmp.report_full_closure)rV   rR   rS   rT   rG   rH   rI   rC   rD   rE   r5   r6   c                 C   s*   || j vrt|ƒ‚| j | | ƒ t| |ƒS r   )Ú	methodmapÚAttributeErrorÚgetattr)r/   Úattrr   r   r   Ú__getattr__ö   s    
zdircmp.__getattr__)NN)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r2   r9   rF   rQ   rU   rW   rY   r]   r^   r_   r:   r`   rd   Úclassmethodr   Ú__class_getitem__r   r   r   r   r   X   s&   !
#
ür   c                 C   sJ   g g g f}|D ]6}t j | |¡}t j ||¡}|t|||ƒ  |¡ q|S )a]  Compare common files in two directories.

    a, b -- directory names
    common -- list of file names found in both directories
    shallow -- if true, do comparison based solely on stat() information

    Returns a tuple of three lists:
      files that compare equal
      files that are different
      filenames that aren't regular files.

    )r   r=   rJ   Ú_cmprL   )r0   r1   rC   r   ÚresrO   ZaxZbxr   r   r   r   ÿ   s    
r   c                 C   s.   z||| ||ƒƒ W S  t y(   Y dS 0 d S )Né   )rK   )r0   r1   ZshÚabsr   r   r   r   rk     s    rk   c                 C   s   t t|j| ƒƒS r   )r?   r   rB   )ZflistÚskipr   r   r   r3   #  s    r3   c                  C   sr   dd l } dd l}| | jdd … d¡\}}t|ƒdkrB| dd ¡‚t|d |d ƒ}d|v rf| ¡  n| ¡  d S )Nr   r   Úrrm   zneed exactly two args)z-rÚ )ÚsysÚgetoptÚargvr   ZGetoptErrorr   r_   r]   )rr   rs   ZoptionsÚargsZddr   r   r   Údemo)  s    
rv   Ú__main__)T)T)rh   r   r   Ú	itertoolsr   Útypesr   Ú__all__r
   r#   r   r   r   r   r   r   r   rn   rk   r3   rv   re   r   r   r   r   Ú<module>   s(   
( (
	