evanferrao
/
dump-mirrorbot
mirror of https://github.com/Box-boi/dump-mirrorbot.git
1
0
Fork 0
Code Issues Actions Packages Projects Releases Wiki Activity

Handle weird path naming of youtube_dl 

Signed-off-by: lzzy12 <jhashivam2020@gmail.com>
This commit is contained in:
lzzy12 2020-03-30 16:32:06 +05:30
parent 141b13b52d
commit 0ec499ff4a
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
bot/helper/mirror_utils/download_utils/youtube_dl_download_helper.py
View File

@ -71,19 +71,21 @@ class YoutubeDLHelper(DownloadHelper):
with YoutubeDL(self.opts) as ydl:
result = ydl.extract_info(link, download=False)
name = ydl.prepare_filename(result)
if 'entries' in result:
video = result['entries'][0]
for v in result['entries']:
if v.get('filesize'):
self.size += float(v['filesize'])
self.name = video.get('playlist_title')
# For playlists, ydl.prepare-filename returns the following format: <Playlist Name>-<Id of playlist>.NA
self.name = name.split(f"-{result['id']}")[0]
self.vid_id = video.get('id')
self.is_playlist = True
else:
video = result
if video.get('filesize'):
self.size = int(video.get('filesize'))
self.name = f"{video.get('title')}.{video.get('ext')}"
self.name = name
self.vid_id = video.get('id')
return video
@ -100,9 +102,9 @@ class YoutubeDLHelper(DownloadHelper):
LOGGER.info(f"Downloading with YT-DL: {link}")
self.__gid = f"{self.vid_id}{self.__listener.uid}"
if not self.is_playlist:
self.opts['outtmpl'] = f"{path}/%(title)s.%(ext)s"
self.opts['outtmpl'] = f"{path}/{self.name}"
else:
self.opts['outtmpl'] = f"{path}/%(playlist_title)s/%(title)s.%(ext)s"
self.opts['outtmpl'] = f"{path}/{self.name}/%(title)s.%(ext)s"
self.__onDownloadStart()
threading.Thread(target=self.__download, args=(link,)).start()