Skip to content

Commit

Permalink
feat: VoiceClient.play wait_finish parameter for awaiting end of stre…
Browse files Browse the repository at this point in the history
…am (#2194)

* wait_end

* style(pre-commit): auto fixes from pre-commit.com hooks

* Changelog

* style(pre-commit): auto fixes from pre-commit.com hooks

* wait_finish

* Changelog

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Lala Sabathil <lala@pycord.dev>
  • Loading branch information
3 people committed Aug 13, 2023
1 parent 1b81356 commit 017e4ad
Show file tree
Hide file tree
Showing 2 changed files with 29 additions and 1 deletion.
2 changes: 2 additions & 0 deletions CHANGELOG.md
Expand Up @@ -72,6 +72,8 @@ These changes are available on the `master` branch, but have not yet been releas
- Added `suppress` and `allowed_mentions` parameters to `Webhook` and
`InteractionResponse` edit methods.
([#2138](https://github.com/Pycord-Development/pycord/pull/2138))
- Added `wait_finish` parameter to `VoiceClient.play` for awaiting the end of a play.
([#2194](https://github.com/Pycord-Development/pycord/pull/2194))

### Changed

Expand Down
28 changes: 27 additions & 1 deletion discord/voice_client.py
Expand Up @@ -624,7 +624,11 @@ def get_ssrc(self, user_id):
]

def play(
self, source: AudioSource, *, after: Callable[[Exception | None], Any] = None
self,
source: AudioSource,
*,
after: Callable[[Exception | None], Any] = None,
wait_finish: bool = False,
) -> None:
"""Plays an :class:`AudioSource`.
Expand All @@ -643,6 +647,14 @@ def play(
The finalizer that is called after the stream is exhausted.
This function must have a single parameter, ``error``, that
denotes an optional exception that was raised during playing.
wait_finish: bool
If True, an awaitable will be returned, which can be used to wait for
audio to stop playing. This awaitable will return an exception if raised,
or None when no exception is raised.
If False, None is returned and the function does not block.
.. versionadded:: v2.5
Raises
------
Expand All @@ -668,8 +680,22 @@ def play(
if not self.encoder and not source.is_opus():
self.encoder = opus.Encoder()

future = None
if wait_finish:
future = asyncio.Future()
after_callback = after

def _after(exc: Exception | None):
if callable(after_callback):
after_callback(exc)

future.set_result(exc)

after = _after

self._player = AudioPlayer(source, self, after=after)
self._player.start()
return future

def unpack_audio(self, data):
"""Takes an audio packet received from Discord and decodes it into pcm audio data.
Expand Down

0 comments on commit 017e4ad

Please sign in to comment.