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

Add ChannelNotFound exception.

Browse source
This commit is contained in:
voussoir 2020-07-20 19:03:53 -07:00
parent ff0d01fdda
commit 795aacb044
1 changed files with 9 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
ycdl/ytapi.py
View file

@ -17,6 +17,9 @@ def int_none(x):
return None return None
return int(x) return int(x)
class ChannelNotFound(Exception):
pass
class VideoNotFound(Exception): class VideoNotFound(Exception):
pass pass
@ -102,14 +105,20 @@ class Youtube:
def get_user_id(self, username): def get_user_id(self, username):
user = self.youtube.channels().list(part='snippet', forUsername=username).execute() user = self.youtube.channels().list(part='snippet', forUsername=username).execute()
if not user.get('items'):
raise ChannelNotFound(f'username: {username}')
return user['items'][0]['id'] return user['items'][0]['id']
def get_user_name(self, uid): def get_user_name(self, uid):
user = self.youtube.channels().list(part='snippet', id=uid).execute() user = self.youtube.channels().list(part='snippet', id=uid).execute()
if not user.get('items'):
raise ChannelNotFound(f'uid: {uid}')
return user['items'][0]['snippet']['title'] return user['items'][0]['snippet']['title']
def get_user_uploads_playlist_id(self, uid): def get_user_uploads_playlist_id(self, uid):
user = self.youtube.channels().list(part='contentDetails', id=uid).execute() user = self.youtube.channels().list(part='contentDetails', id=uid).execute()
if not user.get('items'):
raise ChannelNotFound(f'uid: {uid}')
return user['items'][0]['contentDetails']['relatedPlaylists']['uploads'] return user['items'][0]['contentDetails']['relatedPlaylists']['uploads']
def get_user_videos(self, uid): def get_user_videos(self, uid):