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}"
|
return f"{self._piped_api_url}/{path}"
|
||||||
|
|
||||||
def _request(
|
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)
|
timeout = kwargs.pop('timeout', self._timeout)
|
||||||
if auth:
|
if auth:
|
||||||
|
@ -72,9 +77,14 @@ class YoutubePlugin(Plugin):
|
||||||
if body:
|
if body:
|
||||||
kwargs['data'] = 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()
|
rs.raise_for_status()
|
||||||
return rs.json()
|
|
||||||
|
try:
|
||||||
|
return rs.json()
|
||||||
|
except (TypeError, ValueError):
|
||||||
|
return {}
|
||||||
|
|
||||||
@lru_cache(maxsize=10) # noqa
|
@lru_cache(maxsize=10) # noqa
|
||||||
def _get_channel(self, id: str) -> dict: # pylint: disable=redefined-builtin
|
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
|
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:
|
# vim:sw=4:ts=4:et:
|
||||||
|
|
Loading…
Reference in a new issue