Add voice, Change yml, Change media
This commit is contained in:
@@ -25,8 +25,8 @@ class Anime(commands.Cog):
|
||||
Sends a random anime gif or pic
|
||||
"""
|
||||
# Choose either a gif or a pic -> config/media.py
|
||||
media_type = random.choice(loadconfig.__anime_media__)
|
||||
media = random.choice(media)
|
||||
media_type = random.choice(loadconfig.__media_anime__)
|
||||
media = random.choice(media_type)
|
||||
await ctx.send(media)
|
||||
|
||||
@commands.command(name="waifumedia")
|
||||
@@ -34,10 +34,8 @@ class Anime(commands.Cog):
|
||||
"""
|
||||
Sends a random pic of a waifu (list in config/media.py)
|
||||
"""
|
||||
# Dictionary
|
||||
waifus = loadconfig.__waifu_media__
|
||||
try:
|
||||
media = random.choice(waifus.get(waifu))
|
||||
media = random.choice(loadconfig.__media_waifu__[waifu])
|
||||
await ctx.send(media)
|
||||
except KeyError as error:
|
||||
text = (
|
||||
|
||||
@@ -6,10 +6,13 @@ Some (more or less) handy utility:
|
||||
https://discordpy.readthedocs.io/en/latest/ext/commands/cogs.html
|
||||
"""
|
||||
|
||||
# IMPORTS
|
||||
# IMPORTS - external
|
||||
import discord
|
||||
from discord.ext import commands
|
||||
|
||||
# IMPORTS - internal
|
||||
import loadconfig
|
||||
|
||||
# COG INIT
|
||||
class Utility(commands.Cog):
|
||||
def __init__(self, bot):
|
||||
@@ -17,13 +20,34 @@ class Utility(commands.Cog):
|
||||
|
||||
# COG BODY
|
||||
@commands.command(name="invitelink", aliases=["invite"])
|
||||
async def invite_link(self, ctx):
|
||||
async def invite_link(self, ctx, age: int, uses: int, unique: bool):
|
||||
"""
|
||||
Sends the server's invitelink to chat
|
||||
"""
|
||||
# TODO fetch this from config so more servers are supported
|
||||
link = "Here is our invite link: https://discordapp.com/invite/HbYfyJT"
|
||||
await ctx.send(link)
|
||||
if age is None:
|
||||
age = 60
|
||||
if uses is None:
|
||||
uses = 100
|
||||
if unique is None:
|
||||
unique = True
|
||||
|
||||
link = await bot.create_invite(
|
||||
max_age = age,
|
||||
max_uses = uses,
|
||||
unique = uses,
|
||||
reason = "Created by AquaBot")
|
||||
|
||||
link_embed = discord.Embed(color=discord.Colour.blue())
|
||||
link_embed.add_field(
|
||||
name="Here's and invite to our server:",
|
||||
value=link,
|
||||
inline=True)
|
||||
link_embed.set_footer(
|
||||
text=f"Age: {age}, Uses: {uses}, Unique: {unique}",
|
||||
icon_url=loadconfig.__avater__
|
||||
)
|
||||
|
||||
await ctx.send(embed=link_embed)
|
||||
|
||||
|
||||
@commands.command(name="pat")
|
||||
|
||||
@@ -5,7 +5,7 @@ Make Aqua be able to join voice channel and play audio:
|
||||
- play
|
||||
|
||||
https://discordpy.readthedocs.io/en/latest/ext/commands/cogs.html
|
||||
https://stackoverflow.com/questions/56031159/discord-py-rewrite-what-is-the-source-for-youtubedl-to-play-music
|
||||
https://stackoverflow.com/questions/56060614/how-to-make-a-discord-bot-play-youtube-audio
|
||||
"""
|
||||
|
||||
# IMPORTS
|
||||
@@ -59,38 +59,80 @@ class Voice(commands.Cog):
|
||||
await ctx.send("I'm not connected to a channel!")
|
||||
|
||||
|
||||
# Begin of YouTube Player
|
||||
youtube_dl.utils.bug_reports_message = lambda: ''
|
||||
ytdl_format_options = {
|
||||
'format': 'bestaudio/best',
|
||||
'outtmpl': '%(extractor)s-%(id)s-%(title)s.%(ext)s',
|
||||
'restrictfilenames': True,
|
||||
'noplaylist': True,
|
||||
'nocheckcertificate': True,
|
||||
'ignoreerrors': False,
|
||||
'logtostderr': False,
|
||||
'quiet': True,
|
||||
'no_warnings': True,
|
||||
'default_search': 'auto',
|
||||
'source_address': '0.0.0.0'
|
||||
}
|
||||
ffmpeg_options = {
|
||||
'options': '-vn'
|
||||
}
|
||||
ytdl = youtube_dl.YoutubeDL(ytdl_format_options)
|
||||
class YTDLSource(discord.PCMVolumeTransformer):
|
||||
def __init__(self, source, *, data, volume=0.5):
|
||||
super().__init__(source, volume)
|
||||
|
||||
self.data = data
|
||||
|
||||
self.title = data.get('title')
|
||||
self.url = data.get('url')
|
||||
|
||||
# Maybe i can make a fancy embed out of this?
|
||||
self.uploader = data.get('uploader')
|
||||
self.uploader_url = data.get('uploader_url')
|
||||
date = data.get('upload_date')
|
||||
self.upload_date = date[6:8] + '.' + date[4:6] + '.' + date[0:4]
|
||||
self.title = data.get('title')
|
||||
self.thumbnail = data.get('thumbnail')
|
||||
self.description = data.get('description')
|
||||
self.duration = self.parse_duration(int(data.get('duration')))
|
||||
self.tags = data.get('tags')
|
||||
self.url = data.get('webpage_url')
|
||||
self.views = data.get('view_count')
|
||||
self.likes = data.get('like_count')
|
||||
self.dislikes = data.get('dislike_count')
|
||||
self.stream_url = data.get('url')
|
||||
|
||||
@classmethod
|
||||
async def from_url(cls, url, *, loop=None, stream=False):
|
||||
loop = loop or asyncio.get_event_loop()
|
||||
data = await loop.run_in_executor(
|
||||
None,
|
||||
lambda: ytdl.extract_info(url, download=not stream))
|
||||
|
||||
if 'entries' in data:
|
||||
# take first item from a playlist
|
||||
data = data['entries'][0]
|
||||
|
||||
filename = data['url'] if stream else ytdl.prepare_filename(data)
|
||||
return cls(discord.FFmpegPCMAudio(filename, **ffmpeg_options), data=data)
|
||||
|
||||
@commands.command(name="play", aliases=["p"])
|
||||
@commands.guild_only()
|
||||
async def play(self, ctx, url: str):
|
||||
"""
|
||||
Plays music from YT link specifies
|
||||
Plays music from YouTube
|
||||
"""
|
||||
# TODO
|
||||
try:
|
||||
if os.path.isfile("song.mp3"):
|
||||
os.remove("song.mp3")
|
||||
except PermissionError:
|
||||
await ctx.send("Wait for the current song to end or use the `stop`command")
|
||||
return
|
||||
|
||||
voice = get(bot.voice_clients, guild=ctx.guild)
|
||||
youtube_dl_opts = {
|
||||
'format': 'bestaudio/best',
|
||||
'postprocessors': [{
|
||||
'key': 'FFmpegExtractAudio',
|
||||
'preferredcodec': 'mp3',
|
||||
'preferredquality': '192',
|
||||
}],
|
||||
}
|
||||
with youtube_dl.YouTubeDL(youtube_dl_opts) as ydl:
|
||||
ydl.download([url])
|
||||
for file in os.listdir("./"):
|
||||
if file.endswith(".mp3"):
|
||||
os.rename(file, "song.mp3")
|
||||
|
||||
voice.play(discord.FFmpegPCMAudio("song.mp3"))
|
||||
voice.volume=25
|
||||
voice.is_playing()
|
||||
async with ctx.typing():
|
||||
player = await YTDLSource.from_url(url, loop=self.bot.loop)
|
||||
ctx.voice.play(
|
||||
player,
|
||||
after=lambda e: print('Player error: %s' % e) if e else None))
|
||||
await ctx.send(f"Now playing: {player.title}")
|
||||
|
||||
# End of YouTube Player
|
||||
|
||||
|
||||
# COG ENDING
|
||||
|
||||
@@ -26,7 +26,7 @@ class Welcome(commands.Cog):
|
||||
text = f"Welcome {member.mention} to our useless Discord!"
|
||||
if channel is not None:
|
||||
await channel.send(text)
|
||||
|
||||
await message.add_reaction("\N{THUMBS UP SIGN}")
|
||||
|
||||
@commands.command(name="hello")
|
||||
async def hello(self, ctx):
|
||||
|
||||
Reference in New Issue
Block a user