mirror of
https://github.com/thewesker/Video-Archive-Discord-Bot.git
synced 2025-12-20 04:11:05 -05:00
31 lines
844 B
Python
31 lines
844 B
Python
import youtube_dl
|
|
|
|
def download(videoUrl):
|
|
response = {'fileName': '', 'duration': 0, 'messages': ''}
|
|
ydl = youtube_dl.YoutubeDL({'outtmpl': '%(id)s.mp4', 'merge_output_format': 'mp4'})
|
|
|
|
|
|
with ydl:
|
|
try:
|
|
result = ydl.extract_info(
|
|
videoUrl,
|
|
download=True
|
|
)
|
|
except:
|
|
response['messages'] = 'Error: Download Failed'
|
|
return response
|
|
|
|
if 'entries' in result:
|
|
# Can be a playlist or a list of videos
|
|
response['messages'] = 'Error: More than 1 result found. Please supply a single video only.'
|
|
return response
|
|
else:
|
|
# Just a video
|
|
video = result
|
|
|
|
if('duration' in video):
|
|
response['duration'] = video['duration']
|
|
response['fileName'] = video['id'] + ".mp4"
|
|
|
|
return response
|