7 Commits
A-0.3 ... main

Author SHA1 Message Date
17cf18160c Some changes 2024-01-29 14:46:23 +01:00
Lukas Blacha
ab03ebb4da Still fixing the reactions 2023-02-11 01:49:03 +01:00
Lukas Blacha
d38d93e8cb Added re import 2023-02-11 01:32:33 +01:00
Lukas Blacha
c940d735d2 Added re import 2023-02-11 01:00:33 +01:00
Lukas Blacha
53109d68f9 Deleted unused import 2023-02-11 00:59:50 +01:00
Lukas Blacha
f55d45232c Fixed Reactions 2023-02-10 19:28:10 +01:00
Lukas Blacha
be7002bba5 Preps for youtube search selection 2023-02-10 15:18:24 +01:00

View File

@@ -1,8 +1,7 @@
import asyncio import asyncio
import datetime
import re import re
from datetime import timedelta from pytube import YouTube
from urllib.parse import quote_plus
import aiohttp import aiohttp
import yt_dlp import yt_dlp
import discord import discord
@@ -33,13 +32,13 @@ parser.add_argument("-p", "--prefix", help="Provides the Prefix for the bot (Sta
args = parser.parse_args() args = parser.parse_args()
if args.cores == None: if args.cores is None:
threads = 2 threads = 2
else: else:
threads = int(args.cores) threads = int(args.cores)
if args.prefix == None: if args.prefix is None:
b_prefix = "<" b_prefix = "<"
else: else:
b_prefix = str(args.prefix) b_prefix = str(args.prefix)
@@ -110,9 +109,8 @@ else:
async with aiohttp.ClientSession() as client: async with aiohttp.ClientSession() as client:
async with client.get("https://www.youtube.com/results", params=p, headers=h) as resp: async with client.get("https://www.youtube.com/results", params=p, headers=h) as resp:
dom = await resp.text() dom = await resp.text()
found = re.findall(r"watch\?v=(\S{11})", dom)[0] result = list(set(re.findall(r"watch\?v=(\S{11})", dom)))[0:5]
url = f"https://www.youtube.com/watch?v={found}" return result
return url
intents = discord.Intents.default() intents = discord.Intents.default()
@@ -120,7 +118,7 @@ else:
bot = commands.Bot( bot = commands.Bot(
command_prefix=commands.when_mentioned_or(b_prefix), command_prefix=commands.when_mentioned_or(b_prefix),
description="Relatively simple music bot example", description="BCA Music-Bot",
intents=intents, help_command=None intents=intents, help_command=None
) )
@@ -133,7 +131,8 @@ else:
print(f"Threads for ffmpeg: {threads}") print(f"Threads for ffmpeg: {threads}")
print(f"Prefix: {b_prefix}") print(f"Prefix: {b_prefix}")
print( print(
f"Invitation LINK: https://discord.com/api/oauth2/authorize?client_id={bot.user.id}&permissions=968552344896&scope=bot%20applications.commands") f"Invitation LINK: https://discord.com/api/oauth2/authorize?client_id={bot.user.id}"
f"&permissions=968552344896&scope=bot%20applications.commands")
print("------") print("------")
@@ -197,7 +196,7 @@ else:
while ctx.voice_client.is_playing(): while ctx.voice_client.is_playing():
await asyncio.sleep(3) await asyncio.sleep(3)
else: else:
await ctx.voice_client.disconnect() await ctx.voice_client.disconnect(force=True)
@bot.slash_command(name="stream", description="Streams a song from YouTube [Without preloading]") @bot.slash_command(name="stream", description="Streams a song from YouTube [Without preloading]")
@@ -242,7 +241,61 @@ else:
@bot.slash_command(name="youtube", description="Searches YouTube for a given string and plays the first result") @bot.slash_command(name="youtube", description="Searches YouTube for a given string and plays the first result")
async def youtube(ctx, *, search: str): async def youtube(ctx, *, search: str):
url = await youtube_search(search=search) results = await youtube_search(search=search)
embed = Embed(title=f"", color=discord.Color.green())
embed.add_field(name="", value="", inline=False)
counter = 1
for result in results:
url = f"https://youtube.com/watch?v={result}"
video = YouTube(url)
embed.add_field(name="", value=f"{counter} | `[{str(datetime.timedelta(seconds=video.length))}]` "
f"[{video.title}]({url}{result[counter-1]}) {video.author}", inline=False)
counter += 1
await ctx.respond(f"{success}**🎶 Search results for `{search}`**")
msg = await ctx.send(embed=embed)
await msg.add_reaction("1")
await msg.add_reaction("2")
await msg.add_reaction("3")
await msg.add_reaction("4")
await msg.add_reaction("5")
await msg.add_reaction("")
try:
reaction = await bot.wait_for(event='reaction_add', timeout=60.0)
except asyncio.TimeoutError:
await ctx.send('👎')
else:
vid_id = "https://youtube.com/watch?v="
if str(reaction[0]) == "1":
vid_id += results[0]
if str(reaction[0]) == "2":
vid_id += results[1]
if str(reaction[0]) == "3":
vid_id += results[2]
if str(reaction[0]) == "5":
vid_id += results[3]
if str(reaction[0]) == "5":
vid_id += results[4]
else:
vid_id = None
print(vid_id)
if vid_id is None:
pass
else:
try:
player = await YTDLSource.from_url(vid_id, loop=bot.loop)
ctx.voice_client.play(
player, after=lambda e: print(f"Player error: {e}") if e else None
)
except discord.HTTPException as err:
raise commands.CommandError(err)
finally:
await ctx.send(f"{success}Now playing: `{player.title}`\nRequested by {ctx.author.mention}")
while ctx.voice_client.is_playing():
await asyncio.sleep(3)
else:
await ctx.voice_client.disconnect(force=True)
"""
async with ctx.typing(): async with ctx.typing():
await ctx.respond("🤖 Your song is queued for download... please wait ", ephemeral=True) await ctx.respond("🤖 Your song is queued for download... please wait ", ephemeral=True)
try: try:
@@ -258,11 +311,12 @@ else:
await asyncio.sleep(3) await asyncio.sleep(3)
else: else:
await ctx.voice_client.disconnect() await ctx.voice_client.disconnect()
"""
@play.before_invoke @play.before_invoke
@stream.before_invoke @stream.before_invoke
@youtube.before_invoke #@youtube.before_invoke
async def ensure_voice(ctx: commands.Context): async def ensure_voice(ctx: commands.Context):
if ctx.voice_client is None: if ctx.voice_client is None:
if ctx.author.voice: if ctx.author.voice:
@@ -274,7 +328,7 @@ else:
"""
@bot.event @bot.event
async def on_application_command_error(ctx, error): async def on_application_command_error(ctx, error):
print(f"[on_application_command_error]\n{ctx.author}\n{error}") print(f"[on_application_command_error]\n{ctx.author}\n{error}")
@@ -299,5 +353,6 @@ else:
print(f"[on_command_error]\n{ctx.author}\n{error}") print(f"[on_command_error]\n{ctx.author}\n{error}")
embed = Embed(title=f"{error}", color=15158332) embed = Embed(title=f"{error}", color=15158332)
await ctx.send(embed=embed) await ctx.send(embed=embed)
"""
bot.run(token=args.token, reconnect=True) bot.run(token=args.token, reconnect=True)