rework the whole bot for the new command system

This commit is contained in:
Aaron Riedel 2022-01-22 21:09:45 +01:00
parent 9407111771
commit afc80258ea

279
bot.py
View file

@ -18,32 +18,39 @@ def mid(s, offset, amount):
return s[offset:offset+amount] return s[offset:offset+amount]
token = os.environ['TOKEN'] token = os.environ['TOKEN']
bot = commands.Bot(command_prefix=os.environ['PREFIX'])
client = discord.Client() client = discord.Client()
async def admin(ctx):
def admin(ctx):
if ctx.guild.get_role(261603488331595776) in ctx.author.roles: if ctx.guild.get_role(261603488331595776) in ctx.author.roles:
return True return True
async def member(ctx): def member(ctx):
if ctx.guild.get_role(261603747711418371) in ctx.author.roles: if ctx.guild.get_role(261603747711418371) in ctx.author.roles:
return True return True
def is_admin(): def is_admin():
async def predicate(ctx): async def predicate(ctx):
await admin(ctx) if ctx.guild.get_role(261603488331595776) in ctx.author.roles:
return True
return commands.check(predicate) return commands.check(predicate)
def is_member(): def is_member():
async def predicate(ctx): async def predicate(ctx):
await member(ctx) if ctx.guild.get_role(261603747711418371) in ctx.author.roles:
return True
return commands.check(predicate)
def is_gm():
async def predicate(ctx):
if ctx.guild.get_role(511893805956595722) in ctx.author.roles:
return True
return commands.check(predicate) return commands.check(predicate)
emojinumbers = ["0\u20E3", "1\u20E3" , "2\u20E3" , "3\u20E3" , "4\u20E3" , "5\u20E3" , "6\u20E3" , "7\u20E3" , "8\u20E3" , "9\u20E3"] emojinumbers = ["0\u20E3", "1\u20E3" , "2\u20E3" , "3\u20E3" , "4\u20E3" , "5\u20E3" , "6\u20E3" , "7\u20E3" , "8\u20E3" , "9\u20E3"]
@client.event @bot.event
async def on_ready(): async def on_ready():
print("Bot ready on Version %s..." % discord.__version__) print("Bot ready on Version %s..." % discord.__version__)
bot = commands.Bot(command_prefix='!')
@bot.command() @bot.command()
@is_admin()
async def test(ctx, arg): async def test(ctx, arg):
await ctx.send(arg) await ctx.send(arg)
@ -65,36 +72,11 @@ async def helpme(ctx):
await ctx.author.create_dm() await ctx.author.create_dm()
await ctx.author.dm_channel.send(embed=em) await ctx.author.dm_channel.send(embed=em)
@client.event @bot.command()
async def on_message(message): @is_member()
if message.content == "!help": async def survey(ctx, *, arg):
if message.channel == message.author.dm_channel: await ctx.message.delete()
await message.author.dm_channel.send("Sorry das geht so nicht. Bitte schreibe diese Nachricht in einen Channel auf dem SecondHemd Discord, sonst kann ich nicht sehen was du für Berechtigungen hast. Vielen Dank.") answers = arg.split("|")
else:
await message.delete()
if member(message):
if permission(message):
admin_text = "\n\n!add = Hinzufügen einer Rolle mit Textchannel und hinzufügen von Usern zur Rolle\nNutzung: !add \"<Rolle>\" <User Mentions>\n\n!labor = bringt dich ins Labor :)\nNutzung: !labor\n\n!start = Starte ne Runde PnP\nNutzung: !start @Rolle\n\n!stop = Beende die gestartete Runde PnP\nNutzung: !stop\n\n!purge = Löschen von Nachrichten\nNutzung: !purge all|<Anzahl>|x minutes/hours/days/weeks\n\n!prune - kickt Member ohne Rolle, die 30 Tage nicht online waren\nNutzung: !prune"
else:
admin_text = ""
em = discord.Embed(title="Hilfe",description="!survey = Umfrage mit mehreren Antwortmöglichkeiten\nNutzung: !survey <Frage> | <Antwort1> | <Antwort2> ...\n\n!vote = Ja/Nein Umfrage\nNutzung: !vote <Frage>\n\n!love = zeige einem User Liebe\nNutzung: !love <@User1> <@User2> ...\n\n!roll = Rolle einen oder mehrere Würfel\nNutzung: !roll <anzahl_optional>W<seitenzahl> (z.B. !roll W20 oder !roll 10W6)%s" % admin_text, colour=0x00FF00)
if message.author.dm_channel == None:
await message.author.create_dm()
await message.author.dm_channel.send(embed=em)
else:
if message.author.dm_channel == None:
await message.author.create_dm()
await message.author.dm_channel.send("Sorry, du hast leider nicht die nötigen Berechtigungen. :(")
if message.content.startswith("!survey ") and member(message):
if message.content == "!survey":
error_message = await message.channel.send("Nutzung: !survey <Frage> | <Antwort1> | <Antwort2> ...")
await message.delete()
await asyncio.sleep(15)
await error_message.delete()
else:
answers_raw = right(message.content, len(message.content)-8)
answers = answers_raw.split("|")
question = answers[0] question = answers[0]
answers.remove(question) answers.remove(question)
desc = "" desc = ""
@ -103,23 +85,20 @@ async def on_message(message):
desc = desc + emojinumbers[z] + " - " + y + "\n" desc = desc + emojinumbers[z] + " - " + y + "\n"
z = z + 1 z = z + 1
em = discord.Embed(title=question, description=desc, colour=0x00E0FF) em = discord.Embed(title=question, description=desc, colour=0x00E0FF)
em.set_author(name=message.author.display_name, url=discord.Embed.Empty, icon_url=message.author.avatar_url) em.set_author(name=ctx.author.display_name, url=discord.Embed.Empty, icon_url=ctx.author.avatar_url)
ask_msg = await message.channel.send(content="||@here||",embed=em) ask_msg = await ctx.send(content="||@here||",embed=em)
a = 0 a = 0
for x in emojinumbers: for x in emojinumbers:
if a < z and a != 0: if a < z and a != 0:
await ask_msg.add_reaction(x) await ask_msg.add_reaction(x)
a = a + 1 a = a + 1
await message.delete()
if message.content.startswith("!surveyedit") and member(message): @bot.command()
if message.content == "!surveyedit": @is_member()
await message.delete() async def surveyedit(ctx, *, arg):
await message.channel.send(content="Nutzung: !surveyedit ID | <Frage> | <Antwort1> | <Antwort2> ...", delete_after=15.0) #Nutzung: !surveyedit ID | <Frage> | <Antwort1> | <Antwort2> ...
else: await ctx.message.delete()
await message.delete() answers = arg.split("|")
answers_raw = right(message.content, len(message.content)-12)
answers = answers_raw.split("|")
survey_id = answers[0] survey_id = answers[0]
question = answers[1] question = answers[1]
answers.remove(survey_id) answers.remove(survey_id)
@ -130,202 +109,191 @@ async def on_message(message):
desc = desc + emojinumbers[z] + " - " + y + "\n" desc = desc + emojinumbers[z] + " - " + y + "\n"
z = z + 1 z = z + 1
em = discord.Embed(title=question, description=desc, colour=0x00E0FF) em = discord.Embed(title=question, description=desc, colour=0x00E0FF)
em.set_author(name=message.author.display_name, url=discord.Embed.Empty, icon_url=message.author.avatar_url) em.set_author(name=ctx.author.display_name, url=discord.Embed.Empty, icon_url=ctx.author.avatar_url)
survey_msg = await message.channel.fetch_message((int(survey_id))) survey_msg = await ctx.channel.fetch_message((int(survey_id)))
await survey_msg.edit(embed=em) await survey_msg.edit(embed=em)
#----------------------------------------------------------------------------------------
if message.content.startswith("!vote") and member(message): @bot.command()
if message.content == "!vote": @is_member()
error_message = await message.channel.send("Nutzung: !vote <Frage>") async def vote(ctx, *, arg):
await message.delete() await ctx.message.delete()
await asyncio.sleep(10) em = discord.Embed(description=arg, colour=0x00E0FF)
await error_message.delete() em.set_author(name=ctx.author.display_name, url=discord.Embed.Empty, icon_url=ctx.author.avatar_url)
else: ask_msg = await ctx.send(content="||@here||",embed=em)
em = discord.Embed(description=str(right(message.content,len(message.content)-6)), colour=0x00E0FF)
em.set_author(name=message.author.display_name, url=discord.Embed.Empty, icon_url=message.author.avatar_url)
ask_msg = await message.channel.send(content="||@here||",embed=em)
for x in ["", "", ""]: for x in ["", "", ""]:
await ask_msg.add_reaction(x) await ask_msg.add_reaction(x)
await message.delete()
if message.content.startswith("!say") and permission(message): @bot.command()
em = discord.Embed(description=str(right(message.content,len(message.content)-5)), colour=0x00E0FF) @is_admin()
async def say(ctx, *, arg):
await ctx.message.delete()
em = discord.Embed(description=arg, colour=0x00E0FF)
await message.channel.send(embed=em) await message.channel.send(embed=em)
await message.delete() await message.delete()
#move Bot @bot.command()
if message.content.startswith("!start") and permission(message): @is_gm()
if message.content == ("!start"): async def start(ctx, role_id):
error_message = await message.channel.send( "Nutzung: !start @Rolle") await ctx.message.delete()
await asyncio.sleep(6)
await message.delete()
await error_message.delete()
else:
role_id = message.content.split(" ")[1]
role_id = right(left(role_id, len(role_id)-1), len(role_id) - 4) role_id = right(left(role_id, len(role_id)-1), len(role_id) - 4)
await message.delete() for r in ctx.author.roles:
for r in message.author.roles:
if str(r.id) == role_id: if str(r.id) == role_id:
role = r role = r
em = discord.Embed(description='gespieltes PnP: %s\nTeilnehmer (zum Start):' % role.mention, colour=0x00770d) em = discord.Embed(description='gespieltes PnP: %s\nTeilnehmer (zum Start):' % role.mention, colour=0x00770d)
move_member = [] move_member = []
for c in message.guild.channels: for c in ctx.guild.channels:
if str(c.type) == "voice": if str(c.type) == "voice":
for m in c.members: for m in c.members:
if role in m.roles: if role in m.roles:
move_member.append(m) move_member.append(m)
for m in move_member: for m in move_member:
await m.move_to(client.get_channel(435869507123281920)) await m.move_to(bot.get_channel(435869507123281920))
em.add_field(name=":white_check_mark: " + m.display_name, value = m.mention + "ist am Start! Juhu!", inline=False) em.add_field(name=":white_check_mark: " + m.display_name, value = m.mention + "ist am Start! Juhu!", inline=False)
start_msg = await client.get_channel(435501602485305345).send(content=":game_die: **INFO: PnP Sitzung gestartet!**",embed=em) start_msg = await bot.get_channel(435501602485305345).send(content=":game_die: **INFO: PnP Sitzung gestartet!**",embed=em)
text_file = open("start_id.txt", "w") text_file = open("start_id.txt", "w")
text_file.write(str(start_msg.id)) text_file.write(str(start_msg.id))
text_file.close() text_file.close()
if message.content == "!stop" and permission(message): @bot.command()
@is_gm()
async def stop(ctx):
await ctx.message.delete()
text_file = open("start_id.txt", "r") text_file = open("start_id.txt", "r")
start_id = text_file.read() start_id = text_file.read()
text_file.close() text_file.close()
start_msg = await message.channel.fetch_message(start_id) start_msg = await bot.get_channel(435501602485305345).fetch_message(start_id)
now_time = datetime.datetime.utcnow().replace(microsecond=0) now_time = datetime.datetime.utcnow().replace(microsecond=0)
then_time = start_msg.created_at.replace(microsecond=0) then_time = start_msg.created_at.replace(microsecond=0)
dauer = str(now_time - then_time) dauer = str(now_time - then_time)
for m in client.get_channel(435869507123281920).members: for m in client.get_channel(435869507123281920).members:
await m.move_to(client.get_channel(801869864745697280)) await m.move_to(client.get_channel(801869864745697280))
await message.delete()
await start_msg.edit(content=":game_die: **INFO: PnP Sitzung beendet!** Dauer: %s" % dauer) await start_msg.edit(content=":game_die: **INFO: PnP Sitzung beendet!** Dauer: %s" % dauer)
if message.content == "!labor" and permission(message): @bot.command()
await message.author.move_to(client.get_channel(765601577334865972)) @is_admin()
await message.delete() async def labor(ctx):
await ctx.message.delete()
await ctx.author.move_to(bot.get_channel(765601577334865972))
if message.content.startswith("!add") and permission(message): @bot.command()
if message.content == "!add": @is_admin()
explain_msg = await message.channel.send("Nutzung: !add \"<Rolle>\" <User Mentions>") async def add(ctx, *, arg):
await message.delete() await ctx.message.delete()
await asyncio.sleep(10)
await explain_msg.delete()
else:
try: try:
role_name = message.content.split("\"")[1] role_name = ctx.message.content.split("\"")[1]
text_name = role_name.lower().replace(" ", "-") text_name = role_name.lower().replace(" ", "-")
role = await message.guild.create_role(name=role_name, mentionable=True) role = await ctx.guild.create_role(name=role_name, mentionable=True)
overwrites = { overwrites = {
message.guild.default_role: discord.PermissionOverwrite(read_messages=False), ctx.guild.default_role: discord.PermissionOverwrite(read_messages=False),
role: discord.PermissionOverwrite(read_messages=True) role: discord.PermissionOverwrite(read_messages=True)
} }
channel = await message.guild.create_text_channel(text_name, overwrites=overwrites, topic="Dieser Channel wurde von SecondBot erstellt.") channel = await ctx.guild.create_text_channel(text_name, overwrites=overwrites, topic="Dieser Channel wurde von SecondBot erstellt.")
for user in message.mentions: for user in ctx.message.mentions:
await user.add_roles(role) await user.add_roles(role)
await message.delete()
em = discord.Embed(title=':loudspeaker: Ein neuer Channel für %s!' % role_name, description='Herzlich Willkommen! Es gibt nun eine neue Gruppe %s und diesen wunderbaren Channel hier! Der Channel ist nur für euch und die Admins sichtbar.' % role.mention, colour=0xffa500) em = discord.Embed(title=':loudspeaker: Ein neuer Channel für %s!' % role_name, description='Herzlich Willkommen! Es gibt nun eine neue Gruppe %s und diesen wunderbaren Channel hier! Der Channel ist nur für euch und die Admins sichtbar.' % role.mention, colour=0xffa500)
await channel.send(embed=em) await channel.send(embed=em)
except Exception as e: except Exception as e:
em = discord.Embed(title="Error Code", description=e, color=0xff0000) em = discord.Embed(title="Error Code", description=e, color=0xff0000)
error_message = await message.channel.send(content="Das hat nicht funktioniert. Syntax: !add \"<Rolle>\" <User Mentions>", embed=em) await ctx.channel.send(content="Das hat nicht funktioniert. Syntax: !add \"<Rolle>\" <User Mentions>", embed=em, delete_after=20.0)
await message.delete()
await asyncio.sleep(15)
await error_message.delete()
if message.content.startswith("!purge") and permission(message): @bot.command()
await message.delete() @is_admin()
if message.content == "!purge": async def purge(ctx, *, arg):
await message.channel.send(content="Nutzung: !purge all|<Anzahl>|x minutes/hours/days/weeks", delete_after=10.0) #Nutzung: !purge all|<Anzahl>|x minutes/hours/days/weeks
else: await ctx.message.delete()
by_limit = False by_limit = False
by_time = False by_time = False
try: try:
if message.content.split(" ")[2] == "minutes": if ctx.message.content.split(" ")[2] == "minutes":
by_time = True by_time = True
delta = datetime.timedelta(minutes=int(message.content.split(" ")[1])) delta = datetime.timedelta(minutes=int(ctx.message.content.split(" ")[1]))
if message.content.split(" ")[2] == "hours": if ctx.message.content.split(" ")[2] == "hours":
by_time = True by_time = True
delta = datetime.timedelta(hours=int(message.content.split(" ")[1])) delta = datetime.timedelta(hours=int(ctx.message.content.split(" ")[1]))
if message.content.split(" ")[2] == "days": if ctx.message.content.split(" ")[2] == "days":
by_time = True by_time = True
delta = datetime.timedelta(days=int(message.content.split(" ")[1])) delta = datetime.timedelta(days=int(ctx.message.content.split(" ")[1]))
if message.content.split(" ")[2] == "weeks": if ctx.message.content.split(" ")[2] == "weeks":
by_time = True by_time = True
delta = datetime.timedelta(weeks=int(message.content.split(" ")[1])) delta = datetime.timedelta(weeks=int(ctx.message.content.split(" ")[1]))
except IndexError: except IndexError:
try: try:
if message.content.split(" ")[1] == "all": if ctx.message.content.split(" ")[1] == "all":
by_limit = True by_limit = True
limit = None limit = None
else: else:
by_limit = True by_limit = True
limit = int(message.content.split(" ")[1]) limit = int(ctx.message.content.split(" ")[1])
except: except:
print("List error, ploz ignore") print("List error, ploz ignore")
except: except:
print("List error, ploz ignore") print("List error, ploz ignore")
if by_time: if by_time:
after_time = datetime.datetime.utcnow() - delta after_time = datetime.datetime.utcnow() - delta
question = await message.channel.send("Would you really like to delete all messages of the last {} {}?".format(message.content.split(" ")[1], message.content.split(" ")[2])) question = await ctx.send("Would you really like to delete all messages of the last {} {}?".format(ctx.message.content.split(" ")[1], ctx.message.content.split(" ")[2]))
def check(reaction, user): def check(reaction, user):
return user == message.author and str(reaction.emoji) == "" return user == ctx.author and str(reaction.emoji) == ""
await question.add_reaction("") await question.add_reaction("")
try: try:
reaction, user = await client.wait_for("reaction_add", timeout=60.0, check=check) reaction, user = await bot.wait_for("reaction_add", timeout=60.0, check=check)
except asyncio.TimeoutError: except asyncio.TimeoutError:
await question.delete() await question.delete()
else: else:
await question.delete() await question.delete()
deleted = await message.channel.purge(limit=200, after=after_time) deleted = await ctx.channel.purge(limit=200, after=after_time)
await message.channel.send(content='Ich habe {} Nachrichten gelöscht.'.format(len(deleted)), delete_after=5.0) await ctx.send(content='Ich habe {} Nachrichten gelöscht.'.format(len(deleted)), delete_after=5.0)
if by_limit: if by_limit:
if limit == None: if limit == None:
count = "alle" count = "alle"
else: else:
count = str(limit) count = str(limit)
question = await message.channel.send("Sollen wirklich {} Nachrichten gelöscht werden?".format(count)) question = await ctx.send("Sollen wirklich {} Nachrichten gelöscht werden?".format(count))
def check(reaction, user): def check(reaction, user):
return user == message.author and str(reaction.emoji) == "" return user == ctx.author and str(reaction.emoji) == ""
await question.add_reaction("") await question.add_reaction("")
try: try:
reaction, user = await client.wait_for("reaction_add", timeout=60.0, check=check) reaction, user = await bot.wait_for("reaction_add", timeout=60.0, check=check)
except asyncio.TimeoutError: except asyncio.TimeoutError:
await question.delete() await question.delete()
else: else:
await question.delete() await question.delete()
deleted = await message.channel.purge(limit=limit) deleted = await ctx.channel.purge(limit=limit)
await message.channel.send(content='Ich habe {} Nachrichten gelöscht.'.format(len(deleted)), delete_after=5.0) await ctx.send(content='Ich habe {} Nachrichten gelöscht.'.format(len(deleted)), delete_after=5.0)
if message.content.startswith("!love") and member(message): @bot.command()
await message.delete() @is_member()
for user in message.mentions: async def love(ctx, *, arg):
await ctx.message.delete()
for user in ctx.message.mentions:
if user.dm_channel == None: if user.dm_channel == None:
await user.create_dm() await user.create_dm()
await user.dm_channel.send("❤️") await user.dm_channel.send("❤️")
if message.content.endswith("!?") and member(message): @bot.command()
for x in ["", ""]: @is_admin()
await message.add_reaction(x) async def prune(ctx):
await ctx.message.delete()
if message.content.startswith("!prune") and permission(message): count = await ctx.guild.estimate_pruned_members(days=30)
await message.delete() question = await ctx.channel.send("Sollen wirklich {} Leichen gekickt werden?".format(count))
count = await message.guild.estimate_pruned_members(days=30)
question = await message.channel.send("Sollen wirklich {} Leichen gekickt werden?".format(count))
def check(reaction, user): def check(reaction, user):
return user == message.author and str(reaction.emoji) == "" return user == ctx.author and str(reaction.emoji) == ""
await question.add_reaction("") await question.add_reaction("")
try: try:
reaction, user = await client.wait_for("reaction_add", timeout=60.0, check=check) reaction, user = await bot.wait_for("reaction_add", timeout=60.0, check=check)
except asyncio.TimeoutError: except asyncio.TimeoutError:
await question.delete() await question.delete()
else: else:
await question.delete() await question.delete()
deleted = await message.guild.prune_members(days=30) deleted = await ctx.guild.prune_members(days=30)
await message.channel.send(content='Ich habe {} Leichen beseitigt.'.format(deleted), delete_after=5.0) await ctx.send(content='Ich habe {} Leichen beseitigt.'.format(deleted), delete_after=5.0)
if message.content.startswith("!roll") and member(message): @bot.command()
if message.content == "!roll": @is_member()
await message.delete() async def roll(ctx, arg):
await message.channel.send(content="Nutzung: !roll <anzahl_optional>W<seitenzahl> (z.B. !roll W20 oder !roll 10W6)", delete_after=10.0) await ctx.message.delete()
else:
rollt = 0 rollt = 0
rolle = " " rolle = " "
roll = message.content.split(" ")[1].lower() roll = arg.lower()
if "d" in roll: if "d" in roll:
rolls = "d" rolls = "d"
if "w" in roll: if "w" in roll:
@ -348,17 +316,6 @@ async def on_message(message):
rolltotal = rolle + "\n" + rolltotal rolltotal = rolle + "\n" + rolltotal
rolle = "" rolle = ""
em = discord.Embed(title=rolle, description=rolltotal, colour=0x009933) em = discord.Embed(title=rolle, description=rolltotal, colour=0x009933)
await message.channel.send(embed=em) await ctx.send(embed=em)
#Shiba Bilder vom Shiba Discord auf den SH Discord hauen bot.run(token)
if message.channel.id == 549177598916296723:
if not message.is_system() and len(message.attachments) > 0:
for a in message.attachments:
await client.get_channel(775390209793064980).send(content=a.url)
if message.content.startswith("!print") and permission(message):
print(message.content)
await message.delete()
client.run(token)