U Tg8@sNddlZddlZddlZddlZddlZddlZddlmZGdddZdS)N)YouTubec@sPeZdZdddZddZdd d Zdd dZdddZdddZdddZ d S)VideoDownloader downloads/var/www/html/downloadshttp://8.215.44.123/downloadscCs@||_||_||_tj|ddtjdtjdtt |_ dS)NTexist_okz4%(asctime)s - %(name)s - %(levelname)s - %(message)s)formatlevel) download_path public_path public_urlosmakedirslogging basicConfigINFO getLogger__name__logger)selfr r r r'/var/www/html/dlvid/final_downloader.py__init__ szVideoDownloader.__init__cCsPd|ksd|krdSd|ks,d|ks,d|kr0dSd|krusz>VideoDownloader.create_public_download_link..rH/z.infowzTitle:  zOriginal file: z Created: datezutf-8zExpires: After z hours zrm -f r&znow + z hourszecho "z" | at )shellzScheduled deletion of z after z"Could not schedule file deletion: )successr file_pathZexpires_in_hourszError creating download link: FrRr")rrr r*r+basenamer5uuiduuid4r6shutilcopy2r openwriter, check_outputdecoder-rr)r;r9r")rrStitle expiry_hoursr>r?Z unique_idZ safe_titleZpublic_filenameZpublic_file_pathZ download_urlZinfo_file_pathfZ delete_cmdZat_timerCrrrcreate_public_download_linkls@    $z+VideoDownloader.create_public_download_linkFc Cs6zt|}|r|jjdd}|sB|jd|dddWS|j|jd}tj |\}}|d}t ||d||j d d WS|j }|s|jd |dd dWS|j|jd}d||j d d WSWn\tk r0} z<|jdt| |jd||||WYSd} ~ XYnXdS)zEDownload video from YouTube using pytube with improved error handlingT)Z only_audiozNo audio stream found for FzNo audio stream foundrT)Z output_pathz.mp3mp3rRrSr^r zNo video stream found for zNo video stream foundmp4z,Error downloading from YouTube with pytube: zFalling back to yt-dlp for N)rstreamsfilterfirstrr"downloadr rr*r+renamer^Zget_highest_resolutionr;r5r)download_with_ytdlp) rr audio_onlyZytstreamout_filebaserHZmp3_filerCrrrdownload_youtubes,    z VideoDownloader.download_youtubec sZzddl}||}|dtdtj|j}|ddddddddd d }|r~|d d d ddgdd }n|ddid}| |d}|j d|d|j |dd} | s|j d|dddW5QRWSd| kr8| ds,|j d|dddW5QRWS| dd} |sJ| dd}| dd} |d |} tj| sfd!d"t|jD} | rtj|j| d} |j d#| n*|j d$| dd%dW5QRWSd| | |d&W5QRWSQRXWnLtk rT} z,|j d't| dt| dWYSd} ~ XYnXdS)(zJDownload video using yt-dlp for all platforms with improved error handlingrNrHz.%(ext)sFTr) ZouttmplquietZ no_warningsverboseZ ignoreerrorsZnocheckcertificateZ geo_bypassretriesZsocket_timeoutzbestaudio/bestZFFmpegExtractAudiorbZ192)keyZpreferredcodecZpreferredquality)r Zpostprocessorsr ZbestrdzAttempting to download z with yt-dlp)rhzNo information extracted for z#Failed to extract video informationrTentriesz(Empty playlist or all videos failed for z#Empty playlist or all videos failedextr^Video.csg|]}|r|qSr) startswith)rJr`filenamerr s z7VideoDownloader.download_with_ytdlp..zFound alternative file: zDownloaded file not found at zDownloaded file not foundrczError downloading with yt-dlp: )yt_dlprrurandomhexr*r6r updateZ YoutubeDLrr)Z extract_infor"getr7listdirr;r5)rrrkr~platformZoutput_templateZydl_optsZ final_extZydlr)r^rSZpossible_filesrCrr{rrjsv       z#VideoDownloader.download_with_ytdlpTc CsL|jd|d|d|||}|jd||||}|dsx|dkrx|sx|jd||||}|drH|sH|rH|d}tj|d }||kr$|jd |d d |d |||} | |kr| |d<d|d<tj| d } |jd| d dnd|d<n$|jd |d d|dd|d<|S)z,Main download method with compression optionzStarting download for z , audio_only=z , compress=zDetected platform: rRrz!yt-dlp failed, trying pytube for rSr'z File size (r(zMB) exceeds target size (zMB). Compressing...T compressedz Compression complete. New size: rFz"MB) is already below target size (zMB). Skipping compression.) rr)rrjrorr*r8rD) rrrkcompressr=rrArSZ file_size_mbZcompressed_pathZ new_size_mbrrrrhs,       zVideoDownloader.download)rrr)rN)rE)F)F)FTr) r __module__ __qualname__rrrDrarorjrhrrrrr s  F 3 Pr) rrr,tempfilerVrXZpytuberrrrrrs