import discord from discord.ext import commands from discord.ext.commands import Cog from helpers.checks import check_if_staff from helpers.userlogs import setwatch class ModWatch(Cog): def __init__(self, bot): self.bot = bot @commands.guild_only() @commands.check(check_if_staff) @commands.command() async def watch(self, ctx, target: discord.Member, *, note: str = ""): """Puts a user under watch, staff only.""" setwatch(target.id, ctx.author, True, target.name) await ctx.send(f"{ctx.author.mention}: user is now on watch.") @commands.guild_only() @commands.check(check_if_staff) @commands.command() async def watchid(self, ctx, target: int, *, note: str = ""): """Puts a user under watch by userid, staff only.""" setwatch(target, ctx.author, True, target.name) await ctx.send(f"{target.mention}: user is now on watch.") @commands.guild_only() @commands.check(check_if_staff) @commands.command() async def unwatch(self, ctx, target: discord.Member, *, note: str = ""): """Removes a user from watch, staff only.""" setwatch(target.id, ctx.author, False, target.name) await ctx.send(f"{ctx.author.mention}: user is now not on watch.") @commands.guild_only() @commands.check(check_if_staff) @commands.command() async def unwatchid(self, ctx, target: int, *, note: str = ""): """Removes a user from watch by userid, staff only.""" setwatch(target, ctx.author, False, target.name) await ctx.send(f"{target.mention}: user is now not on watch.") async def setup(bot): await bot.add_cog(ModWatch(bot))