jmueller/1.Fjg-Rekrubot
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Implemented No Stimme

Browse Source
This commit is contained in:
Jonathan Müller
2023-11-30 21:15:30 +01:00
parent e4c6598885
commit 2c725bbafb
1 changed files with 138 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

149
main.py
View File

@@ -39,9 +39,12 @@ bot = discord.Bot(intents = intents)
@bot.event
async def on_ready():
global select
select = Yes_select()
bot.add_view(select)
global Yes_select_inst
Yes_select_inst = Yes_select()
bot.add_view(Yes_select_inst)
global no_select_inst
no_select_inst = No_select()
bot.add_view(no_select_inst)
print(f"{bot.user} is ready and online!")
conn = connect_db()
@@ -62,10 +65,18 @@ def build_option_list():
records = cur.fetchall()
for row in records:
options.append(discord.SelectOption(value=str(row[0]), label=str(row[1])))
print("Ich wurde benutzt!")
return options
def insert_yes_vote(recruit_id, voter_id):
cur.execute("SELECT discord_id_recruit, discord_id_voter FROM yes_votes WHERE discord_id_recruit = ? AND discord_id_voter = ?", (recruit_id, voter_id))
result = cur.fetchone()
if result is None:
cur.execute("INSERT INTO yes_votes(discord_id_recruit, discord_id_voter) VALUES (?, ?)", (recruit_id, voter_id))
conn.commit()
cur.execute("SELECT nickname FROM recruits WHERE discord_id = ?", (recruit_id, ))
return cur.fetchone()[0]
class Yes_select(discord.ui.View):
def __init__(self):
super().__init__(timeout=None) # timeout of the view must be set to None
@@ -78,8 +89,16 @@ class Yes_select(discord.ui.View):
)
async def select_callback(self, select, interaction): # the function called when the user is done selecting options
await interaction.response.send_message(f"Awesome! I like {select.values[0]} too!")
message = "Du hast für folgende Rekruten mit **Ja** abgestimmt:\n\n"
for recruit in select.values:
member = insert_yes_vote(recruit, interaction.user.id)
message = message + member + "\n"
embed = discord.Embed(
title="Aktion abgeschlossen",
description=message,
color=discord.Colour.green(),
)
await interaction.response.send_message(embed = embed, ephemeral = True)
def create_yes_embed():
@@ -92,7 +111,7 @@ def create_yes_embed():
async def send_yes_message(channel):
message = await channel.send(embed=create_yes_embed(), view=Yes_select())
message = await channel.send(embed=create_yes_embed(), view=Yes_select_inst)
cur.execute("UPDATE settings SET VALUE = ? WHERE name = 'message_voting_yes'", (message.id, ))
@@ -102,17 +121,123 @@ async def edit_yes_message():
cur.execute("SELECT value FROM settings WHERE name='message_voting_yes'")
message = await channel.fetch_message(int(cur.fetchone()[0]))
conn.commit()
global select
#result = await message.edit(embed=create_yes_embed(), view=Yes_select())
select.children[0].options = build_option_list()
await message.edit(embed=create_yes_embed(), view=select)
global Yes_select_inst
Yes_select_inst.children[0].options = build_option_list()
Yes_select_inst.children[0].max_values = get_count_recruits()
await message.edit(embed=create_yes_embed(), view=Yes_select_inst)
print("Edited!")
def insert_no_vote(recruit_id, voter_id, reason):
cur.execute("INSERT INTO no_votes(discord_id_recruit, discord_id_voter, reason) VALUES (?, ?, ?)", (recruit_id, voter_id, reason))
conn.commit()
cur.execute("SELECT nickname FROM recruits WHERE discord_id = ?", (recruit_id, ))
return cur.fetchone()[0]
class No_reason(discord.ui.Modal):
recruit_name = ""
recruit_id = ""
def __init__(self, *args, **kwargs) -> None:
super().__init__(*args, **kwargs)
self.add_item(discord.ui.InputText(label="Begründung", style=discord.InputTextStyle.long))
async def callback(self, interaction: discord.Interaction):
insert_no_vote(self.recruit_id, interaction.user.id, self.children[0].value)
message = ("Du hast deine Nein-Stimme erfolgreich abgegeben.")
embed = discord.Embed(
title="Aktion abgeschlossen",
description=message,
color=discord.Colour.green(),
)
embed.add_field(name="Rekrut:", value=self.recruit_name)
embed.add_field(name="Dein Grund", value=self.children[0].value)
embed.add_field(name="Hinweis:", value="Die Rekrutierungsleiter werden dich demnächst kontaktieren.")
await interaction.response.send_message(embeds=[embed], ephemeral=True)
async def reason(discord_id, interaction):
cur.execute("SELECT nickname FROM recruits WHERE discord_id = ?", (discord_id,))
nickname = cur.fetchone()[0]
no_reason_instance = No_reason(title="Begründung für " + nickname)
no_reason_instance.recruit_id = discord_id
no_reason_instance.recruit_name = nickname
await interaction.response.send_modal(no_reason_instance)
class no_question(discord.ui.View):
discord_id = ""
@discord.ui.button(label="Ja", style=discord.ButtonStyle.green,)
async def yes(self, button, interaction):
await reason(self.discord_id, interaction)
@discord.ui.button(label="Nein", style=discord.ButtonStyle.grey)
async def no(self, button, interaction):
embed = discord.Embed(
title="Aktion abgebrochen",
description="Es wurde nichts gespeichert.\nDeine Nein-Stimme mit dem angegebenen Grund bleibt bestehen.\nWenn du diese löschen möchtest, wähle die Option 'Stimme entfernen'",
color=discord.Colour.blurple(), # Pycord provides a class with default colors you can choose from
)
await interaction.response.send_message(embed=embed, ephemeral=True)
class No_select(discord.ui.View):
def __init__(self):
super().__init__(timeout=None) # timeout of the view must be set to None
@discord.ui.select(
custom_id="no1",
placeholder="Rekruten wählen...",
min_values=1,
max_values=1,
options=build_option_list()
)
async def select_callback(self, select, interaction): # the function called when the user is done selecting options
cur.execute("SELECT no_votes.reason, recruits.nickname FROM recruits, no_votes WHERE recruits.discord_id = no_votes.discord_id_recruit AND no_votes.discord_id_voter = ? AND no_votes.discord_id_recruit = ?", (interaction.user.id, select.values[0]))
vote = cur.fetchall()
if not vote:
await reason(select.values[0], interaction)
else:
no_question_inst = no_question()
no_question_inst.discord_id = select.values[0]
embed = discord.Embed(
title="Du hast bereits für den Rekruten abgestimmt.",
description="Möchtest du deine jetzige Stimme löschen und einen neuen Grund angeben?",
color=discord.Colour.blurple(), # Pycord provides a class with default colors you can choose from
)
embed.add_field(name="Rekrut", value=vote[0][1])
embed.add_field(name="Angegebener Grund", value=vote[0][0])
await interaction.response.send_message(embed=embed, view=no_question_inst, ephemeral = True)
def create_no_embed():
embed = discord.Embed(
title="Rekrutenbesichtigung: Nein-Stimme",
description="Geb hier deine Nein-Stimme für einen Rekruten ab!",
color=discord.Colour.red(),
)
return embed
async def send_no_message(channel):
message = await channel.send(embed=create_no_embed(), view=no_select_inst)
cur.execute("UPDATE settings SET VALUE = ? WHERE name = 'message_voting_no'", (message.id, ))
async def edit_no_message():
cur.execute("SELECT value FROM settings WHERE name='channel_voting'")
channel = bot.get_channel(int(cur.fetchone()[0]))
cur.execute("SELECT value FROM settings WHERE name='message_voting_no'")
message = await channel.fetch_message(int(cur.fetchone()[0]))
conn.commit()
global no_select_inst
no_select_inst.children[0].options = build_option_list()
await message.edit(embed=create_no_embed(), view=no_select_inst)
print("Edited!")
@settings.command(description="Sendet die Nachricht zum Abstimmen in den aktuellen Channel.")
@default_permissions(manage_roles=True)
async def message_voting(ctx):
cur.execute("UPDATE settings SET VALUE = ? WHERE name = 'channel_voting'", (ctx.channel_id, ))
await send_yes_message(ctx.channel)
await send_no_message(ctx.channel)
conn.commit()
await ctx.respond(f"Done.", ephemeral=True)
@@ -167,6 +292,7 @@ async def refresh(ctx):
cur.execute("UPDATE recruits SET recruit = ?, nickname = ? WHERE discord_id = ?", (0, member.display_name, member.id))
conn.commit()
await edit_yes_message()
await edit_no_message()
await ctx.respond(f"Done.", ephemeral=True)
@@ -190,6 +316,7 @@ async def on_member_update(previous, after):
recruit_update(after, 0, cur, conn)
break
await edit_yes_message()
await edit_no_message()
bot.run(os.getenv('TOKEN')) # run the bot with the token