U g@sdddlZddlZddlZddlmZddlmZmZmZm Z m Z m Z m Z m Z mZGdddeZdS)N) PostProcessor) PopenPostProcessingErrorcheck_executable cli_optionencodeArgumentencodeFilenameprepend_extension shell_quote str_or_nonec@s<eZdZdZdZd ddZddd Zejdd d d Z dS) SponSkrubPPZsponsZ sponskrubNFcCspt||||_||_t|p"d|_|||_|s@|d|sl|jdkrl|rdt d|dnt ddS)Nrzyt_dlp.postprocessor.SponSkrubPP support is deprecated and may be removed in a future version. Use yt_dlp.postprocessor.SponsorBlock and yt_dlp.postprocessor.ModifyChaptersPP insteadzsponskrub not found in ""zNsponskrub not found. Please install or provide the path using --sponskrub-path) r__init__forcecutoutr argsget_exepathZdeprecation_warningr)self downloaderrrZ ignoreerrorZcutrZ _from_clirB/tmp/pip-unpacked-wheel-q5ljy6pj/yt_dlp/postprocessor/sponskrub.pyrs  zSponSkrubPP.__init__cCs4|rt|dgs0tj||j}t|dgs0dS|S)Nz-h)rosrjoin _exe_name)rrrrrr*s  zSponSkrubPP.get_exe)ZimagescCs|jdkrg|fS|d}tjt|s2g|fS|ddkrT|dg|fS|jr~|js~|dds~| dg|fS|d|jrd nd |jr| d |dds| d t ||j }tjt|rt t||jg}|js|d g7}|t |jjdd7}|t|j7}||j|jdd7}|d|d||g7}dd|D}|dt|tj|d|drdntjd\}}}|st|||d|jrdndnN|dkr|dn8t|| drdnd p d!|g|fS)"NfilepathZ extractor_keyZyoutubez2Skipping sponskrub since it is not a YouTube videoZ__real_downloadFzhSkipping sponskrub since the video was already downloaded. Use --sponskrub-force to run sponskrub anywayzTrying to %s sponsor sectionsremovemarkzHCutting out sponsor segments will cause the subtitles to go out of sync.zSIf sponskrub is run multiple times, unintended parts of the video could be cut out.z-chapterz-proxyproxy)Z use_compatz--idcSsg|] }t|qSr)r ).0irrr Tsz#SponSkrubPP.run..zsponskrub command line: Tverbose)textstdoutzSponsor sections have been %sremovedZmarkedz(No segments in the SponsorBlock databaseZ unrecognisedrz!sponskrub failed with error code )!rrexistsr lowerZ to_screenrrgetZreport_warningr _temp_extrrZ _downloaderparamsshlexsplitrZ_configuration_argsrZ write_debugr rrun get_param subprocessPIPEreplacerstrip splitlines startswith)rZ informationfilenameZ temp_filenamecmdr(_ returncoderrrr31sR       (   *zSponSkrubPP.run)rNFFFF)r) __name__ __module__ __qualname__r/rrrrZ _restrict_tor3rrrrrs    r)rr1r5commonrutilsrrrrr r r r r rrrrrs  ,