U $Pf@spddlmZddlZddlZddlmZddlmZmZm Z m Z m Z m Z m Z Gddde ZGd d d eZdS) )unicode_literalsN)FFmpegPostProcessor)check_executableencodeArgumentencodeFilenamePostProcessingErrorprepend_extensionreplace_extension shell_quotec@s eZdZdS)EmbedThumbnailPPErrorN)__name__ __module__ __qualname__rrK/tmp/pip-unpacked-wheel-7zdooeg3/youtube_dl/postprocessor/embedthumbnail.pyr sr cs&eZdZdfdd ZddZZS)EmbedThumbnailPPNFcstt||||_dS)N)superr__init___already_have_thumbnail)self downloaderZalready_have_thumbnail __class__rrrszEmbedThumbnailPP.__init__c Cs |d}t|d}|ds0|jdg|fS|ddd}tjt|sd|jdg|fSdd }tj |\}}|r|d d }|d kr||r|jd |t |d }t t|t||}d }|d kr\| dd} t t|t| t | d} |jd| || | ddgtt| t |d} t t| t| | }|ddkrddddddddddg } |jd||||g|| |jstt|tt|t t|t|n8|ddkrtd d!d"Dd} | dkrtd#t| d$t|d$td%t|d$td&t|d$g}|jd'||jjd(d)rz|jd*t|tj|tjtjd+}|\}}|jd,kr|d-d.}t||jstt|d/|kr|jd0n"tt|t t|t|ntd1g|fS)2NfilepathtempZ thumbnailsz5[embedthumbnail] There aren't any thumbnails to embedfilenamez=Skipping embedding the thumbnail because the file is missing.c SsDtt|d}|d}W5QRX|dddkoB|dddkS)Nrb rsRIFFsWEBP)openrread)pathfbrrris_webp-sz%EmbedThumbnailPP.run..is_webprZwebpzJ[ffmpeg] Correcting extension to webp and escaping path for thumbnail "%s")jpgZpng%#r)z*[ffmpeg] Converting thumbnail "%s" to JPEGz-bsf:vZ mjpeg2jpegextZmp3z-ccopyz-map01z -metadata:s:vztitle="Album cover"zcomment="Cover (Front)"z![ffmpeg] Adding thumbnail to "%s")Zm4aZmp4css|]}t|dgr|VqdS)z-vN)r).0xrrr \s z'EmbedThumbnailPP.run..)Z AtomicParsley atomicparsleyz,AtomicParsley was not found. Please install.Tz --artworkz-oz([atomicparsley] Adding thumbnail to "%s"verboseFz&[debug] AtomicParsley command line: %s)stdoutstderrrzutf-8replaces No changesz5The file format doesn't support embedding a thumbnailzCOnly mp3 and m4a/mp4 are supported for thumbnail embedding for now.)r getZ _downloaderZ to_screenosr%existsrZreport_warningsplitextlowerr renamer7Z run_ffmpegremoveZrun_ffmpeg_multiple_filesrnextr rparamsr subprocessPopenPIPE communicate returncodedecodestrip)rinforZ temp_filenameZthumbnail_filenamer(_Z thumbnail_extZthumbnail_webp_filenameZescaped_thumbnail_filenameZescaped_thumbnail_jpg_filenameZthumbnail_jpg_filenameoptionsr3cmdpr5r6msgrrrruns            zEmbedThumbnailPP.run)NF)rrrrrN __classcell__rrrrrsr) __future__rr9rAZffmpegrutilsrrrr r r r r rrrrrs  $