a
    ÿëÞh.  ã                   @   sD   d dl Z ddlmZ ddlmZ ddlmZmZ G dd„ deƒZdS )	é    Né   )ÚPostProcessoré   )Úshutil)ÚPostProcessingErrorÚmake_dirc                   @   s*   e Zd Zd	dd„Zedd„ ƒZdd„ ZdS )
ÚMoveFilesAfterDownloadPPNTc                 C   s   t  | |¡ || _d S ©N)r   Ú__init__Ú_downloaded)ÚselfZ
downloaderÚ
downloaded© r   ú`/mnt/pikpak/tmp/myenv/lib/python3.9/site-packages/yt_dlp/postprocessor/movefilesafterdownload.pyr
      s    z!MoveFilesAfterDownloadPP.__init__c                 C   s   dS )NZ	MoveFilesr   )Úclsr   r   r   Úpp_key   s    zMoveFilesAfterDownloadPP.pp_keyc                    s:  t j |d ¡\}}| d|¡‰ t j ˆ |¡}| jrD||d |d < ‡ fdd„}|d  ¡ D ]Ì\}}|sp||ƒ}t j |¡t j |¡krŠq\t j |¡sª|  	d|› d¡ q\t j |¡rú|  
dd	¡rà|  	d
|› d¡ t  |¡ n|  	d|› d|› d¡ q\t|tƒ |  d|› d|› d¡ t ||¡ q\||d< g |fS )NÚfilepathZ
__finaldirZ__files_to_movec                    s   t j ˆ t j | ¡¡S r	   )ÚosÚpathÚjoinÚbasename)Úold©Zfinaldirr   r   Ú<lambda>   ó    z.MoveFilesAfterDownloadPP.run.<locals>.<lambda>zFile "z" cannot be foundZ
overwritesTzReplacing existing file "ú"zCannot move file "z$" out of temporary directory since "z" already exists. zMoving file "z" to ")r   r   ÚsplitÚgetr   r   ÚitemsÚabspathÚexistsZreport_warningÚ	get_paramÚremover   r   Z	to_screenr   Úmove)r   ÚinfoZdl_pathZdl_nameZ	finalpathZmake_newfilenameZoldfileÚnewfiler   r   r   Úrun   s6    ÿ
zMoveFilesAfterDownloadPP.run)NT)Ú__name__Ú
__module__Ú__qualname__r
   Úclassmethodr   r&   r   r   r   r   r      s   

r   )	r   Úcommonr   Úcompatr   Úutilsr   r   r   r   r   r   r   Ú<module>   s   