From 9fba7b3075bb3d71529c50b39bd6e2282b165fb2 Mon Sep 17 00:00:00 2001 From: Ave Ozkal Date: Mon, 24 Dec 2018 01:11:01 +0300 Subject: [PATCH] warn: add --- .gitignore | 2 ++ README.md | 4 ++- cogs/links.py | 9 ++++--- cogs/mod.py | 74 +++++++++++++++++++++++++++++++++++++++++++++++++-- 4 files changed, 83 insertions(+), 6 deletions(-) diff --git a/.gitignore b/.gitignore index 349aaf6..a4bcda6 100755 --- a/.gitignore +++ b/.gitignore @@ -100,3 +100,5 @@ files/* priv-* config.py +# Prevent new versions of this from being committed +data/warnsv2.json diff --git a/README.md b/README.md index 9cba824..022f8ae 100755 --- a/README.md +++ b/README.md @@ -35,7 +35,7 @@ Based on https://gitlab.com/ao/dpybotbase - [x] Moderation: nickname - [x] Moderation: clear/purge - [ ] Moderation: restrictions (people who leave with muted role will get muted role on join) -- [ ] Warns: warn +- [x] Warns: warn - [ ] Warns: delwarnid-delwarn - [ ] Warns: listwarns-listwarnsid - [ ] Warns: clearwarns-clearwarnsid @@ -47,8 +47,10 @@ Main goal of this project is to get Robocop functionality done, secondary goal i - [ ] New verification feature: Using log module from akbbot for logging attempts and removing old attempts - [ ] New moderation feature: mutetime (mute with time) +- [ ] New feature: Highlights (problematic words automatically get highlighted) - [ ] New feature: Modmail - [ ] New feature: Submiterr +- [ ] New moderation feature: Display of mutes on listwarns - [ ] New moderation feature: User notes - [ ] New moderation feature: watch-unwatch (using log module from akbbot) - [ ] New moderation feature: timelock (channel lockdown with time) diff --git a/cogs/links.py b/cogs/links.py index 93fbb44..50701a0 100644 --- a/cogs/links.py +++ b/cogs/links.py @@ -15,14 +15,17 @@ class Links: await ctx.send("https://github.com/reswitched/pegaswitch") @commands.command(hidden=True, aliases=["atmos"]) - async def Atmosphere(self, ctx): + async def atmosphere(self, ctx): """Link to the Atmosphere repo""" await ctx.send("https://github.com/atmosphere-nx/atmosphere") - + @commands.command(hidden=True, aliases=["guides"]) async def guide(self, ctx): """Link to the guide(s)""" - await ctx.send("*AtlasNX's SDSetup-Guide:* https://guide.teamatlasnx.com \n *Noirscape's Guide:* http://switchguide.xyz/ \n *Pegaswitch Guide:* https://switch.hacks.guide/") + await ctx.send("*AtlasNX's Guide:* https://guide.teamatlasnx.com\n" + "*Noirscape's Guide:* http://switchguide.xyz/\n" + "*Pegaswitch Guide:* https://switch.hacks.guide/") + def setup(bot): bot.add_cog(Links(bot)) diff --git a/cogs/mod.py b/cogs/mod.py index 57c181c..f17ee09 100644 --- a/cogs/mod.py +++ b/cogs/mod.py @@ -1,6 +1,8 @@ import discord from discord.ext import commands import config +import json +import time class ModCog: @@ -138,7 +140,7 @@ class ModCog: try: await target.send(dm_message) except discord.errors.Forbidden: - # Prevents kick issues in cases where user blocked bot + # Prevents ban issues in cases where user blocked bot # or has DMs disabled pass @@ -301,10 +303,78 @@ class ModCog: if not channel: channel = ctx.channel await channel.purge(limit=limit) - msg = f"🗑 **Cleared**: {ctx.author.mention} cleared {limit} "\ + msg = f"🗑 **Purged**: {ctx.author.mention} purged {limit} "\ f"messages in {channel.mention}." await log_channel.send(msg) + @commands.guild_only() + @commands.check(check_if_staff) + @commands.command() + async def warn(self, ctx, target: discord.Member, *, reason: str = ""): + """Warn a user. Staff only.""" + if self.check_if_target_is_staff(target): + return await ctx.send("I can't warn this user as " + "they're a member of staff.") + + log_channel = self.bot.get_channel(config.log_channel) + with open("data/warnsv2.json", "r") as f: + warns = json.load(f) + if str(target.id) not in warns: + warns[str(target.id)] = {"warns": []} + warns[str(target.id)]["name"] = str(target) + timestamp = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()) + warn_data = {"issuer_id": ctx.author.id, + "issuer_name": ctx.author.name, + "reason": reason, + "timestamp": timestamp} + warns[str(target.id)]["warns"].append(warn_data) + with open("data/warnsv2.json", "w") as f: + json.dump(warns, f) + + warn_count = len(warns[str(target.id)]["warns"]) + + msg = f"You were warned on {ctx.guild.name}." + if reason: + msg += " The given reason is: " + reason + msg += f"\n\nPlease read the rules in {config.rules_url}. "\ + f"This is warn #{warn_count}." + if warn_count == 2: + msg += " __The next warn will automatically kick.__" + if warn_count == 3: + msg += "\n\nYou were kicked because of this warning. "\ + "You can join again right away. "\ + "Two more warnings will result in an automatic ban." + if warn_count == 4: + msg += "\n\nYou were kicked because of this warning. "\ + "This is your final warning. "\ + "You can join again, but "\ + "**one more warn will result in a ban**." + if warn_count == 5: + msg += "\n\nYou were automatically banned due to five warnings." + try: + await target.send(msg) + except discord.errors.Forbidden: + # Prevents log issues in cases where user blocked bot + # or has DMs disabled + pass + if warn_count == 3 or warn_count == 4: + await target.kick() + if warn_count >= 5: # just in case + await target.ban(reason="exceeded warn limit", + delete_message_days=0) + await ctx.send(f"{target.mention} warned. " + f"User has {warn_count} warning(s).") + msg = f"⚠️ **Warned**: {ctx.author.mention} warned {target.mention}"\ + f" (warn #{warn_count}) | {self.bot.escape_message(target)}" + + if reason: + msg += f"✏️ __Reason__: \"{reason}\"" + else: + msg += "Please add an explanation below. In the future"\ + ", it is recommended to use `.ban [reason]`"\ + " as the reason is automatically sent to the user." + await log_channel.send(msg) + def setup(bot): bot.add_cog(ModCog(bot))