Compare commits
2 commits
b2b59d651c
...
8cea668e37
Author | SHA1 | Date | |
---|---|---|---|
8cea668e37 | |||
b890326e71 |
2 changed files with 166 additions and 3 deletions
Binary file not shown.
|
@ -60,7 +60,12 @@ class YoutubePlugin(Plugin):
|
|||
return f"{self._piped_api_url}/{path}"
|
||||
|
||||
def _request(
|
||||
self, path: str, body: Optional[str] = None, auth: bool = True, **kwargs
|
||||
self,
|
||||
path: str,
|
||||
method: str = 'get',
|
||||
body: Optional[str] = None,
|
||||
auth: bool = True,
|
||||
**kwargs,
|
||||
):
|
||||
timeout = kwargs.pop('timeout', self._timeout)
|
||||
if auth:
|
||||
|
@ -72,9 +77,14 @@ class YoutubePlugin(Plugin):
|
|||
if body:
|
||||
kwargs['data'] = body
|
||||
|
||||
rs = requests.get(self._api_url(path), timeout=timeout, **kwargs)
|
||||
func = getattr(requests, method.lower())
|
||||
rs = func(self._api_url(path), timeout=timeout, **kwargs)
|
||||
rs.raise_for_status()
|
||||
return rs.json()
|
||||
|
||||
try:
|
||||
return rs.json()
|
||||
except (TypeError, ValueError):
|
||||
return {}
|
||||
|
||||
@lru_cache(maxsize=10) # noqa
|
||||
def _get_channel(self, id: str) -> dict: # pylint: disable=redefined-builtin
|
||||
|
@ -196,5 +206,158 @@ class YoutubePlugin(Plugin):
|
|||
|
||||
return PipedChannelSchema().dump(response) or {} # type: ignore
|
||||
|
||||
@action
|
||||
def add_to_playlist(self, video_id: str, playlist_id: str):
|
||||
"""
|
||||
Add a video to a playlist.
|
||||
|
||||
:param video_id: YouTube video ID.
|
||||
:param playlist_id: Piped playlist ID.
|
||||
"""
|
||||
self._request(
|
||||
'playlists/add',
|
||||
method='post',
|
||||
json={
|
||||
'videoIds': [video_id],
|
||||
'playlistId': playlist_id,
|
||||
},
|
||||
)
|
||||
|
||||
@action
|
||||
def remove_from_playlist(
|
||||
self,
|
||||
playlist_id: str,
|
||||
video_id: Optional[str] = None,
|
||||
index: Optional[int] = None,
|
||||
):
|
||||
"""
|
||||
Remove a video from a playlist.
|
||||
|
||||
Note that either the video ID or the index must be provided.
|
||||
|
||||
:param video_id: YouTube video ID.
|
||||
:param index: (0-based) index of the video in the playlist.
|
||||
:param playlist_id: Piped playlist ID.
|
||||
"""
|
||||
assert video_id or index, 'Either the video ID or the index must be provided'
|
||||
|
||||
if index is None:
|
||||
index = next(
|
||||
(
|
||||
i
|
||||
for i, v in enumerate(
|
||||
self._request(f'playlists/{playlist_id}').get(
|
||||
'relatedStreams', []
|
||||
)
|
||||
)
|
||||
if v.get('id') == video_id
|
||||
),
|
||||
None,
|
||||
)
|
||||
|
||||
if index is None:
|
||||
self.logger.warning(
|
||||
'Video %s not found in the playlist %s', video_id, playlist_id
|
||||
)
|
||||
return
|
||||
|
||||
self._request(
|
||||
'playlists/remove',
|
||||
method='post',
|
||||
json={
|
||||
'index': index,
|
||||
'playlistId': playlist_id,
|
||||
},
|
||||
)
|
||||
|
||||
@action
|
||||
def create_playlist(self, name: str) -> dict:
|
||||
"""
|
||||
Create a new playlist.
|
||||
|
||||
:param name: Playlist name.
|
||||
:return: Playlist information.
|
||||
"""
|
||||
playlist_id = self._request(
|
||||
'playlists/create',
|
||||
method='post',
|
||||
json={'name': name},
|
||||
).get('playlistId')
|
||||
|
||||
assert playlist_id, 'Failed to create the playlist'
|
||||
playlists = self._request('user/playlists')
|
||||
new_playlist = next((p for p in playlists if p.get('id') == playlist_id), None)
|
||||
|
||||
assert new_playlist, 'Failed to retrieve the new playlist'
|
||||
return dict(PipedPlaylistSchema().dump(new_playlist) or {})
|
||||
|
||||
@action
|
||||
def rename_playlist(
|
||||
self, id: str, name: Optional[str] = None, description: Optional[str] = None
|
||||
): # pylint: disable=redefined-builtin
|
||||
"""
|
||||
Rename a playlist.
|
||||
|
||||
:param id: Piped playlist ID.
|
||||
:param name: New playlist name.
|
||||
:param description: New playlist description.
|
||||
"""
|
||||
args = {}
|
||||
if name:
|
||||
args['newName'] = name
|
||||
if description:
|
||||
args['description'] = description
|
||||
if not args:
|
||||
self.logger.info('No new name or description provided')
|
||||
return
|
||||
|
||||
self._request(
|
||||
'playlists/rename',
|
||||
method='post',
|
||||
json={
|
||||
'playlistId': id,
|
||||
**args,
|
||||
},
|
||||
)
|
||||
|
||||
@action
|
||||
def delete_playlist(self, id: str): # pylint: disable=redefined-builtin
|
||||
"""
|
||||
Delete a playlist.
|
||||
|
||||
:param id: Piped playlist ID.
|
||||
"""
|
||||
self._request(
|
||||
'playlists/delete',
|
||||
method='post',
|
||||
json={'playlistId': id},
|
||||
)
|
||||
|
||||
@action
|
||||
def subscribe(self, channel_id: str):
|
||||
"""
|
||||
Subscribe to a channel.
|
||||
|
||||
:param channel_id: YouTube channel ID.
|
||||
"""
|
||||
self._request(
|
||||
'subscribe',
|
||||
method='post',
|
||||
json={'channelId': channel_id},
|
||||
)
|
||||
|
||||
@action
|
||||
def unsubscribe(self, channel_id: str):
|
||||
"""
|
||||
Unsubscribe from a channel.
|
||||
|
||||
:param channel_id: YouTube channel ID.
|
||||
"""
|
||||
self._request(
|
||||
'unsubscribe',
|
||||
method='post',
|
||||
json={'channelId': channel_id},
|
||||
)
|
||||
|
||||
|
||||
# vim:sw=4:ts=4:et:
|
||||
|
|
Loading…
Reference in a new issue