mirror of
https://github.com/Rapptz/discord.py.git
synced 2025-06-04 11:02:37 +00:00
Add voice playlist example code.
This commit is contained in:
parent
1ac2f6437f
commit
9175b83387
105
examples/playlist.py
Normal file
105
examples/playlist.py
Normal file
@ -0,0 +1,105 @@
|
||||
import asyncio
|
||||
import discord
|
||||
|
||||
if not discord.opus.is_loaded():
|
||||
# the 'opus' library here is opus.dll on windows
|
||||
# or libopus.so on linux in the current directory
|
||||
# you should replace this with the location the
|
||||
# opus library is located in.
|
||||
discord.opus.load_opus('opus')
|
||||
|
||||
class VoiceEntry:
|
||||
def __init__(self, message, song):
|
||||
self.requester = message.author
|
||||
self.channel = message.channel
|
||||
self.song = song
|
||||
|
||||
class Bot(discord.Client):
|
||||
def __init__(self):
|
||||
super().__init__()
|
||||
self.songs = asyncio.Queue()
|
||||
self.play_next_song = asyncio.Event()
|
||||
self.starter = None
|
||||
self.player = None
|
||||
self.current = None
|
||||
|
||||
def toggle_next_song(self):
|
||||
self.loop.call_soon_threadsafe(self.play_next_song.set)
|
||||
|
||||
def can_control_song(self, author):
|
||||
return author == self.starter or (self.current is not None and author == self.current.requester)
|
||||
|
||||
def is_playing(self):
|
||||
return self.player is not None and self.player.is_playing()
|
||||
|
||||
@asyncio.coroutine
|
||||
def on_message(self, message):
|
||||
if message.author == self.user:
|
||||
return
|
||||
|
||||
if message.channel.is_private:
|
||||
yield from self.send_message(message.channel, 'You cannot use this bot in private messages.')
|
||||
|
||||
if message.content.startswith('$join'):
|
||||
if self.is_voice_connected():
|
||||
yield from self.send_message(message.channel, 'Already connected to a voice channel')
|
||||
channel_name = message.content[5:].strip()
|
||||
check = lambda c: c.name == channel_name and c.type == discord.ChannelType.voice
|
||||
|
||||
channel = discord.utils.find(check, message.server.channels)
|
||||
if channel is None:
|
||||
yield from self.send_message(message.channel, 'Cannot find a voice channel by that name.')
|
||||
|
||||
yield from self.join_voice_channel(channel)
|
||||
self.starter = message.author
|
||||
|
||||
elif message.content.startswith('$leave'):
|
||||
if not self.can_control_song(message.author):
|
||||
return
|
||||
self.starter = None
|
||||
yield from self.voice.disconnect()
|
||||
elif message.content.startswith('$pause'):
|
||||
if not self.can_control_song(message.author):
|
||||
fmt = 'Only the requester ({0.current.requester}) can control this song'
|
||||
yield from self.send_message(message.channel, fmt.format(self))
|
||||
|
||||
if self.player.is_playing():
|
||||
self.player.pause()
|
||||
elif message.content.startswith('$resume'):
|
||||
if not self.can_control_song(message.author):
|
||||
fmt = 'Only the requester ({0.current.requester}) can control this song'
|
||||
yield from self.send_message(message.channel, fmt.format(self))
|
||||
|
||||
if self.player is not None and not self.is_playing():
|
||||
self.player.resume()
|
||||
elif message.content.startswith('$next'):
|
||||
filename = message.content[5:].strip()
|
||||
yield from self.songs.put(VoiceEntry(message, filename))
|
||||
yield from self.send_message(message.channel, 'Successfully registered {}'.format(filename))
|
||||
elif message.content.startswith('$play'):
|
||||
if self.player is not None and self.player.is_playing():
|
||||
yield from self.send_message(message.channel, 'Already playing a song')
|
||||
return
|
||||
while True:
|
||||
if not self.is_voice_connected():
|
||||
yield from self.send_message(message.channel, 'Not connected to a voice channel')
|
||||
return
|
||||
|
||||
self.play_next_song.clear()
|
||||
self.current = yield from self.songs.get()
|
||||
self.player = self.voice.create_ffmpeg_player(self.current.song, after=self.toggle_next_song)
|
||||
self.player.start()
|
||||
fmt = 'Playing song "{0.song}" from {0.requester}'
|
||||
yield from self.send_message(self.current.channel, fmt.format(self.current))
|
||||
yield from self.play_next_song.wait()
|
||||
|
||||
@asyncio.coroutine
|
||||
def on_ready(self):
|
||||
print('Logged in as')
|
||||
print(self.user.name)
|
||||
print(self.user.id)
|
||||
print('------')
|
||||
|
||||
|
||||
bot = Bot()
|
||||
bot.run('email', 'password')
|
Loading…
x
Reference in New Issue
Block a user