113 lines
4.4 KiB
Python
113 lines
4.4 KiB
Python
import discord
|
|
import os # default module
|
|
from dotenv import load_dotenv
|
|
import json
|
|
import mariadb
|
|
|
|
if os.path.exists(".env"):
|
|
load_dotenv()
|
|
intents = discord.Intents.default()
|
|
intents.members = True
|
|
bot = discord.Bot(intents = intents)
|
|
|
|
configfile = "config.json"
|
|
if os.path.exists(configfile):
|
|
with open('config.json') as user_file:
|
|
config = json.loads(user_file.read())
|
|
else:
|
|
config = {"db_user": os.environ["db_user"], "db_password": os.environ["db_password"], "db_host": os.environ["db_host"], "db_name": os.environ["db_name"]}
|
|
|
|
class DBConnection:
|
|
def __init__(self):
|
|
self.user = config["db_user"]
|
|
self.password = config["db_password"]
|
|
self.host = config["db_host"]
|
|
self.database = config["db_name"]
|
|
|
|
def __enter__(self):
|
|
self.conn = mariadb.connect(user=self.user, password=self.password,
|
|
host=self.host, database=self.database)
|
|
self.cursor = self.conn.cursor()
|
|
return self.cursor
|
|
|
|
def __exit__(self, exc_type, exc_val, exc_tb):
|
|
self.conn.commit()
|
|
self.cursor.close()
|
|
self.conn.close()
|
|
|
|
|
|
@bot.event
|
|
async def on_ready():
|
|
bot.add_view(CounterView())
|
|
print(f"{bot.user} is ready and online!")
|
|
|
|
|
|
def create_embed(guild):
|
|
embed = discord.Embed(
|
|
title="Schimpfwortkasse",
|
|
color=discord.Colour.magenta(),
|
|
)
|
|
|
|
with DBConnection() as cur:
|
|
cur.execute("SELECT user_id, count FROM member")
|
|
members = cur.fetchall()
|
|
cur.execute("SELECT sum(count) as Totalcount from member")
|
|
count_member = cur.fetchone()[0]
|
|
if count_member is None:
|
|
count_member = 0
|
|
cur.execute("SELECT value FROM settings WHERE name='pre-count'")
|
|
precount = cur.fetchone()[0]
|
|
if precount is None:
|
|
precount = 0
|
|
total = precount + count_member
|
|
embed.add_field(name="Aktueller Stand:", value=str(total) + ".00 Euro")
|
|
description = ""
|
|
if precount != 0:
|
|
description = description + "**Vorheriger Stand:** " + str(precount) + " Euro\n"
|
|
for member in members:
|
|
description = description + "**" + guild.get_member(member[0]).display_name + "**: " + str(member[1]) + " Euro \n"
|
|
embed.add_field(name="Bestenliste:", value=description)
|
|
return embed
|
|
|
|
|
|
class CounterView(discord.ui.View):
|
|
def __init__(self):
|
|
super().__init__(timeout=None) # timeout of the view must be set to None
|
|
|
|
@discord.ui.button(label="+1", custom_id="count_plus_one", style=discord.ButtonStyle.primary)
|
|
async def button_callback(self, button, interaction):
|
|
with DBConnection() as cur:
|
|
cur.execute("INSERT INTO member (user_id, count) VALUES (?, 1) ON DUPLICATE KEY UPDATE count=count + 1", (interaction.user.id, ))
|
|
await updatemessage(interaction.guild)
|
|
await interaction.response.send_message("Ab ins Sparschwein damit!", ephemeral=True)
|
|
|
|
async def updatemessage(guild):
|
|
with DBConnection() as cur:
|
|
cur.execute("SELECT value from settings WHERE name='message_id'")
|
|
message_id = cur.fetchone()[0]
|
|
cur.execute("SELECT value from settings WHERE name='channel_id'")
|
|
channelid = cur.fetchone()[0]
|
|
channel = await guild.fetch_channel(int(channelid))
|
|
message = await channel.fetch_message(int(message_id))
|
|
await message.edit(embed=create_embed(guild), view=CounterView())
|
|
|
|
@bot.slash_command(name="set_channel", description="Setze den Channel, in dem der Zähler erstellt wird.")
|
|
async def set_channel(ctx):
|
|
channel = ctx.channel
|
|
message = await channel.send(embed=create_embed(ctx.guild), view=CounterView())
|
|
with DBConnection() as cur:
|
|
cur.execute("UPDATE settings SET VALUE = ? WHERE name = 'channel_id'", (ctx.channel_id,))
|
|
cur.execute("UPDATE settings SET VALUE = ? WHERE name = 'message_id'", (message.id,))
|
|
await ctx.respond(content="Done", ephemeral=True)
|
|
|
|
@bot.slash_command(name="update", description="Aktualisiere die Nachricht")
|
|
async def set_channel(ctx):
|
|
await updatemessage(ctx.guild)
|
|
await ctx.respond(content="Done", ephemeral=True)
|
|
|
|
@bot.slash_command(name="reset_counter", description="Setze den Zähler auf Null.")
|
|
async def reset_counter(ctx):
|
|
await ctx.respond(content="Dies kann nur durch Lord gemacht werden. Er muss die Daten in der Datenbank löschen. Sprech ihn bitte an.", ephemeral=True)
|
|
|
|
bot.run(os.getenv('TOKEN')) # run the bot with the token
|