def read(self):
ui_setting = self.ui_setting
cw = self.customWidget
print_ = get_print(cw)
if self.yt_type == 'video':
res = get_resolution()
info = get_videos(self.url,
type=self.yt_type,
max_res=res,
only_mp4=False,
audio_included=not True,
cw=cw)
else:
abr = get_abr()
info = get_videos(self.url, type=self.yt_type, max_abr=abr, cw=cw)
videos = info['videos']
cw.enableSegment(overwrite=True)
# first video must be valid
while videos:
video = videos[0]
try:
video.url()
break
except Exception as e:
print(e)
videos.remove(video)
else:
raise Exception('No videos')
if len(videos) > 1:
p2f = get_p2f(cw)
if p2f:
self.single = False
self.title = clean_title(info['title'])
self.urls = [video.url for video in videos]
video = videos[0]
self.setIcon(video.thumb)
return
else:
video = videos.pop(0)
cw.gal_num = cw.url = video.url._url
if videos and cw.alive:
s = u', '.join(video.url._url for video in videos)
self.exec_queue.put(([s, {
'youtube': cw.format
}], 'downButton(cw[0], format_selector=cw[1])'))
self.urls.append(video.url)
self.artist = video.username
self.setIcon(video.thumb)
self.title = video.title
def read(self):
ui_setting = self.ui_setting
cw = self.cw
print_ = get_print(cw)
if self.yt_type == 'video':
res = get_resolution()
info = get_videos(self.url,
type=self.yt_type,
max_res=res,
only_mp4=False,
audio_included=not True,
cw=cw)
else:
abr = get_abr()
info = get_videos(self.url, type=self.yt_type, max_abr=abr, cw=cw)
videos = info['videos']
if not videos:
raise Exception('No videos')
self.enableSegment(overwrite=True)
# first video must be valid
while videos:
video = videos[0]
try:
video.url()
break
except Exception as e:
e_ = e
self.print_(print_error(e)[0])
videos.remove(video)
else:
raise e_
if info['type'] != 'single':
video = self.process_playlist(info['title'], videos)
else:
self.urls.append(video.url)
self.title = video.title
if video.stream.live:
self.lock = False
self.artist = video.username
self.setIcon(video.thumb)
def pp(self, filename):
if self.format == 'mp4':
return
name, ext_old = os.path.splitext(filename)
filename_new = '{}.mp3'.format(name)
ffmpeg.convert(filename,
filename_new,
'-shortest -preset ultrafast -b:a {}k'.format(
get_abr()),
cw=self.cw)
if utils.ui_setting.albumArt.isChecked():
self.thumb.seek(0) #
ffmpeg.add_cover(filename_new,
self.thumb, {
'artist': self.username,
'title': self.title
},
cw=self.cw)
return filename_new
def _pp(self, filename):
cw = self.cw
print_ = get_print(cw)
ui_setting = utils.ui_setting
ext = os.path.splitext(filename)[1].lower()
if not os.path.isfile(filename):
print(u'no file: {}'.format(filename))
return
filename_new = None
if self.type == 'video' and (self.audio is not None
or ext != '.mp4'): # UHD or non-mp4
if self.audio is not None: # merge
print_(u'Download audio: {}'.format(self.audio))
hash = uuid()
path = os.path.join(os.path.dirname(filename),
'{}_a.tmp'.format(hash))
if cw is not None:
cw.trash_can.append(path)
if constants.FAST:
downloader_v3.download(self.audio,
chunk=1024 * 1024,
n_threads=2,
outdir=os.path.dirname(path),
fileName=os.path.basename(path),
customWidget=cw,
overwrite=True)
else:
downloader.download(self.audio,
outdir=os.path.dirname(path),
fileName=os.path.basename(path),
customWidget=cw,
overwrite=True)
ext, out = ffmpeg.merge(filename,
path,
cw=cw,
vcodec=self.vcodec)
#print(out)
name, ext_old = os.path.splitext(filename)
if ext_old.lower() != ext.lower():
print_(u'rename ext {} --> {}'.format(ext_old, ext))
filename_new = u'{}{}'.format(name, ext)
if os.path.isfile(filename_new):
os.remove(filename_new)
os.rename(filename, filename_new)
else: # convert non-mp4 video -> mp4
name, ext_old = os.path.splitext(filename)
filename_new = u'{}.mp4'.format(name)
print_(u'Convert video: {} -> {}'.format(
filename, filename_new))
ffmpeg.convert(filename, filename_new, cw=cw)
elif self.type == 'audio' and ext != '.mp3': # convert non-mp3 audio -> mp3
name, ext_old = os.path.splitext(filename)
filename_new = u'{}.mp3'.format(name)
ffmpeg.convert(filename,
filename_new,
'-shortest -preset ultrafast -b:a {}k'.format(
get_abr()),
cw=cw)
if self.type == 'audio' and ui_setting.albumArt.isChecked():
try:
self.thumb.seek(0) #
ffmpeg.add_cover(filename_new,
self.thumb, {
'artist': self.username,
'title': self.title
},
cw=cw)
except Exception as e:
s = print_error(e)[-1]
print_(s)
if ui_setting and ui_setting.subtitle.isChecked():
lang = {
'korean': 'ko',
'english': 'en',
'japanese': 'ja'
}[compatstr(ui_setting.subtitleCombo.currentText()).lower()]
if lang in self.subtitles:
try:
subtitle = self.subtitles[lang]
filename_sub = u'{}.vtt'.format(
os.path.splitext(filename)[0])
downloader.download(
subtitle,
os.path.dirname(filename_sub),
fileName=os.path.basename(filename_sub),
overwrite=True)
filename_sub_new = u'{}.srt'.format(
os.path.splitext(filename_sub)[0])
cw.imgs.append(filename_sub_new)
cw.dones.add(
os.path.realpath(filename_sub_new).replace(
'\\\\?\\', ''))
srt_converter.convert(filename_sub, filename_sub_new)
cw.setSubtitle(True)
finally:
try:
os.remove(filename_sub)
except:
pass
return filename_new
def pp(self, filename):
cw = self.cw
print_ = get_print(cw)
ui_setting = utils.ui_setting
ext = os.path.splitext(filename)[1].lower()
if not os.path.isfile(filename):
print('no file: {}'.format(filename))
return
filename_new = None
if self.type == 'video' and (self.audio is not None
or ext != '.mp4'): # UHD or non-mp4
if self.audio is not None: # merge
print_('Download audio: {}'.format(self.audio))
hash = uuid()
path = os.path.join(os.path.dirname(filename),
'{}_a.tmp'.format(hash))
if cw is not None:
cw.trash_can.append(path)
if constants.FAST:
downloader_v3.download(self.audio,
chunk=1024 * 1024,
n_threads=2,
outdir=os.path.dirname(path),
fileName=os.path.basename(path),
customWidget=cw,
overwrite=True)
else:
downloader.download(self.audio,
outdir=os.path.dirname(path),
fileName=os.path.basename(path),
customWidget=cw,
overwrite=True)
ext, out = ffmpeg.merge(filename,
path,
cw=cw,
vcodec=self.vcodec)
#print(out)
name, ext_old = os.path.splitext(filename)
if ext_old.lower() != ext.lower():
print_('rename ext {} --> {}'.format(ext_old, ext))
filename_new = '{}{}'.format(name, ext)
if os.path.isfile(filename_new):
os.remove(filename_new)
os.rename(filename, filename_new)
else: # convert non-mp4 video -> mp4
name, ext_old = os.path.splitext(filename)
filename_new = '{}.mp4'.format(name)
print_('Convert video: {} -> {}'.format(
filename, filename_new))
ffmpeg.convert(filename, filename_new, cw=cw)
elif self.type == 'audio' and ext != '.mp3': # convert non-mp3 audio -> mp3
name, ext_old = os.path.splitext(filename)
filename_new = '{}.mp3'.format(name)
ffmpeg.convert(filename,
filename_new,
'-shortest -preset ultrafast -b:a {}k'.format(
get_abr()),
cw=cw)
if self.type == 'audio' and ui_setting.albumArt.isChecked():
try:
self.thumb.seek(0) #
ffmpeg.add_cover(filename_new,
self.thumb, {
'artist': self.username,
'title': self.title
},
cw=cw)
except Exception as e:
s = print_error(e)[-1]
print_(s)
utils.pp_subtitle(self, filename, cw)
return filename_new