U g)@sddlZddlZddlZddlZddlmZddlmZmZddl m Z ddl m Z ddl mZmZmZmZmZmZmZe rdd lmZmZdd lmZmZdd lmZdd lmZGd ddeZGdddeZ dS)N) PostProcessor)FFmpegPostProcessorFFmpegThumbnailsConvertorPP)imghdr)mutagen)PopenPostProcessingErrorcheck_executableencodeArgumentencodeFilenameprepend_extension shell_quote)FLACPicture)MP4MP4Cover)OggOpus) OggVorbisc@s eZdZdS)EmbedThumbnailPPErrorN)__name__ __module__ __qualname__rrG/tmp/pip-unpacked-wheel-q5ljy6pj/yt_dlp/postprocessor/embedthumbnail.pyrsrc@s:eZdZd ddZddZddZejdd d d ZdS) EmbedThumbnailPPNFcCst||||_dS)N)r__init___already_have_thumbnail)self downloaderZalready_have_thumbnailrrrr!s zEmbedThumbnailPP.__init__c sfdd}z8d}|j|ddgdd}t||}|dkrB|WSWn<tk r}z|d||WYSd}~XYnXt|dt|d fS) Ncs*dd}}|r&|r&||fSdS)Nwidthheightget)r!r"thumbnail_dictrrguess&sz9EmbedThumbnailPP._get_thumbnail_resolution..guessz!,\s*(?P\d+)x(?P\d+)\s*[,\[]z -hide_banner)r)Zexpected_retcodesz)unable to find the thumbnail resolution; wh) run_ffmpegresearchr report_warningintgroup)rfilenamer&r'Z size_regexZ size_resultZmobjerrrr%r_get_thumbnail_resolution%s   z*EmbedThumbnailPP._get_thumbnail_resolutioncCs||d|ddS)Nz: Adding thumbnail to "") to_screen)rZexer0rrr _report_run6szEmbedThumbnailPP._report_run)ZimagescCsL|d}t|d}|ds.|dg|fStddt|dddddDd}|dkrp|d g|fS|d|d}tjt|s| d g|fSt |j }| |||d|d}}tj |ddd}|d d kr |d kr ||d}d}tt|j} d} |d dkrpddddddddddddddg} |d||||g|| nn|d d kr&t|} d|dd } ||d!| \} }| dk r| dd"| g|d8}| d#||d$|d%| d$|d&|g|d||||| n|d d'krd(|d)gk}trP|rVd*} n|d+|tjtjd,}zt|d-}| }W5QRXt!j"|d.}|st#d/n||krt#d0|t$|}t|||d1g|j%d2<|&|}Wn8t'k r*}z| d3|d*} W5d}~XYnX| s0d} td4dd5Dd}|dkrd|d6d*} n|st|d7t|dt|dt(d8t|dt(d9t|dg}|d:d;|)d<D7}|d=||*d>t+|t,j-|dt.j/t.j/d?\}}}|r| d@|0d*} ndA|kr0| dBd*} | sd} zz|dP} ||dCd\} }| dk r| dd"| g|d8}| dD|dEg|d||||g|| Wn6t1k r}zd*} t2dF|W5d}~XYnXn|d dGkrts t2dH|d+|t3t4t5dI|d |}t6}dt!"||_7t|d-}| |_8W5QRXdJ|_9|:||d|}|dk r|\|_;|_<|d dKkr|=|nt>?|@AdL|dM<|&|}nt2dN| r||krt|||B|| | ||k}|jC|s"|jDs&|nd|r:|jDs:|nd|dOg|fS)QNfilepathtempZ thumbnailsz$There aren't any thumbnails to embedcss"|]\}}|dr| VqdS)r6Nr#).0itrrr Bs z'EmbedThumbnailPP.run..rzThere are no thumbnails on diskz=Skipping embedding the thumbnail because the file is missing.ext)ZmkvZmka)jpgjpegpngr@TZmp3z-ccopy-mapz0:0z1:0z -write_id3v11z-id3v2_version3z -metadata:s:vztitle="Album cover"zcomment=Cover (front)ffmpegzimage/r>r?)tagsmimetypez-0:z-attachz -metadata:s:z mimetype=zfilename=cover.)Zm4aZmp4Zm4vZmovzembed-thumbnail-atomicparsleyZ compat_optsFr)r?r@rb)r)zcould not determine image typezincompatible image type: )dataZ imageformatZcovrzunable to embed using mutagen; css|]}t|dgr|VqdS)z-vN)r )r8xrrrr;s ) AtomicParsley atomicparsleyzlibatomicparsley.sozCNeither mutagen nor AtomicParsley was found. Falling back to ffmpegz4mutagen was not found. Falling back to AtomicParsleyz --artworkz-ocSsg|] }t|qSr)r )r8orrr sz(EmbedThumbnailPP.run..rKrLzAtomicParsley command line: )textstdoutstderrz0Unable to embed thumbnails using AtomicParsley; z No changesz5The file format doesn't support embedding a thumbnail) disposition attached_picz -disposition:rSz(Unable to embed using ffprobe & ffmpeg; )oggopusflaczSmodule mutagen was not found. Please install using `python3 -m pip install mutagen`)rUrVrTrVasciiZMETADATA_BLOCK_PICTUREz]Supported filetypes for thumbnail embedding are: mp3, mkv/mka, ogg/opus/flac, m4a/mp4/m4v/mov)info)rBrC)Err$r4next enumerateospathexistsr r-rZ _downloaderZ fixup_webpsplitextZconvert_thumbnailstatst_mtimer5Zrun_ffmpeg_multiple_fileslistZstream_copy_optsreplaceZget_stream_numberextendZ_ffmpeg_filename_argumentr* get_paramrrZ FORMAT_JPEGZ FORMAT_PNGopenreadrwhat ValueErrorrrFsave Exceptionr Z_configuration_argsZ write_debugrr run subprocessPIPEstripr rrrrrmimerItyper2r!r"Z add_picturebase64 b64encodewritedecodeZ try_utimeZ_delete_downloaded_filesr)rrYr0Z temp_filenameidxZthumbnail_filenameZ convertorZoriginal_thumbnailZ thumbnail_extmtimesuccessoptionsrGZ old_streamZ new_streamZprefer_atomicparsleyfZ thumbfileZ thumb_datatype_metar1rLcmdrPrQ returncodeZpicresZ convertedrrrrl9s<   (                        "      zEmbedThumbnailPP.run)NF) rrrrr2r5rZ _restrict_torlrrrrrs   r)!rrr\r+rmcommonrrErrcompatrZ dependenciesrutilsr r r r r rrZ mutagen.flacrrZ mutagen.mp4rrZmutagen.oggopusrZmutagen.oggvorbisrrrrrrrs   $