dj_rakosbeka/main.py
2025-05-01 12:22:58 +02:00

116 lines
No EOL
3.8 KiB
Python

import discord
from discord.ext import commands
from discord import FFmpegPCMAudio
import yt_dlp
# Set up intents and bot
intents = discord.Intents.default()
intents.message_content = True # Needed to read .play/.stop commands
bot = commands.Bot(command_prefix="..", intents=intents)
# Global variable to hold the current voice clients
voice_clients = {}
# Sets the default volume to 20%, to make headphone users survive
# the volume does not change when the bot is already playing, only new audio
current_volume = 0.20
@bot.event
async def on_ready():
# Construct the invite URL dynamically
application_info = await bot.application_info()
client_id = application_info.id
invite_url = f"https://discord.com/oauth2/authorize?client_id={client_id}&permissions=3145728&scope=bot"
print(f"Invite URL: {invite_url}")
@bot.command()
async def play(ctx, url: str):
guild_id = ctx.guild.id
if not ctx.author.voice:
await ctx.send("lépjél már be egy kurva csatornába te szerencsétlen fasz")
return
if guild_id not in voice_clients or not voice_clients[guild_id].is_connected():
voice_clients[guild_id] = await ctx.author.voice.channel.connect()
vc = voice_clients[guild_id]
if vc.is_playing():
vc.stop()
try:
# Load blocked keywords from file
with open("blocked_keywords.txt", "r", encoding="utf-8") as f:
blocked_keywords = [line.strip().lower() for line in f if line.strip()]
# YTDLP options
ydl_opts = {
"format": "bestaudio/best",
"quiet": True,
"noplaylist": True,
"default_search": "auto",
}
# Use yt-dlp to extract audio URL and check for blocked keywords
with yt_dlp.YoutubeDL(ydl_opts) as ydl:
info = ydl.extract_info(url, download=False)
title = info["title"].lower()
if any(keyword in title for keyword in blocked_keywords):
await ctx.send(f"na ezt a fost nem jatszom le: `{info['title']}`")
return
audio_url = info["url"]
print("Audio URL:", audio_url)
# Set up FFmpeg options with the current volume
ffmpeg_options = {
"before_options": "-reconnect 1 -reconnect_streamed 1 -reconnect_delay_max 5 -protocol_whitelist file,http,https,tcp,tls,crypto",
"options": f"-vn -af volume={current_volume}"
}
# Play the audio
vc.play(
FFmpegPCMAudio(audio_url, **ffmpeg_options),
after=lambda e: print(f"Finished playing: {e}")
)
await ctx.send(f"most ez a fos szól: {info['title']}")
except Exception as e:
await ctx.send(f"most ezzel mi a kurva anyámat kezdjek?: {e}")
@bot.command()
async def stop(ctx):
guild_id = ctx.guild.id
if guild_id in voice_clients and voice_clients[guild_id].is_playing():
voice_clients[guild_id].stop()
await ctx.send("hát akkor ne hallgassad basszalak szájba")
else:
await ctx.send("nem is szól semmi te buzeráns fasz")
#Does not work in multi-server setup, needs fix
@bot.command()
async def volume(ctx, vol: float):
global current_volume
if 0.0 <= vol <= 2.0:
current_volume = vol
await ctx.send(f"hangerő most {vol*100:.0f}%")
else:
await ctx.send("0.0 és 2.0 között add meg te szerencsétlen")
@bot.command()
async def leave(ctx):
guild_id = ctx.guild.id
if guild_id in voice_clients and voice_clients[guild_id].is_connected():
await voice_clients[guild_id].disconnect()
del voice_clients[guild_id]
await ctx.send("akkor csá gecik")
else:
await ctx.send("nem is vagyok voice channelben basszam ki a szádat")
# Run the bot
TOKEN = open("TOKEN").read().strip()
bot.run(TOKEN)