christoph/discord-heidi
1
Fork 0
Code Issues 9 Actions Packages Releases 3 Activity

Take current input into account for autocomplete + reformat
All checks were successful
Build Heidi Docker image / build-docker (push) Successful in 17s
Details

Browse Source
This commit is contained in:
Christoph Urlacher
2023-11-25 19:37:02 +01:00
parent 273b3ff406
commit a0e184f2fc
1 changed files with 108 additions and 63 deletions
Download Patch File Download Diff File Expand all files Collapse all files

171
bot.py
View File

@ -5,7 +5,7 @@ from discord import app_commands
from discord.app_commands import Choice from discord.app_commands import Choice
from functools import reduce from functools import reduce
from dotenv import load_dotenv from dotenv import load_dotenv
from typing import Optional, Union from typing import List, Optional, Union
# TODO: Reenable + extend textgen # TODO: Reenable + extend textgen
# from textgen import textgen # from textgen import textgen
@ -17,28 +17,27 @@ from typing import Optional, Union
# We're fancy today # We're fancy today
from rich.traceback import install from rich.traceback import install
install(show_locals=True) install(show_locals=True)
# ================================================================================================ # # ================================================================================================ #
# ================================================================================================ # # ================================================================================================ #
# NOTE: Always set this correctly: # # NOTE: Always set this correctly: #
DOCKER = True # DOCKER = False #
# ================================================================================================ # # ================================================================================================ #
# ================================================================================================ # # ================================================================================================ #
# DONE: Migrate back to discord.py # DONE: Migrate back to discord.py
# DONE: Rewrite bot with slash commands (and making actual use of discord.py) # DONE: Rewrite bot with slash commands (and making actual use of discord.py)
# TODO: Insult statistics (you have insulted 20 times)
# TODO: Only post in heidi-spam channel # TODO: Only post in heidi-spam channel
# TODO: yt-dlp music support # TODO: yt-dlp music support
# TODO: Somehow upload voicelines more easily (from discord voice message?), also need to be distributed to folders so no more than 25 lines per folder # TODO: Somehow upload voicelines more easily (from discord voice message?)
# TODO: Reenable text/quote generation, allow uploading of training text files, allow switching "personalities" (/elon generates elon quote?)
# TODO: Zalgo generator
# IDs of the servers Heidi is used on # IDs of the servers Heidi is used on
LINUS_GUILD = discord.Object(id=431154792308408340) LINUS_GUILD = discord.Object(id=431154792308408340)
TEST_GUILD = discord.Object(id=821511861178204161) TEST_GUILD = discord.Object(id=821511861178204161)
class HeidiClient(discord.Client): class HeidiClient(discord.Client):
def __init__(self, *, intents: discord.Intents): def __init__(self, *, intents: discord.Intents):
super().__init__(status="Nur eine kann GNTM werden!", intents=intents) super().__init__(status="Nur eine kann GNTM werden!", intents=intents)
@ -53,33 +52,33 @@ class HeidiClient(discord.Client):
# if the predicate is true the action is performed # if the predicate is true the action is performed
self.auto_triggers = { self.auto_triggers = {
# lambda m: m.author.nick.lower() in self.models.get_in_names(): self.autoreact_to_girls, # lambda m: m.author.nick.lower() in self.models.get_in_names(): self.autoreact_to_girls,
lambda m: "jeremy" in m.author.nick.lower(): self._autoreact_to_jeremy lambda m: "jeremy"
in m.author.nick.lower(): self._autoreact_to_jeremy
} }
# Textgen # Textgen
self.textgen_models: dict[str, textgen] = { # self.textgen_models: dict[str, textgen] = {
# The name must correspond to the name of the training text file # # The name must correspond to the name of the training text file
# "kommunistisches_manifest": LSTMTextGenerator(10), # "kommunistisches_manifest": LSTMTextGenerator(10),
# "musk": LSTMTextGenerator(10), # "musk": LSTMTextGenerator(10),
# "bibel": LSTMTextGenerator(10) # "bibel": LSTMTextGenerator(10)
# "bibel": MarkovTextGenerator(3), # Prefix length of 3
# "kommunistisches_manifest": MarkovTextGenerator(3),
# "musk": MarkovTextGenerator(3)
# }
# "bibel": MarkovTextGenerator(3), # Prefix length of 3 # for name, model in self.textgen_models.items():
# "kommunistisches_manifest": MarkovTextGenerator(3), # model.init(name) # Loads the textfile
# "musk": MarkovTextGenerator(3)
}
for name, model in self.textgen_models.items(): # if os.path.exists(f"weights/{name}_lstm_model.pt"):
model.init(name) # Loads the textfile # model.load()
# elif not DOCKER:
# model.train()
# else:
# print("Error: Can't load model", name)
if os.path.exists(f"weights/{name}_lstm_model.pt"): # print("Generating test sentence for", name)
model.load() # self.textgen_models[name].generate_sentence()
elif not DOCKER:
model.train()
else:
print("Error: Can't load model", name)
print("Generating test sentence for", name)
self.textgen_models[name].generate_sentence()
# Synchronize commands to guilds # Synchronize commands to guilds
async def setup_hook(self): async def setup_hook(self):
@ -113,7 +112,6 @@ class HeidiClient(discord.Client):
# picture.set_footer(text=name) # picture.set_footer(text=name)
# await message.channel.send(embed=picture) # await message.channel.send(embed=picture)
# Automatic Actions -------------------------------------------------------------------------- # Automatic Actions --------------------------------------------------------------------------
# @staticmethod # @staticmethod
@ -130,15 +128,16 @@ class HeidiClient(discord.Client):
""" """
await message.add_reaction("🧀") await message.add_reaction("🧀")
# ------------------------------------------------------------------------------------------------ # ------------------------------------------------------------------------------------------------
# Log to file # Log to file
handler = logging.FileHandler(filename='discord.log', encoding='utf-8', mode='w') handler = logging.FileHandler(filename="discord.log", encoding="utf-8", mode="w")
# Intents specification is no longer optional # Intents specification is no longer optional
intents = discord.Intents.default() intents = discord.Intents.default()
intents.members = True # Allow to react to member join/leave etc intents.members = True # Allow to react to member join/leave etc
intents.message_content = True # Allow to read message content from arbitrary messages intents.message_content = True # Allow to read message content from arbitrary messages
# Setup our client # Setup our client
client = HeidiClient(intents=intents) client = HeidiClient(intents=intents)
@ -146,9 +145,14 @@ client = HeidiClient(intents=intents)
# Events ----------------------------------------------------------------------------------------- # Events -----------------------------------------------------------------------------------------
# NOTE: I defined the events outside of the Client class, don't know if I like it or not... # NOTE: I defined the events outside of the Client class, don't know if I like it or not...
@client.event @client.event
async def on_ready(): async def on_ready():
print(f"{client.user} (id: {client.user.id}) has connected to Discord!") if client.user != None:
print(f"{client.user} (id: {client.user.id}) has connected to Discord!")
else:
print(f"client.user is None!")
@client.event @client.event
async def on_message(message): async def on_message(message):
@ -164,14 +168,22 @@ async def on_message(message):
await action(message) await action(message)
break break
# Commands --------------------------------------------------------------------------------------- # Commands ---------------------------------------------------------------------------------------
@client.tree.command(name = "giblinkbruder", description = "Heidi hilft mit dem Link zu deiner Lieblingsshow im Qualitätsfernsehen.")
@client.tree.command(
name="giblinkbruder",
description="Heidi hilft mit dem Link zu deiner Lieblingsshow im Qualitätsfernsehen.",
)
async def show_link(interaction: discord.Interaction): async def show_link(interaction: discord.Interaction):
link_pro7 = "https://www.prosieben.de/tv/germanys-next-topmodel/livestream" link_pro7 = "https://www.prosieben.de/tv/germanys-next-topmodel/livestream"
link_joyn = "https://www.joyn.de/serien/germanys-next-topmodel" link_joyn = "https://www.joyn.de/serien/germanys-next-topmodel"
await interaction.response.send_message(f"ProSieben: {link_pro7}\nJoyn: {link_joyn}") await interaction.response.send_message(
f"ProSieben: {link_pro7}\nJoyn: {link_joyn}"
)
@client.tree.command(name="heidi", description="Heidi!") @client.tree.command(name="heidi", description="Heidi!")
async def heidi_exclaim(interaction: discord.Interaction): async def heidi_exclaim(interaction: discord.Interaction):
@ -182,13 +194,14 @@ async def heidi_exclaim(interaction: discord.Interaction):
"Dann zieh dich mal aus!", "Dann zieh dich mal aus!",
"Warum denn so schüchtern?", "Warum denn so schüchtern?",
"Im TV ist das legal!", "Im TV ist das legal!",
"Das Stroh ist nur fürs Shooting!" "Das Stroh ist nur fürs Shooting!",
] ]
await interaction.response.send_message(random.choice(messages)) await interaction.response.send_message(random.choice(messages))
@client.tree.command(name="miesmuschel", description="Was denkt Heidi?") @client.tree.command(name="miesmuschel", description="Was denkt Heidi?")
@app_commands.rename(question = "frage") @app_commands.rename(question="frage")
@app_commands.describe(question = "Heidi wird es beantworten!") @app_commands.describe(question="Heidi wird es beantworten!")
async def magic_shell(interaction: discord.Interaction, question: str): async def magic_shell(interaction: discord.Interaction, question: str):
choices = [ choices = [
"Ja!", "Ja!",
@ -202,17 +215,23 @@ async def magic_shell(interaction: discord.Interaction, question: str):
] ]
question = question.strip() question = question.strip()
question_mark = "" if question[-1] == "?" else "?" question_mark = "" if question[-1] == "?" else "?"
await interaction.response.send_message(f"{question}{question_mark}\nHeidi sagt: {random.choice(choices)}") await interaction.response.send_message(
f"{question}{question_mark}\nHeidi sagt: {random.choice(choices)}"
)
# TODO: Allow , separated varargs, need to parse manually as slash commands don't support varargs # TODO: Allow , separated varargs, need to parse manually as slash commands don't support varargs
@client.tree.command(name="wähle", description="Heidi trifft die Wahl!") @client.tree.command(name="wähle", description="Heidi trifft die Wahl!")
@app_commands.rename(option_a = "entweder") @app_commands.rename(option_a="entweder")
@app_commands.describe(option_a = "Ist es vielleicht dies?") @app_commands.describe(option_a="Ist es vielleicht dies?")
@app_commands.rename(option_b = "oder") @app_commands.rename(option_b="oder")
@app_commands.describe(option_b = "Oder doch eher das?") @app_commands.describe(option_b="Oder doch eher das?")
async def choose(interaction: discord.Interaction, option_a: str, option_b: str): async def choose(interaction: discord.Interaction, option_a: str, option_b: str):
options = [option_a.strip(), option_b.strip()] options = [option_a.strip(), option_b.strip()]
await interaction.response.send_message(f"{options[0]} oder {options[1]}?\nHeidi sagt: {random.choice(options)}") await interaction.response.send_message(
f"{options[0]} oder {options[1]}?\nHeidi sagt: {random.choice(options)}"
)
# async def quote_model_autocomplete(interaction: discord.Interaction, current: str) -> list[Choice[str]]: # async def quote_model_autocomplete(interaction: discord.Interaction, current: str) -> list[Choice[str]]:
# models = client.textgen_models.keys() # models = client.textgen_models.keys()
@ -239,22 +258,33 @@ async def choose(interaction: discord.Interaction, option_a: str, option_b: str)
# joined_quote = " ".join(generated_quote) # joined_quote = " ".join(generated_quote)
# await interaction.response.send_message(f"Heidi sagt: \"{joined_quote}\"") # await interaction.response.send_message(f"Heidi sagt: \"{joined_quote}\"")
SOUNDDIR: str = "/sounds/" if DOCKER else "./voicelines/" SOUNDDIR: str = "/sounds/" if DOCKER else "./heidi-sounds/"
# Example: https://discordpy.readthedocs.io/en/latest/interactions/api.html?highlight=autocomplete#discord.app_commands.autocomplete # Example: https://discordpy.readthedocs.io/en/latest/interactions/api.html?highlight=autocomplete#discord.app_commands.autocomplete
async def board_autocomplete(interaction: discord.Interaction, current: str) -> list[Choice[str]]: async def board_autocomplete(
boards = os.listdir(SOUNDDIR) interaction: discord.Interaction, current: str
return [Choice(name=board, value=board) for board in boards] ) -> list[Choice[str]]:
boards: List[str] = os.listdir(SOUNDDIR)
async def sound_autocomplete(interaction: discord.Interaction, current: str) -> list[Choice[str]]:
board = interaction.namespace.board # TODO: Can't work?
sounds = map(lambda x: x.split(".")[0], os.listdir(SOUNDDIR + board + "/"))
return [Choice(name=sound, value=sound) for sound in sounds]
@client.tree.command(name = "sag", description = "Heidi drückt den Knopf auf dem Soundboard.") return [Choice(name=board, value=board) for board in boards if board.startswith(current)]
@app_commands.describe(sound = "Was soll Heidi sagen?")
@app_commands.autocomplete(board = board_autocomplete)
@app_commands.autocomplete(sound = sound_autocomplete) async def sound_autocomplete(
interaction: discord.Interaction, current: str
) -> list[Choice[str]]:
board: str = interaction.namespace.board
sounds: List[str] = list(map(lambda x: x.split(".")[0], os.listdir(SOUNDDIR + board + "/")))
return [Choice(name=sound, value=sound) for sound in sounds if sound.startswith(current)]
@client.tree.command(
name="sag", description="Heidi drückt den Knopf auf dem Soundboard."
)
@app_commands.describe(sound="Was soll Heidi sagen?")
@app_commands.autocomplete(board=board_autocomplete)
@app_commands.autocomplete(sound=sound_autocomplete)
async def say_voiceline(interaction: discord.Interaction, board: str, sound: str): async def say_voiceline(interaction: discord.Interaction, board: str, sound: str):
# Only Members can access voice channels # Only Members can access voice channels
if not isinstance(interaction.user, discord.Member): if not isinstance(interaction.user, discord.Member):
@ -265,7 +295,11 @@ async def say_voiceline(interaction: discord.Interaction, board: str, sound: str
member: discord.Member = interaction.user member: discord.Member = interaction.user
# Member needs to be in voice channel to hear audio (Heidi needs to know the channel to join) # Member needs to be in voice channel to hear audio (Heidi needs to know the channel to join)
if (not member.voice) or (not member.voice.channel) or (not isinstance(member.voice.channel, discord.VoiceChannel)): if (
(not member.voice)
or (not member.voice.channel)
or (not isinstance(member.voice.channel, discord.VoiceChannel))
):
print("User not in (valid) voice channel!") print("User not in (valid) voice channel!")
await interaction.response.send_message("Heidi sagt: Komm in den Channel!") await interaction.response.send_message("Heidi sagt: Komm in den Channel!")
return return
@ -276,12 +310,16 @@ async def say_voiceline(interaction: discord.Interaction, board: str, sound: str
open(SOUNDDIR + board + "/" + sound + ".mkv") open(SOUNDDIR + board + "/" + sound + ".mkv")
except IOError: except IOError:
print("Error: Invalid soundfile!") print("Error: Invalid soundfile!")
await interaction.response.send_message(f"Heidi sagt: \"{board}/{sound}\" kanninich finden bruder") await interaction.response.send_message(
f'Heidi sagt: "{board}/{sound}" kanninich finden bruder'
)
return return
await interaction.response.send_message(f"Heidi sagt: \"{board}/{sound}\"") await interaction.response.send_message(f'Heidi sagt: "{board}/{sound}"')
audio_source = discord.FFmpegPCMAudio(SOUNDDIR + board + "/" + sound + ".mkv") # only works from docker audio_source = discord.FFmpegPCMAudio(
SOUNDDIR + board + "/" + sound + ".mkv"
) # only works from docker
voice_client = await voice_channel.connect() voice_client = await voice_channel.connect()
voice_client.play(audio_source) voice_client.play(audio_source)
@ -290,12 +328,16 @@ async def say_voiceline(interaction: discord.Interaction, board: str, sound: str
await voice_client.disconnect() await voice_client.disconnect()
# Contextmenu ------------------------------------------------------------------------------------ # Contextmenu ------------------------------------------------------------------------------------
# TODO: More insults # TODO: More insults
# Callable on members # Callable on members
@client.tree.context_menu(name="beleidigen") @client.tree.context_menu(name="beleidigen")
async def insult(interaction: discord.Interaction, member: discord.Member): # with message: discord.Message this can be called on a message async def insult(
interaction: discord.Interaction, member: discord.Member
): # with message: discord.Message this can be called on a message
if not member.dm_channel: if not member.dm_channel:
await member.create_dm() await member.create_dm()
@ -317,11 +359,14 @@ async def insult(interaction: discord.Interaction, member: discord.Member): # wi
"Opfer!", "Opfer!",
"Du miese Raupe!", "Du miese Raupe!",
"Geh Steckdosen befruchten!", "Geh Steckdosen befruchten!",
"Richtiger Gesichtsgünther ey!" "Richtiger Gesichtsgünther ey!",
] ]
await member.dm_channel.send(random.choice(insults)) await member.dm_channel.send(random.choice(insults))
await interaction.response.send_message("Anzeige ist raus!") # with ephemeral = True only the caller can see the answer await interaction.response.send_message(
"Anzeige ist raus!"
) # with ephemeral = True only the caller can see the answer
# ------------------------------------------------------------------------------------------------ # ------------------------------------------------------------------------------------------------