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

Add method get_playlist_videos.

Browse source
This commit is contained in:
voussoir 2020-03-17 20:53:10 -07:00
parent 08ae8a9394
commit 1a7662b6bd
1 changed files with 13 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
ycdl/ytapi.py
View file

@ -61,19 +61,14 @@ class Youtube:
user = self.youtube.channels().list(part='snippet', id=uid).execute()
return user['items'][0]['snippet']['title']
def get_user_videos(self, username=None, uid=None):
if username:
user = self.youtube.channels().list(part='contentDetails', forUsername=username).execute()
else:
user = self.youtube.channels().list(part='contentDetails', id=uid).execute()
upload_playlist = user['items'][0]['contentDetails']['relatedPlaylists']['uploads']
def _iter_playlist(self, playlist_id):
page_token = None
while True:
response = self.youtube.playlistItems().list(
maxResults=50,
pageToken=page_token,
part='contentDetails',
playlistId=upload_playlist,
playlistId=playlist_id,
).execute()
page_token = response.get('nextPageToken', None)
video_ids = [item['contentDetails']['videoId'] for item in response['items']]
@ -86,6 +81,17 @@ class Youtube:
if page_token is None:
break
def get_playlist_videos(self, playlist_id):
yield from self._iter_playlist(playlist_id)
def get_user_videos(self, username=None, uid=None):
if username:
user = self.youtube.channels().list(part='contentDetails', forUsername=username).execute()
else:
user = self.youtube.channels().list(part='contentDetails', id=uid).execute()
upload_playlist_id = user['items'][0]['contentDetails']['relatedPlaylists']['uploads']
yield from self._iter_playlist(upload_playlist_id)
def get_related_videos(self, video_id, count=50):
if isinstance(video_id, Video):
video_id = video_id.id