yf(ddlZddlZddlZddlZddlmZddlmZmZddl m Z ddl m Z ddl mZmZmZmZmZmZmZmZe rdd lmZmZdd lmZmZdd lmZdd lmZGd deZ GddeZ!y)N) PostProcessor)FFmpegPostProcessorFFmpegThumbnailsConvertorPP)imghdr)mutagen)PopenPostProcessingErrorcheck_executableencodeArgumentencodeFilenameerror_to_compat_strprepend_extension shell_quote)FLACPicture)MP4MP4Cover)OggOpus) OggVorbisc eZdZy)EmbedThumbnailPPErrorN)__name__ __module__ __qualname__E/usr/lib/python3/dist-packages/yt_dlp/postprocessor/embedthumbnail.pyrrsrrcReZdZddZdZdZejddZy) EmbedThumbnailPPNFc>tj||||_y)N)r__init___already_have_thumbnail)self downloaderalready_have_thumbnails rr#zEmbedThumbnailPP.__init__"s$$T:6'=$rc\fd} d}|j|ddgd}tj||}||S t |jdt |jdfS#t$r.}|j dt |z|cYd}~Sd}~wwxYw) Nc\jdjd}}|r|r||fSyy)Nwidthheightget)r*r+thumbnail_dicts rguessz9EmbedThumbnailPP._get_thumbnail_resolution..guess's8*..w79K9KH9U6Ef}$ urz!,\s*(?P\d+)x(?P\d+)\s*[,\[]z -hide_banner)r)expected_retcodesz+unable to find the thumbnail resolution; %swh) run_ffmpegresearchr report_warningrintgroup)r%filenamer.r/ size_regex size_resultmobjerrs ` r_get_thumbnail_resolutionz*EmbedThumbnailPP._get_thumbnail_resolution&s %  =J//(D>:J^b/cK99Z5D|w 4::c?#SC%999#     MPcdgPh h i7N s6A44 B+=#B& B+&B+c2|j|d|dy)Nz: Adding thumbnail to "") to_screen)r%exer9s r _report_runzEmbedThumbnailPP._report_run7s #5hZqABr)imagesc 8|d}t|d}|jds|jdg|fStdt |dddddDd}||jdg|fS|d|d}t j jt|s|jd g|fSt|j}|j|||d|dx}}t j j|ddd}|d d vr|d vr|j|d }d }t jt|j } d} |d dk(r-gd} |j#d||j%||g|| n|d d vrt'|j)} d|j+dd} |j-|d| \} }| | j/dd| zg|dz}| j/d|j1|d|zd| zd|zd|zg|j#d||j3||| n|d dvrd|j5dgv}t6r|rd} n |j#d |t9|}t:j<t:j>d!tAjB|}tE|d"5}|jG}dddt;|#g|jHd$<|jK|}| sOd} td&d'Dd}||jd(d} n#|s|jd)t|dt|dtQd*t|dtQd+t|dg}||jSd,Dcgc] }tQ|c}z }|j#d-||jUd.tW|ztYjZ|dt\j^t\j^/\}}}|r"|jd0|jad1|vr|jd2d} | sd} g|j)dd3} |j-|d4d\} }| | j/dd| zg|dz}| j/d5|zd6g|j#d||j%||g|| n6|d d8vr#t6s ted9|j#d |tgthtjd:|d |}tm}d;tAjB|z|_7tE|d"5}|jG|_8dddd<|_9|ju||d|}||\|_;|_<|d d=k(r|j{|n5t}j~|jjd>|d?<|jK|}n ted@| r||k7rt j*|||j|| | ||k7}|j|s |js|nd|r|js|nd|Ag|fS#1swYxYw#tL$r*}|jd%tO|zd} Yd}~d}~wwxYwcc}w#tb$r}d} ted7|d}~wwxYw#1swY}xYw)BNfilepathtemp thumbnailsz$There aren't any thumbnails to embedc3LK|]\}}|jds| yw)rFNr,).0its r z'EmbedThumbnailPP.run..Cs"_41aQUUS]M^QB_s$$rzThere are no thumbnails on diskz=Skipping embedding the thumbnail because the file is missing.ext)mkvmka)jpgjpegpngrTTmp3)z-ccopy-mapz0:0rWz1:0z -write_id3v11z-id3v2_version3 -metadata:s:vztitle="Album cover"rZzcomment=Cover (front)ffmpegzimage/rRrS)tagsmimetyperWz-0:%dz-attachz-metadata:s:%dz mimetype=%szfilename=cover.%s)m4amp4m4vmovzembed-thumbnail-atomicparsley compat_optsFr )rSrTrb)data imageformatcovrz!unable to embed using mutagen; %sc3:K|]}t|dgr|yw)z-vN)r )rJxs rrMz'EmbedThumbnailPP.run..s$&4'D62&4s) AtomicParsley atomicparsleyzlibatomicparsley.sozCNeither mutagen nor AtomicParsley was found. Falling back to ffmpegz4mutagen was not found. Falling back to AtomicParsleyz --artworkz-orirjzAtomicParsley command line: %s)textstdoutstderrz0Unable to embed thumbnails using AtomicParsley; z No changesz5The file format doesn't support embedding a thumbnailrX) disposition attached_picz-disposition:%sroz(Unable to embed using ffprobe & ffmpeg; )oggopusflaczSmodule mutagen was not found. Please install using `python3 -m pip install mutagen`)rqrrrpzimage/%srrasciiMETADATA_BLOCK_PICTUREz]Supported filetypes for thumbnail embedding are: mp3, mkv/mka, ogg/opus/flac, m4a/mp4/m4v/mov)info)Err-rAnext enumerateospathexistsrr6r _downloader fixup_webpsplitextconvert_thumbnailstatst_mtimerCrun_ffmpeg_multiple_filesliststream_copy_optsreplaceget_stream_numberextend_ffmpeg_filename_argumentr3 get_paramr rr FORMAT_JPEG FORMAT_PNGrwhatopenreadr\save Exceptionrr _configuration_args write_debugrr run subprocessPIPEstripr rrrrrmimerdtyper>r*r+ add_picturebase64 b64encodewritedecode try_utime_delete_downloaded_filesr$)r%rvr9 temp_filenameidxthumbnail_filename convertororiginal_thumbnail thumbnail_extmtimesuccessoptionsr] old_stream new_streamprefer_atomicparsleymetaf thumbfile thumb_datar=rjcmdorlrm returncodepicres converteds rrzEmbedThumbnailPP.run:sn #)(F; xx % NNB Ct8O_9T,-?"-Eq#I_aef ; NN< =t8O!,/4Z@ww~~n-?@A    _ `t8O00@0@A T3'26|2DS2I*2UU/(();a NN499:LM :-}x/G :-/B]/RT U   Xx 0 OOHmW = %[8 8#BdnnUbdfFg#g 2 $$$Y9x=D"*!5!5h>Q>QRSYS^S^_qSrsA0$769%.^^%5 6)1zq)Q(RDIIf%IIK$,M  $&4X&46:!; !(NN#hi#G/']^)->)(D9)+6)* @C t7O7OP_7`a!N1-aaC$$_h?$$%E TWHX%XY163TR\RaRajtjyjy1z.FFJ!++.^_e_k_k_m^n,op$v-++,de"' bE 5 5 7EEEG-1-C-C "A1.F*J !-*0D'EF"a NN$5 $BN#ST$$Xx822H>P3QS`bij %[3 3+-BCC   Y 1 $yA$u+NxXA)C!FKK0B$CCCH($/ ,9$>>+ ,CH001CT,EWX[E\]C(+% 3:E{f$ c"/5.>.>syy{.K.R.RSZ.[*+ FFH$M()HI I }0 JJ}h / x.&*<<  %%"+43O3O UY"+D4P4P VZ & 4xE66 !$''(KNabeNf(fg#G$.b<+b#G/2Z[^Z_0`aab , ,s\=A Z3Z&.5Z3;[);B[.>\&Z0+Z33 [&<[!![&. \ 7\\ \)NF) rrrr#r>rCr _restrict_torrrrr!r! s5>:"C ]u-h.hrr!)"rryr4rcommonrr[rrcompatr dependenciesr utilsr r r r rrrr mutagen.flacrr mutagen.mp4rrmutagen.oggopusrmutagen.oggvorbisrrr!rrrrsY !D"    *)'+ / C*Cr