Spaces:
Sleeping
Sleeping
File size: 2,730 Bytes
36a4bf9 ea07962 36a4bf9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 |
from __future__ import unicode_literals
import youtube_dl as youtube_dl
import os
import time
import os
import shutil
max_retries = 3
delay = 2
class MyLogger(object):
def __init__(self, external_logger=lambda x: None):
self.external_logger = external_logger
def debug(self, msg):
print("[debug]: ", msg)
self.external_logger(msg)
def warning(self, msg):
print("[warning]: ", msg)
def error(self, msg):
print("[error]: ", msg)
def my_hook(d):
print("hook", d["status"])
if d["status"] == "finished":
print("Done downloading, now converting ...")
def get_ydl_opts(external_logger=lambda x: None):
return {
"format": "bestaudio/best",
"postprocessors": [
{
"key": "FFmpegExtractAudio",
"preferredcodec": "mp3",
"preferredquality": "192", # set the preferred bitrate to 192kbps
}
],
"logger": MyLogger(external_logger),
"outtmpl": "./downloads/audio/%(title)s.%(ext)s", # Set the output filename directly
"progress_hooks": [my_hook],
}
def download_video_audio(url, external_logger=lambda x: None):
retries = 0
while retries < max_retries:
try:
ydl_opts = get_ydl_opts(external_logger)
with youtube_dl.YoutubeDL(ydl_opts) as ydl:
print("Going to download ", url)
info = ydl.extract_info(url, download=False)
filename = ydl.prepare_filename(info)
res = ydl.download([url])
print("youtube-dl result :", res)
mp3_filename = os.path.splitext(filename)[0] + '.mp3'
print('mp3 file name - ', mp3_filename)
return mp3_filename
except Exception as e:
retries += 1
print(
f"An error occurred during downloading (Attempt {retries}/{max_retries}):",
str(e),
)
if retries >= max_retries:
raise e
time.sleep(delay)
def delete_download(path):
try:
if os.path.isfile(path):
os.remove(path)
print(f"File {path} has been deleted.")
elif os.path.isdir(path):
shutil.rmtree(path)
print(f"Directory {path} and its contents have been deleted.")
else:
print(f"The path {path} is neither a file nor a directory.")
except PermissionError:
print(f"Permission denied: Unable to delete {path}.")
except FileNotFoundError:
print(f"File or directory not found: {path}")
except Exception as e:
print(f"An error occurred while trying to delete {path}: {str(e)}")
|