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

migrate to discord.py, don't launch without token

Browse Source
This commit is contained in:
Christoph Urlacher
2022-10-28 14:15:47 +02:00
parent 76b2eee34d
commit e6d8618229
1 changed files with 81 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files

134
bot.py
View File

@ -1,51 +1,63 @@
import os # Example: https://github.com/Rapptz/discord.py/blob/master/examples/app_commands/basic.py
import re
import random import os, re, random, logging, asyncio, discord
import asyncio from discord import app_commands
from functools import reduce from functools import reduce
from dotenv import load_dotenv from dotenv import load_dotenv
from models import Models from typing import Optional
import discord # TODO: Reenable and extend scraper
# from models import Models
# used to start the bot locally, for docker the variables have to be set when the container is run # We're fancy today
load_dotenv() from rich.traceback import install
TOKEN = os.getenv("DISCORD_TOKEN") install(show_locals=True)
# TODO: Migrate back to discord.py
# TODO: Rewrite bot with slash commands (and making actual use of discord.py)
# TODO: Send messages only to heidispam channel
# TODO: Print status messages to heidispam
# TODO: Somehow upload voicelines more easily (from discord voice message?)
# TODO: Reenable text/quote generation, allow uploading of training text files, allow switching "personalities"
# TODO: Zalgo generator
# IDs of the servers Heidi is used on
LINUS_GUILD = discord.Object(id=431154792308408340)
TEST_GUILD = discord.Object(id=821511861178204161)
class HeidiClient(discord.Client): class HeidiClient(discord.Client):
def __init__(self): def __init__(self, *, intents: discord.Intents):
super().__init__( super().__init__(status="Nur eine kann GNTM werden!", intents=intents)
status="Nur eine kann GNTM werden!",
)
self.prefix = "Heidi, " self.prefix = "Heidi, "
self.prefix_regex = "^" + self.prefix self.prefix_regex = "^" + self.prefix
self.models = Models() # scraped model list # self.models = Models() # scraped model list
# automatic actions # automatic actions
self.triggers = { self.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
}
# commands # commands
self.matchers = {"Hilfe$": self.show_help, self.matchers = {
"Heidi!$": self.say_name, "Hilfe$": self.show_help,
"Heidi!$": self.say_name,
# GNTM stuff # GNTM stuff
"wer ist dabei\\?$": self.list_models_in, # "wer ist dabei\\?$": self.list_models_in,
"wer ist raus\\?$": self.list_models_out, # "wer ist raus\\?$": self.list_models_out,
"gib Bild von .+$": self.show_model_picture, # "gib Bild von .+$": self.show_model_picture,
"gib Link$": self.show_link, "gib Link$": self.show_link,
# Fun stuff # Fun stuff
"welche Farbe .+\\?": self.random_color, "welche Farbe .+\\?": self.random_color,
".+, ja oder nein\\?": self.magic_shell, ".+, ja oder nein\\?": self.magic_shell,
"wähle: (.+,?)+$": self.choose, "wähle: (.+,?)+$": self.choose,
"sprechen": self.list_voicelines, "sprechen": self.list_voicelines,
"sag .+$": self.say_voiceline} "sag .+$": self.say_voiceline
}
# Helpers ------------------------------------------------------------------------------------ # Helpers ------------------------------------------------------------------------------------
@ -54,10 +66,10 @@ class HeidiClient(discord.Client):
Generate help-string from docstrings of matchers and triggers Generate help-string from docstrings of matchers and triggers
""" """
docstrings_triggers = [ docstrings_triggers = [
" - " + func.__doc__.strip() for func in self.triggers.values() " - " + str(func.__doc__).strip() for func in self.triggers.values()
] ]
docstrings_matchers = [ docstrings_matchers = [
" - " + func.__doc__.strip() for func in self.matchers.values() " - " + str(func.__doc__).strip() for func in self.matchers.values()
] ]
response = 'Präfix: "' + self.prefix + '" (mit Leerzeichen)\n' response = 'Präfix: "' + self.prefix + '" (mit Leerzeichen)\n'
@ -83,6 +95,7 @@ class HeidiClient(discord.Client):
print(f"{self.user} (id: {self.user.id}) has connected to Discord!") print(f"{self.user} (id: {self.user.id}) has connected to Discord!")
async def on_message(self, message): async def on_message(self, message):
# Skip Heidis own messages
if message.author == client.user: if message.author == client.user:
return return
@ -111,27 +124,27 @@ class HeidiClient(discord.Client):
""" """
await message.channel.send("HEIDI!") await message.channel.send("HEIDI!")
async def list_models_in(self, message): # async def list_models_in(self, message):
""" # """
wer ist dabei? # wer ist dabei?
""" # """
await message.channel.send("\n".join(self.models.get_in_names())) # await message.channel.send("\n".join(self.models.get_in_names()))
async def list_models_out(self, message): # async def list_models_out(self, message):
""" # """
wer ist raus? (Liste der Keks welche ge*ickt wurden) # wer ist raus? (Liste der Keks welche ge*ickt wurden)
""" # """
await message.channel.send("\n".join(self.models.get_out_names())) # await message.channel.send("\n".join(self.models.get_out_names()))
async def show_model_picture(self, message): # async def show_model_picture(self, message):
""" # """
gib Bild von <Name> # gib Bild von <Name>
""" # """
name = message.content.split()[-1] # name = message.content.split()[-1]
picture = discord.Embed() # picture = discord.Embed()
picture.set_image(url=self.models.get_image(name)) # picture.set_image(url=self.models.get_image(name))
picture.set_footer(text=name) # picture.set_footer(text=name)
await message.channel.send(embed=picture) # await message.channel.send(embed=picture)
@staticmethod @staticmethod
async def magic_shell(message): async def magic_shell(message):
@ -253,5 +266,20 @@ class HeidiClient(discord.Client):
await message.add_reaction("🧀") await message.add_reaction("🧀")
client = HeidiClient() # Used to start the bot locally, for docker the variables have to be set when the container is run
client.run(TOKEN) load_dotenv()
TOKEN: str = os.getenv("DISCORD_TOKEN") or ""
# Start client if TOKEN valid
if TOKEN != "":
# Log to file
handler = logging.FileHandler(filename='discord.log', encoding='utf-8', mode='w')
# Intents specification is no longer optional
intents = discord.Intents.default()
intents.message_content = True
client = HeidiClient(intents=intents)
client.run(TOKEN, log_handler=handler)
else:
print("DISCORD_TOKEN not found, exiting...")