a h")@sddlZddlZddlZddlZddlmZddlmZmZddl m Z ddl m Z ddl mZmZmZmZmZmZe rdd lmZmZdd lmZmZdd lmZdd lmZGd ddeZGdddeZdS)N) PostProcessor)FFmpegPostProcessorFFmpegThumbnailsConvertorPP)imghdr)mutagen)PopenPostProcessingErrorcheck_executableencodeArgumentprepend_extension shell_quote)FLACPicture)MP4MP4Cover)OggOpus) OggVorbisc@s eZdZdS)EmbedThumbnailPPErrorN)__name__ __module__ __qualname__rrX/mnt/pikpak/tmp/myenv/lib/python3.9/site-packages/yt_dlp/postprocessor/embedthumbnail.pyrsrc@s:eZdZd ddZddZddZejdd d d ZdS) EmbedThumbnailPPNFcCst||||_dS)N)r__init___already_have_thumbnail)selfZ downloaderZalready_have_thumbnailrrrr s zEmbedThumbnailPP.__init__c sfdd}z8d}|j|ddgdd}t||}|durB|WSWn:ty~}z"|d||WYd}~Sd}~00t|dt|d fS) Ncs*dd}}|r&|r&||fSdS)Nwidthheightget)rr 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)rZexer.rrr _report_run5szEmbedThumbnailPP._report_run)Zimagesc CsN|d}t|d}|ds.|dg|fStddt|dddddDd}|durp|d g|fS|d|d}tj|s|d g|fSt |j }| |||d|d}}tj |ddd}|d d vr|d vr| |d}d}t|j} d} |d dkrPgd} |d||||g|| n|d d vrt|} d|dd} ||d| \} }| dur| dd| g|d8}| d||d|d| d|d|g|d||||| n|d dvrd|dgv}tr0|r6d } n|d!|tjtjd"}zt|d#}|}Wdn1s|0Yt j!|d$}|st"d%n||vrt"d&|t#|}t|||d'g|j$d(<|%|}Wn:t&y"}z |d)|d } WYd}~n d}~00| sd} td*dd+Dd}|dur\|d,d } n|sl|d-||t'd.|t'd/|g}|d0d1|(d2D7}|d3||)d4t*|t+j,|dt-j.t-j.d5\}}}|r|d6|/d } nd7|vr|d8d } | sd} zg|dd9} ||d:d\} }| durh| dd| g|d8}| d;|dvrtst1d?|d!|t2t3t4d@|d |}t5}dt !||_6t|d#}||_7Wdn1s^0YdA|_8|9||d|}|dur|\|_:|_;|d dBkr|<|nt=>|?@dC|dD<|%|}nt1dE| r||krt|||A|| | ||k}|jB|s$|jCs(|nd|r<|jCs<|nd|dFg|fS)GNfilepathtempZ thumbnailsz$There aren't any thumbnails to embedcss"|]\}}|dr| VqdS)r4Nr!).0itrrr Az'EmbedThumbnailPP.run..rzThere are no thumbnails on diskz=Skipping embedding the thumbnail because the file is missing.ext)ZmkvZmka)jpgjpegpngr?TZmp3)z-ccopy-mapz0:0rAz1:0z -write_id3v11z-id3v2_version3 -metadata:s:vztitle=Album coverrDzcomment=Cover (front)ffmpegzimage/r=r>)tagsmimetyperAz-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 )r6xrrrr9s ) AtomicParsley atomicparsleyzlibatomicparsley.sozCNeither mutagen nor AtomicParsley was found. Falling back to ffmpegz4mutagen was not found. Falling back to AtomicParsleyz --artworkz-ocSsg|] }t|qSr)r )r6orrr r:z(EmbedThumbnailPP.run..rKrLzAtomicParsley command line: )textstdoutstderrz0Unable to embed thumbnails using AtomicParsley; z No changesz5The file format doesn't support embedding a thumbnailrB) 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)Dr r"r2next enumerateospathexistsr+rZ _downloaderZ fixup_webpsplitextZconvert_thumbnailstatst_mtimer3Zrun_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 rrrrrmimerItyper0rr Z add_picturebase64 b64encodewritedecodeZ try_utimeZ_delete_downloaded_filesr)rrYr.Z temp_filenameidxZthumbnail_filenameZ convertorZoriginal_thumbnailZ thumbnail_extmtimesuccessoptionsrGZ old_streamZ new_streamZprefer_atomicparsleyfZ thumbfileZ thumb_datatype_metar/rLcmdrPrQ returncodeZpicresZ convertedrrrrl8s   (               (           (  *    zEmbedThumbnailPP.run)NF) rrrrr0r3rZ _restrict_torlrrrrrs   r) rrr\r)rmcommonrrErrcompatrZ dependenciesrutilsr r r r r rZ mutagen.flacrrZ mutagen.mp4rrZmutagen.oggopusrZmutagen.oggvorbisrrrrrrrs