o i")@sddlZddlZddlZddlZddlmZddlmZmZddl m Z ddl m Z ddl mZmZmZmZmZmZe rXdd 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__rrW/home/ubuntu/.local/lib/python3.10/site-packages/yt_dlp/postprocessor/embedthumbnail.pyrsrc@s:eZdZd ddZddZddZejdd d d ZdS) EmbedThumbnailPPNFcCst||||_dS)N)r__init___already_have_thumbnail)self downloaderalready_have_thumbnailrrrr s  zEmbedThumbnailPP.__init__c sfdd}zd}|j|ddgdd}t||}|dur!|WSWnty?}z|d||WYd}~Sd}~wwt|dt|d fS) Ncs.dd}}|r|r||fSdSdS)Nwidthheightget)r!r"thumbnail_dictrrguess%sz9EmbedThumbnailPP._get_thumbnail_resolution..guessz!,\s*(?P\d+)x(?P\d+)\s*[,\[]z -hide_banner)r)expected_retcodesz)unable to find the thumbnail resolution; wh) run_ffmpegresearchr report_warningintgroup)rfilenamer&r' size_regex size_resultmobjerrrr%r_get_thumbnail_resolution$s  z*EmbedThumbnailPP._get_thumbnail_resolutioncCs||d|ddS)Nz: Adding thumbnail to "") to_screen)rexer1rrr _report_run5szEmbedThumbnailPP._report_run)imagesc Cs4|d}t|d}|ds|dg|fStddt|dddddDd}|dur8|d g|fS|d|d}tj|sO|d g|fSt |j }| |||d|d}}tj |ddd}|d d vr|d vr| |d}d}t|j} d} |d dkrgd} |d||||g|| n>|d d vrt|} d|dd} ||d| \} }| dur| dd| g|d8}| d||d|d| d|d|g|d||||| n|d dvr`d|dgv}tr|rd } nv|d!|tjtjd"}zKt|d# }|}Wdn 1s9wYt j!|d$}|sKt"d%||vrWt"d&|t#|}t|||d'g|j$d(<|%|}Wnt&y}z|d)|d } WYd}~nd}~ww| sd} td*dd+Dd}|dur|d,d } nZ|s|d-||t'd.|t'd/|g}|d0d1|(d2D7}|d3||)d4t*|t+j,|dt-j.t-j.d5\}}}|r|d6|/d } n d7|vr|d8d } | s_d} zAg|dd9} ||d:d\} }| dur.| dd| g|d8}| d;|dvrtsnt1d?|d!|t2t3t4d@|d |}t5}dt !||_6t|d# }||_7Wdn 1swYdA|_8|9||d|}|dur|\|_:|_;|d dBkr|<|n t=>|?@dC|dD<|%|}nt1dE| r||krt|||A|| | ||k}|jB|s|jCs|nd|r|jCs|nd|dFg|fS)GNfilepathtemp thumbnailsz$There aren't any thumbnails to embedcss$|] \}}|dr| VqdS)r<Nr#).0itrrr As"z'EmbedThumbnailPP.run..rzThere are no thumbnails on diskz=Skipping embedding the thumbnail because the file is missing.ext)mkvmka)jpgjpegpngrITmp3)z-ccopy-mapz0:0rLz1:0z -write_id3v11z-id3v2_version3 -metadata:s:vztitle=Album coverrOzcomment=Cover (front)ffmpegzimage/rGrH)tagsmimetyperLz-0:z-attachz -metadata:s:z mimetype=zfilename=cover.)m4amp4m4vmovzembed-thumbnail-atomicparsley compat_optsFr)rHrIrb)r*zcould not determine image typezincompatible image type: )data imageformatcovrzunable to embed using mutagen; css |] }t|dgr|VqdS)z-vN)r )r?xrrrrBs  ) AtomicParsley atomicparsleyzlibatomicparsley.sozCNeither mutagen nor AtomicParsley was found. Falling back to ffmpegz4mutagen was not found. Falling back to AtomicParsleyz --artworkz-ocSsg|]}t|qSr)r )r?orrr sz(EmbedThumbnailPP.run..r]r^zAtomicParsley command line: )textstdoutstderrz0Unable to embed thumbnails using AtomicParsley; z No changesz5The file format doesn't support embedding a thumbnailrM) disposition attached_picz -disposition:rez(Unable to embed using ffprobe & ffmpeg; )oggopusflaczSmodule mutagen was not found. Please install using `python3 -m pip install mutagen`)rgrhrfrhasciiMETADATA_BLOCK_PICTUREz]Supported filetypes for thumbnail embedding are: mp3, mkv/mka, ogg/opus/flac, m4a/mp4/m4v/mov)info)Dr r$r8next enumerateospathexistsr.r _downloader fixup_webpsplitextconvert_thumbnailstatst_mtimer:run_ffmpeg_multiple_filesliststream_copy_optsreplaceget_stream_numberextend_ffmpeg_filename_argumentr+ get_paramrr FORMAT_JPEG FORMAT_PNGopenreadrwhat ValueErrorrrQsave Exceptionr _configuration_args write_debugrr run subprocessPIPEstripr rrrrrmimerYtyper6r!r" add_picturebase64 b64encodewritedecode try_utime_delete_downloaded_filesr)rrlr1 temp_filenameidxthumbnail_filename convertororiginal_thumbnail thumbnail_extmtimesuccessoptionsrR old_stream new_streamprefer_atomicparsleyf thumbfile thumb_datatype_metar5r^cmdrbrc returncodepicres convertedrrrr8s    (                                  zEmbedThumbnailPP.run)NF) rrrrr6r:r _restrict_torrrrrrs   r) rror,rcommonrrPrrcompatr dependenciesrutilsr r r r r r mutagen.flacrr mutagen.mp4rrmutagen.oggopusrmutagen.oggvorbisrrrrrrrs