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

Pull out separate method Channel.refresh_uploads_playlist.

Browse source
This commit is contained in:
voussoir 2021-03-31 14:34:39 -07:00
parent 9a47ff00ee
commit 341c40ad1c
No known key found for this signature in database
GPG key ID: 5F7554F8C26DACCB
1 changed files with 11 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
ycdl/objects.py
View file

@ -73,9 +73,8 @@ class Channel(Base):
def refresh(self, *, force=False, rss_assisted=True, commit=True): def refresh(self, *, force=False, rss_assisted=True, commit=True):
self.ycdldb.log.info('Refreshing %s.', self.id) self.ycdldb.log.info('Refreshing %s.', self.id)
if not self.uploads_playlist: if force or (not self.uploads_playlist):
self.uploads_playlist = self.ycdldb.youtube.get_user_uploads_playlist_id(self.id) self.reset_uploads_playlist_id()
self.set_uploads_playlist_id(self.uploads_playlist)
if force or not rss_assisted: if force or not rss_assisted:
video_generator = self.ycdldb.youtube.get_playlist_videos(self.uploads_playlist) video_generator = self.ycdldb.youtube.get_playlist_videos(self.uploads_playlist)
@ -111,6 +110,14 @@ class Channel(Base):
if commit: if commit:
self.ycdldb.commit() self.ycdldb.commit()
def reset_uploads_playlist_id(self):
'''
Reset the stored uploads_playlist id with current data from the API.
'''
self.uploads_playlist = self.ycdldb.youtube.get_user_uploads_playlist_id(self.id)
self.set_uploads_playlist_id(self.uploads_playlist)
return self.uploads_playlist
def set_automark(self, state, commit=True): def set_automark(self, state, commit=True):
if state not in constants.VIDEO_STATES: if state not in constants.VIDEO_STATES:
raise exceptions.InvalidVideoState(state) raise exceptions.InvalidVideoState(state)
@ -143,6 +150,7 @@ class Channel(Base):
self.ycdldb.commit() self.ycdldb.commit()
def set_uploads_playlist_id(self, playlist_id, commit=True): def set_uploads_playlist_id(self, playlist_id, commit=True):
self.ycdldb.log.debug('Setting %s upload playlist to %s.', self.id, playlist_id)
if not isinstance(playlist_id, str): if not isinstance(playlist_id, str):
raise TypeError(f'Playlist id must be a string, not {type(playlist_id)}.') raise TypeError(f'Playlist id must be a string, not {type(playlist_id)}.')