mirror of
https://github.com/myned/modufur.git
synced 2024-12-24 14:27:27 +00:00
Prefix per guild command
This commit is contained in:
parent
a74c08867a
commit
1ecd790f83
2 changed files with 181 additions and 163 deletions
|
@ -13,168 +13,180 @@ from utils import utils as u
|
||||||
|
|
||||||
class Administration:
|
class Administration:
|
||||||
|
|
||||||
def __init__(self, bot):
|
def __init__(self, bot):
|
||||||
self.bot = bot
|
self.bot = bot
|
||||||
self.RATE_LIMIT = u.RATE_LIMIT
|
self.RATE_LIMIT = u.RATE_LIMIT
|
||||||
self.queue = asyncio.Queue()
|
self.queue = asyncio.Queue()
|
||||||
self.deleting = False
|
|
||||||
|
|
||||||
if u.tasks['auto_del']:
|
|
||||||
for channel in u.tasks['auto_del']:
|
|
||||||
temp = self.bot.get_channel(channel)
|
|
||||||
self.bot.loop.create_task(self.queue_for_deletion(temp))
|
|
||||||
print('AUTO-DELETING : #{}'.format(temp.id))
|
|
||||||
self.bot.loop.create_task(self.delete())
|
|
||||||
self.deleting = True
|
|
||||||
|
|
||||||
@commands.command(name=',prunefromguild', aliases=[',pfg', ',prunefromserver', ',pfs'], brief='Prune a user\'s messages from the guild', description='about flag centers on message 50 of 101 messages\n\npfg \{user id\} [before|after|about] [\{message id\}]\n\nExample:\npfg \{user id\} before \{message id\}')
|
|
||||||
@commands.is_owner()
|
|
||||||
@checks.del_ctx()
|
|
||||||
async def prune_all_user(self, ctx, user, when=None, reference=None):
|
|
||||||
def yes(msg):
|
|
||||||
if msg.content.lower() == 'y' and msg.channel is ctx.channel and msg.author is ctx.author:
|
|
||||||
return True
|
|
||||||
elif msg.content.lower() == 'n' and msg.channel is ctx.channel and msg.author is ctx.author:
|
|
||||||
raise exc.CheckFail
|
|
||||||
else:
|
|
||||||
return False
|
|
||||||
|
|
||||||
channels = ctx.guild.text_channels
|
|
||||||
if reference is not None:
|
|
||||||
for channel in channels:
|
|
||||||
try:
|
|
||||||
ref = await channel.get_message(reference)
|
|
||||||
|
|
||||||
except err.NotFound:
|
|
||||||
continue
|
|
||||||
|
|
||||||
history = []
|
|
||||||
try:
|
|
||||||
pru_sent = await ctx.send('\N{HOURGLASS} **Pruning** <@{}>**\'s messages will take some time.**'.format(user))
|
|
||||||
ch_sent = await ctx.send('\N{FILE CABINET} **Caching channels...**')
|
|
||||||
|
|
||||||
if when is None:
|
|
||||||
for channel in channels:
|
|
||||||
async for message in channel.history(limit=None):
|
|
||||||
if message.author.id == int(user):
|
|
||||||
history.append(message)
|
|
||||||
await ch_sent.edit(content='\N{FILE CABINET} **Cached** `{}/{}` **channels.**'.format(channels.index(channel) + 1, len(channels)))
|
|
||||||
await asyncio.sleep(self.RATE_LIMIT)
|
|
||||||
elif when == 'before':
|
|
||||||
for channel in channels:
|
|
||||||
async for message in channel.history(limit=None, before=ref.created_at):
|
|
||||||
if message.author.id == int(user):
|
|
||||||
history.append(message)
|
|
||||||
await ch_sent.edit(content='\N{FILE CABINET} **Cached** `{}/{}` **channels.**'.format(channels.index(channel) + 1, len(channels)))
|
|
||||||
await asyncio.sleep(self.RATE_LIMIT)
|
|
||||||
elif when == 'after':
|
|
||||||
for channel in channels:
|
|
||||||
async for message in channel.history(limit=None, after=ref.created_at):
|
|
||||||
if message.author.id == int(user):
|
|
||||||
history.append(message)
|
|
||||||
await ch_sent.edit(content='\N{FILE CABINET} **Cached** `{}/{}` **channels.**'.format(channels.index(channel) + 1, len(channels)))
|
|
||||||
await asyncio.sleep(self.RATE_LIMIT)
|
|
||||||
elif when == 'about':
|
|
||||||
for channel in channels:
|
|
||||||
async for message in channel.history(limit=None, about=ref.created_at):
|
|
||||||
if message.author.id == int(user):
|
|
||||||
history.append(message)
|
|
||||||
await ch_sent.edit(content='\N{FILE CABINET} **Cached** `{}/{}` **channels.**'.format(channels.index(channel) + 1, len(channels)))
|
|
||||||
await asyncio.sleep(self.RATE_LIMIT)
|
|
||||||
|
|
||||||
est_sent = await ctx.send('\N{STOPWATCH} **Estimated time to delete history:** `{}m {}s`'.format(int(self.RATE_LIMIT * len(history) / 60), int(self.RATE_LIMIT * len(history) % 60)))
|
|
||||||
cont_sent = await ctx.send('{} **Continue?** `Y` or `N`'.format(ctx.author.mention))
|
|
||||||
await self.bot.wait_for('message', check=yes, timeout=10 * 60)
|
|
||||||
await cont_sent.delete()
|
|
||||||
del_sent = await ctx.send('\N{WASTEBASKET} **Deleting messages...**')
|
|
||||||
await del_sent.pin()
|
|
||||||
c = 0
|
|
||||||
for message in history:
|
|
||||||
with suppress(err.NotFound):
|
|
||||||
await message.delete()
|
|
||||||
c += 1
|
|
||||||
await del_sent.edit(content='\N{WASTEBASKET} **Deleted** `{}/{}` **messages.**'.format(history.index(message) + 1, len(history)))
|
|
||||||
await asyncio.sleep(self.RATE_LIMIT)
|
|
||||||
await del_sent.unpin()
|
|
||||||
|
|
||||||
await ctx.send('\N{WASTEBASKET} `{}` **of** <@{}>**\'s messages left in** {}**.**'.format(len(history) - c, user, ctx.guild.name))
|
|
||||||
await ctx.message.add_reaction('\N{WHITE HEAVY CHECK MARK}')
|
|
||||||
|
|
||||||
except exc.CheckFail:
|
|
||||||
await ctx.send('**Deletion aborted.**', delete_after=10)
|
|
||||||
await ctx.message.add_reaction('\N{CROSS MARK}')
|
|
||||||
|
|
||||||
except TimeoutError:
|
|
||||||
await ctx.send('**Deletion timed out.**', delete_after=10)
|
|
||||||
await ctx.message.add_reaction('\N{CROSS MARK}')
|
|
||||||
|
|
||||||
async def delete(self):
|
|
||||||
while self.deleting:
|
|
||||||
message = await self.queue.get()
|
|
||||||
await asyncio.sleep(self.RATE_LIMIT)
|
|
||||||
with suppress(err.NotFound):
|
|
||||||
if not message.pinned:
|
|
||||||
await message.delete()
|
|
||||||
|
|
||||||
print('STOPPED : deleting')
|
|
||||||
|
|
||||||
async def queue_for_deletion(self, channel):
|
|
||||||
def check(msg):
|
|
||||||
if msg.content.lower() == 'stop' and msg.channel is channel and msg.author.guild_permissions.administrator:
|
|
||||||
raise exc.Abort
|
|
||||||
elif msg.channel is channel and not msg.pinned:
|
|
||||||
return True
|
|
||||||
return False
|
|
||||||
|
|
||||||
try:
|
|
||||||
async for message in channel.history(limit=None):
|
|
||||||
if message.content.lower() == 'stop' and message.author.guild_permissions.administrator:
|
|
||||||
raise exc.Abort
|
|
||||||
if not message.pinned:
|
|
||||||
await self.queue.put(message)
|
|
||||||
|
|
||||||
while not self.bot.is_closed():
|
|
||||||
message = await self.bot.wait_for('message', check=check)
|
|
||||||
await self.queue.put(message)
|
|
||||||
|
|
||||||
except exc.Abort:
|
|
||||||
u.tasks['auto_del'].remove(channel.id)
|
|
||||||
u.dump(u.tasks, 'cogs/tasks.pkl')
|
|
||||||
if not u.tasks['auto_del']:
|
|
||||||
self.deleting = False
|
self.deleting = False
|
||||||
print('STOPPED : deleting #{}'.format(channel.id))
|
|
||||||
await channel.send('**Stopped queueing messages for deletion in** {}**.**'.format(channel.mention), delete_after=5)
|
|
||||||
|
|
||||||
@commands.command(name='autodelete', aliases=['autodel', 'ad'])
|
if u.tasks['auto_del']:
|
||||||
@commands.has_permissions(administrator=True)
|
for channel in u.tasks['auto_del']:
|
||||||
@checks.del_ctx()
|
temp = self.bot.get_channel(channel)
|
||||||
async def auto_delete(self, ctx):
|
self.bot.loop.create_task(self.queue_for_deletion(temp))
|
||||||
try:
|
print('AUTO-DELETING : #{}'.format(temp.id))
|
||||||
if ctx.channel.id not in u.tasks['auto_del']:
|
self.bot.loop.create_task(self.delete())
|
||||||
u.tasks['auto_del'].append(ctx.channel.id)
|
self.deleting = True
|
||||||
u.dump(u.tasks, 'cogs/tasks.pkl')
|
|
||||||
self.bot.loop.create_task(self.queue_for_deletion(ctx.channel))
|
@commands.command(name=',prunefromguild', aliases=[',pfg', ',prunefromserver', ',pfs'], brief='Prune a user\'s messages from the guild', description='about flag centers on message 50 of 101 messages\n\npfg \{user id\} [before|after|about] [\{message id\}]\n\nExample:\npfg \{user id\} before \{message id\}')
|
||||||
if not self.deleting:
|
@commands.is_owner()
|
||||||
self.bot.loop.create_task(self.delete())
|
@checks.del_ctx()
|
||||||
self.deleting = True
|
async def prune_all_user(self, ctx, user, when=None, reference=None):
|
||||||
print('AUTO-DELETING : #{}'.format(ctx.channel.id))
|
def yes(msg):
|
||||||
await ctx.send('**Auto-deleting all messages in {}.**'.format(ctx.channel.mention), delete_after=5)
|
if msg.content.lower() == 'y' and msg.channel is ctx.channel and msg.author is ctx.author:
|
||||||
|
return True
|
||||||
|
elif msg.content.lower() == 'n' and msg.channel is ctx.channel and msg.author is ctx.author:
|
||||||
|
raise exc.CheckFail
|
||||||
|
else:
|
||||||
|
return False
|
||||||
|
|
||||||
|
channels = ctx.guild.text_channels
|
||||||
|
if reference is not None:
|
||||||
|
for channel in channels:
|
||||||
|
try:
|
||||||
|
ref = await channel.get_message(reference)
|
||||||
|
|
||||||
|
except err.NotFound:
|
||||||
|
continue
|
||||||
|
|
||||||
|
history = []
|
||||||
|
try:
|
||||||
|
pru_sent = await ctx.send('\N{HOURGLASS} **Pruning** <@{}>**\'s messages will take some time.**'.format(user))
|
||||||
|
ch_sent = await ctx.send('\N{FILE CABINET} **Caching channels...**')
|
||||||
|
|
||||||
|
if when is None:
|
||||||
|
for channel in channels:
|
||||||
|
async for message in channel.history(limit=None):
|
||||||
|
if message.author.id == int(user):
|
||||||
|
history.append(message)
|
||||||
|
await ch_sent.edit(content='\N{FILE CABINET} **Cached** `{}/{}` **channels.**'.format(channels.index(channel) + 1, len(channels)))
|
||||||
|
await asyncio.sleep(self.RATE_LIMIT)
|
||||||
|
elif when == 'before':
|
||||||
|
for channel in channels:
|
||||||
|
async for message in channel.history(limit=None, before=ref.created_at):
|
||||||
|
if message.author.id == int(user):
|
||||||
|
history.append(message)
|
||||||
|
await ch_sent.edit(content='\N{FILE CABINET} **Cached** `{}/{}` **channels.**'.format(channels.index(channel) + 1, len(channels)))
|
||||||
|
await asyncio.sleep(self.RATE_LIMIT)
|
||||||
|
elif when == 'after':
|
||||||
|
for channel in channels:
|
||||||
|
async for message in channel.history(limit=None, after=ref.created_at):
|
||||||
|
if message.author.id == int(user):
|
||||||
|
history.append(message)
|
||||||
|
await ch_sent.edit(content='\N{FILE CABINET} **Cached** `{}/{}` **channels.**'.format(channels.index(channel) + 1, len(channels)))
|
||||||
|
await asyncio.sleep(self.RATE_LIMIT)
|
||||||
|
elif when == 'about':
|
||||||
|
for channel in channels:
|
||||||
|
async for message in channel.history(limit=None, about=ref.created_at):
|
||||||
|
if message.author.id == int(user):
|
||||||
|
history.append(message)
|
||||||
|
await ch_sent.edit(content='\N{FILE CABINET} **Cached** `{}/{}` **channels.**'.format(channels.index(channel) + 1, len(channels)))
|
||||||
|
await asyncio.sleep(self.RATE_LIMIT)
|
||||||
|
|
||||||
|
est_sent = await ctx.send('\N{STOPWATCH} **Estimated time to delete history:** `{}m {}s`'.format(int(self.RATE_LIMIT * len(history) / 60), int(self.RATE_LIMIT * len(history) % 60)))
|
||||||
|
cont_sent = await ctx.send('{} **Continue?** `Y` or `N`'.format(ctx.author.mention))
|
||||||
|
await self.bot.wait_for('message', check=yes, timeout=10 * 60)
|
||||||
|
await cont_sent.delete()
|
||||||
|
del_sent = await ctx.send('\N{WASTEBASKET} **Deleting messages...**')
|
||||||
|
await del_sent.pin()
|
||||||
|
c = 0
|
||||||
|
for message in history:
|
||||||
|
with suppress(err.NotFound):
|
||||||
|
await message.delete()
|
||||||
|
c += 1
|
||||||
|
await del_sent.edit(content='\N{WASTEBASKET} **Deleted** `{}/{}` **messages.**'.format(history.index(message) + 1, len(history)))
|
||||||
|
await asyncio.sleep(self.RATE_LIMIT)
|
||||||
|
await del_sent.unpin()
|
||||||
|
|
||||||
|
await ctx.send('\N{WASTEBASKET} `{}` **of** <@{}>**\'s messages left in** {}**.**'.format(len(history) - c, user, ctx.guild.name))
|
||||||
|
await ctx.message.add_reaction('\N{WHITE HEAVY CHECK MARK}')
|
||||||
|
|
||||||
|
except exc.CheckFail:
|
||||||
|
await ctx.send('**Deletion aborted.**', delete_after=10)
|
||||||
|
await ctx.message.add_reaction('\N{CROSS MARK}')
|
||||||
|
|
||||||
|
except TimeoutError:
|
||||||
|
await ctx.send('**Deletion timed out.**', delete_after=10)
|
||||||
|
await ctx.message.add_reaction('\N{CROSS MARK}')
|
||||||
|
|
||||||
|
async def delete(self):
|
||||||
|
while self.deleting:
|
||||||
|
message = await self.queue.get()
|
||||||
|
await asyncio.sleep(self.RATE_LIMIT)
|
||||||
|
with suppress(err.NotFound):
|
||||||
|
if not message.pinned:
|
||||||
|
await message.delete()
|
||||||
|
|
||||||
|
print('STOPPED : deleting')
|
||||||
|
|
||||||
|
async def queue_for_deletion(self, channel):
|
||||||
|
def check(msg):
|
||||||
|
if msg.content.lower() == 'stop' and msg.channel is channel and msg.author.guild_permissions.administrator:
|
||||||
|
raise exc.Abort
|
||||||
|
elif msg.channel is channel and not msg.pinned:
|
||||||
|
return True
|
||||||
|
return False
|
||||||
|
|
||||||
|
try:
|
||||||
|
async for message in channel.history(limit=None):
|
||||||
|
if message.content.lower() == 'stop' and message.author.guild_permissions.administrator:
|
||||||
|
raise exc.Abort
|
||||||
|
if not message.pinned:
|
||||||
|
await self.queue.put(message)
|
||||||
|
|
||||||
|
while not self.bot.is_closed():
|
||||||
|
message = await self.bot.wait_for('message', check=check)
|
||||||
|
await self.queue.put(message)
|
||||||
|
|
||||||
|
except exc.Abort:
|
||||||
|
u.tasks['auto_del'].remove(channel.id)
|
||||||
|
u.dump(u.tasks, 'cogs/tasks.pkl')
|
||||||
|
if not u.tasks['auto_del']:
|
||||||
|
self.deleting = False
|
||||||
|
print('STOPPED : deleting #{}'.format(channel.id))
|
||||||
|
await channel.send('**Stopped queueing messages for deletion in** {}**.**'.format(channel.mention), delete_after=5)
|
||||||
|
|
||||||
|
@commands.command(name='autodelete', aliases=['autodel', 'ad'])
|
||||||
|
@commands.has_permissions(administrator=True)
|
||||||
|
@checks.del_ctx()
|
||||||
|
async def auto_delete(self, ctx):
|
||||||
|
try:
|
||||||
|
if ctx.channel.id not in u.tasks['auto_del']:
|
||||||
|
u.tasks['auto_del'].append(ctx.channel.id)
|
||||||
|
u.dump(u.tasks, 'cogs/tasks.pkl')
|
||||||
|
self.bot.loop.create_task(self.queue_for_deletion(ctx.channel))
|
||||||
|
if not self.deleting:
|
||||||
|
self.bot.loop.create_task(self.delete())
|
||||||
|
self.deleting = True
|
||||||
|
print('AUTO-DELETING : #{}'.format(ctx.channel.id))
|
||||||
|
await ctx.send('**Auto-deleting all messages in {}.**'.format(ctx.channel.mention), delete_after=5)
|
||||||
|
await ctx.message.add_reaction('\N{WHITE HEAVY CHECK MARK}')
|
||||||
|
else:
|
||||||
|
raise exc.Exists
|
||||||
|
|
||||||
|
except exc.Exists:
|
||||||
|
await ctx.send('**Already auto-deleting in {}.** Type `stop` to stop.'.format(ctx.channel.mention), delete_after=10)
|
||||||
|
await ctx.message.add_reaction('\N{CROSS MARK}')
|
||||||
|
|
||||||
|
@commands.command(name='deletecommands', aliases=['delcmds'])
|
||||||
|
@commands.has_permissions(administrator=True)
|
||||||
|
async def delete_commands(self, ctx):
|
||||||
|
if ctx.guild.id not in u.settings['del_ctx']:
|
||||||
|
u.settings['del_ctx'].append(ctx.guild.id)
|
||||||
|
else:
|
||||||
|
u.settings['del_ctx'].remove(ctx.guild.id)
|
||||||
|
u.dump(u.settings, 'settings.pkl')
|
||||||
|
|
||||||
|
await ctx.send('**Delete command invocations:** `{}`'.format(ctx.guild.id in u.settings['del_ctx']))
|
||||||
await ctx.message.add_reaction('\N{WHITE HEAVY CHECK MARK}')
|
await ctx.message.add_reaction('\N{WHITE HEAVY CHECK MARK}')
|
||||||
else:
|
|
||||||
raise exc.Exists
|
|
||||||
|
|
||||||
except exc.Exists:
|
@commands.command(name='setprefix', aliases=['setpre', 'spre'])
|
||||||
await ctx.send('**Already auto-deleting in {}.** Type `stop` to stop.'.format(ctx.channel.mention), delete_after=10)
|
@commands.has_permissions(administrator=True)
|
||||||
await ctx.message.add_reaction('\N{CROSS MARK}')
|
async def set_prefix(self, ctx, prefix=None):
|
||||||
|
if prefix is not None:
|
||||||
|
u.settings['prefixes'][ctx.guild.id] = prefix
|
||||||
|
else:
|
||||||
|
with suppress(KeyError):
|
||||||
|
del u.settings['prefixes'][ctx.guild.id]
|
||||||
|
|
||||||
@commands.command(name='deletecommands', aliases=['delcmds'])
|
await ctx.send(f'**Prefix set to:** `{"` or `".join(prefix if ctx.guild.id in u.settings["prefixes"] else u.config["prefix"])}`')
|
||||||
@commands.has_permissions(administrator=True)
|
await ctx.message.add_reaction('\N{WHITE HEAVY CHECK MARK}')
|
||||||
async def delete_commands(self, ctx):
|
|
||||||
if ctx.guild.id not in u.settings['del_ctx']:
|
|
||||||
u.settings['del_ctx'].append(ctx.guild.id)
|
|
||||||
else:
|
|
||||||
u.settings['del_ctx'].remove(ctx.guild.id)
|
|
||||||
u.dump(u.settings, 'settings.pkl')
|
|
||||||
|
|
||||||
await ctx.send('**Delete command invocations:** `{}`'.format(ctx.guild.id in u.settings['del_ctx']))
|
|
||||||
await ctx.message.add_reaction('\N{WHITE HEAVY CHECK MARK}')
|
|
||||||
|
|
|
@ -18,13 +18,19 @@ from utils import utils as u
|
||||||
|
|
||||||
# log.basicConfig(level=log.INFO)
|
# log.basicConfig(level=log.INFO)
|
||||||
|
|
||||||
bot = commands.Bot(command_prefix=u.config['prefix'], description='Experimental miscellaneous bot')
|
|
||||||
|
|
||||||
|
def get_prefix(bot, message):
|
||||||
|
if isinstance(message.guild, d.Guild) and message.guild.id in u.settings['prefixes']:
|
||||||
|
return u.settings['prefixes'][message.guild.id]
|
||||||
|
return u.config['prefix']
|
||||||
|
|
||||||
|
|
||||||
|
bot = commands.Bot(command_prefix=get_prefix, description='Experimental miscellaneous bot')
|
||||||
|
|
||||||
# Send and print ready message to #testing and console after logon
|
# Send and print ready message to #testing and console after logon
|
||||||
@bot.event
|
@bot.event
|
||||||
async def on_ready():
|
async def on_ready():
|
||||||
from cogs import booru, info, management, owner, tools
|
from cogs import booru, info, management, owner, tools
|
||||||
|
|
||||||
bot.add_cog(tools.Utils(bot))
|
bot.add_cog(tools.Utils(bot))
|
||||||
bot.add_cog(owner.Bot(bot))
|
bot.add_cog(owner.Bot(bot))
|
||||||
|
@ -33,7 +39,7 @@ async def on_ready():
|
||||||
bot.add_cog(info.Info(bot))
|
bot.add_cog(info.Info(bot))
|
||||||
bot.add_cog(booru.MsG(bot))
|
bot.add_cog(booru.MsG(bot))
|
||||||
|
|
||||||
# bot.loop.create_task(u.clear(booru.temp_urls, 30*60))
|
# bot.loop.create_task(u.clear(booru.temp_urls, 30*60))
|
||||||
|
|
||||||
if u.config['playing'] is not 'None':
|
if u.config['playing'] is not 'None':
|
||||||
await bot.change_presence(game=d.Game(name=u.config['playing']))
|
await bot.change_presence(game=d.Game(name=u.config['playing']))
|
||||||
|
|
Loading…
Reference in a new issue