o
    !d                     @   s2   d dl Z d dlmZ e eZG dd deZdS )    N)advance_iteratorc                   @   s(   e Zd ZdZdd Zdd Zdd ZdS )	
ComparatorzN
    This class performs all of the comparisons behind the sync operation
    c                 C   s   || _ || _|| _d S )N)_sync_strategy_not_at_dest_sync_strategy_not_at_src_sync_strategy)selfZ"file_at_src_and_dest_sync_strategyZfile_not_at_dest_sync_strategyZfile_not_at_src_sync_strategy r   Gusr/lib/python3.10/site-packages/awscli/customizations/s3/comparator.py__init__   s   
zComparator.__init__c                 c   sX   d}d}d}d}	 z
|s|rt |}W n ty!   d}d}Y nw z
|s+|r+t |}W n ty9   d}d}Y nw |s|sd}d}| ||}	|	dkrY| j||}
|
rX|V  nR|	dkrnd}d}| j|d}
|
rm|V  n=|	dkrd}d}| jd|}
|
r|V  n(|s|rd}| j|d}
|
r|V  n|r|sd}| jd|}
|
r|V  ndS q
)a  
        This function preforms the actual comparisons.  The parameters it takes
        are the generated files for both the source and the destination.  The
        key concept in this function is that no matter the type of where the
        files are coming from, they are listed in the same order, least to
        greatest in collation order.  This allows for easy comparisons to
        determine if file needs to be added or deleted.  Comparison keys are
        used to determine if two files are the same and each file has a
        unique comparison key.  If they are the same compare the size and
        last modified times to see if a file needs to be updated.   Ultimately,
        it will yield a sequence of file info objectsthat will be sent to
        the ``S3Handler``.

        :param src_files: The generated FileInfo objects from the source.
        :param dest_files: The generated FileInfo objects from the dest.

        :returns: Yields the FilInfo objects of the files that need to be
            operated on

        Algorithm:
            Try to take next from both files. If it is empty signal
            corresponding done flag.  If both generated lists are not done
            compare compare_keys.  If equal, compare size and time to see if
            it needs to be updated.  If source compare_key is less than dest
            compare_key, the file needs to be added to the destination.  Take
            the next source file but not not destination file.  If the source
            compare_key is greater than dest compare_key, that destination file
            needs to be deleted from the destination.  Take the next dest file
            but not the source file.  If the source list is empty delete the
            rest of the files in the dest list from the destination.  If the
            dest list is empty add the rest of the file in source list to
            the destination.
        FTNequal	less_thangreater_than)r   StopIterationcompare_comp_keyr   Zdetermine_should_syncr   r   )r   Z	src_filesZ
dest_filesZsrc_doneZ	dest_doneZsrc_takeZ	dest_takesrc_file	dest_fileZcompare_keysZshould_syncr   r   r	   call    sz   #zComparator.callc                 C   s(   |j }|j }||krdS ||k rdS dS )z
        Determines if the source compare_key is less than, equal to,
        or greater than the destination compare_key
        r   r   r   )Zcompare_key)r   r   r   Zsrc_comp_keyZdest_comp_keyr   r   r	   r      s   zComparator.compare_comp_keyN)__name__
__module____qualname____doc__r
   r   r   r   r   r   r	   r      s
    br   )loggingZawscli.compatr   	getLoggerr   ZLOGobjectr   r   r   r   r	   <module>   s   
