ekzyis
/
musicube
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Rename stream to play

This commit is contained in:
ekzyis 2021-09-24 23:48:12 +02:00
parent 514c6c15c2
commit 07f8f3e524
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/bot.py
View File

@ -15,7 +15,7 @@ class Music(commands.Cog):
self.bot = bot self.bot = bot
@commands.command() @commands.command()
async def stream(self, ctx, *, url): async def play(self, ctx, *, url):
async with ctx.typing(): async with ctx.typing():
player = await YTDLSource.from_url(url, loop=self.bot.loop, stream=True) player = await YTDLSource.from_url(url, loop=self.bot.loop, stream=True)
ctx.voice_client.play(player, after=lambda e: print(f"Player error: {e}") if e else None) ctx.voice_client.play(player, after=lambda e: print(f"Player error: {e}") if e else None)
@ -26,7 +26,7 @@ class Music(commands.Cog):
async def stop(self, ctx): async def stop(self, ctx):
await ctx.voice_client.disconnect() await ctx.voice_client.disconnect()
@stream.before_invoke @play.before_invoke
async def ensure_voice(self, ctx): async def ensure_voice(self, ctx):
if ctx.voice_client is None: if ctx.voice_client is None:
if ctx.author.voice: if ctx.author.voice:

4
test/test_bot.py
View File

@ -39,7 +39,7 @@ async def test_bot_ensure_voice(bot, ctx):
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_bot_stream(bot, ctx): async def test_bot_play(bot, ctx):
mbot = Music(bot) mbot = Music(bot)
with patch('bot.YTDLSource', new_callable=AsyncMock) as ytdl_source: with patch('bot.YTDLSource', new_callable=AsyncMock) as ytdl_source:
@ -47,7 +47,7 @@ async def test_bot_stream(bot, ctx):
ytdl_source.from_url.return_value = player ytdl_source.from_url.return_value = player
url = 'A Day To Remember - All I Want' url = 'A Day To Remember - All I Want'
# pylint: disable=too-many-function-args # pylint: disable=too-many-function-args
await mbot.stream(mbot, ctx, url=url) await mbot.play(mbot, ctx, url=url)
assert ytdl_source.from_url.await_args.args == (url,) assert ytdl_source.from_url.await_args.args == (url,)
assert ytdl_source.from_url.await_args.kwargs == {"loop": bot.loop, "stream": True} assert ytdl_source.from_url.await_args.kwargs == {"loop": bot.loop, "stream": True}
assert ctx.voice_client.play.call_args.args == (player,) assert ctx.voice_client.play.call_args.args == (player,)