voussoir/ycdl
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Clean up input type of ycdldb.download_video.

Browse source
This commit is contained in:
voussoir 2021-03-29 23:09:04 -07:00
parent 03ee7e387f
commit 18f617a45b
No known key found for this signature in database
GPG key ID: 5F7554F8C26DACCB
1 changed files with 10 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
ycdl/ycdldb.py
View file

@ -310,14 +310,17 @@ class YCDLDBVideoMixin:
Create the queuefile within the channel's associated directory, or
the default directory from the config file.
'''
if isinstance(video, ytapi.Video):
video_id = video.id
if isinstance(video, objects.Video):
pass
elif isinstance(video, ytapi.Video):
video = self.get_video(video.id)
elif isinstance(video, str):
video = self.get_video(video)
else:
video_id = video
raise TypeError(video)
video = self.get_video(video_id)
if video.state != 'pending' and not force:
self.log.debug('%s does not need to be downloaded.', video_id)
self.log.debug('%s does not need to be downloaded.', video.id)
return
try:
@ -328,12 +331,12 @@ class YCDLDBVideoMixin:
download_directory = self.config['download_directory']
extension = self.config['queuefile_extension']
self.log.info('Creating queuefile for %s.', video_id)
self.log.info('Creating queuefile for %s.', video.id)
download_directory = pathclass.Path(download_directory)
download_directory.makedirs(exist_ok=True)
queuefile = download_directory.with_child(video_id).replace_extension(extension)
queuefile = download_directory.with_child(video.id).replace_extension(extension)
queuefile.touch()
video.mark_state('downloaded', commit=False)