mirror of
https://github.com/myned/modufur.git
synced 2024-12-24 22:27:28 +00:00
Refactor kwargs pre-command method
This commit is contained in:
parent
57faf39b96
commit
d8c5b743d4
2 changed files with 21 additions and 6 deletions
|
@ -1405,6 +1405,7 @@ class MsG:
|
||||||
async def __get_blacklist_aliases(self, ctx, *args):
|
async def __get_blacklist_aliases(self, ctx, *args):
|
||||||
guild = ctx.guild if isinstance(
|
guild = ctx.guild if isinstance(
|
||||||
ctx.guild, d.Guild) else ctx.channel
|
ctx.guild, d.Guild) else ctx.channel
|
||||||
|
args, lst = u.kwargs(args)
|
||||||
|
|
||||||
aliases = {}
|
aliases = {}
|
||||||
# Creates temp aliases based on context
|
# Creates temp aliases based on context
|
||||||
|
@ -1413,12 +1414,14 @@ class MsG:
|
||||||
aliases[tag] = list(self.aliases[tag])
|
aliases[tag] = list(self.aliases[tag])
|
||||||
|
|
||||||
# paginator.add_line(f'{tag}\n```{" ".join(alias_list)}```')
|
# paginator.add_line(f'{tag}\n```{" ".join(alias_list)}```')
|
||||||
|
args, lst = u.kwargs(args)
|
||||||
|
|
||||||
await formatter.paginate(ctx, aliases, start='\N{NO ENTRY SIGN} **Contextual blacklist aliases:**\n')
|
await formatter.paginate(ctx, aliases, start='\N{NO ENTRY SIGN} **Contextual blacklist aliases:**\n')
|
||||||
|
|
||||||
@_get_blacklist.command(name='global', aliases=['gl', 'g'], brief='Get current global blacklist', description='Get current global blacklist\n\nThis applies to all booru commands, in accordance with Discord\'s ToS agreement\n\nExample:\n\{p\}bl get global')
|
@_get_blacklist.command(name='global', aliases=['gl', 'g'], brief='Get current global blacklist', description='Get current global blacklist\n\nThis applies to all booru commands, in accordance with Discord\'s ToS agreement\n\nExample:\n\{p\}bl get global')
|
||||||
async def __get_global_blacklist(self, ctx, *args):
|
async def __get_global_blacklist(self, ctx, *args):
|
||||||
await ctx.send('\N{NO ENTRY SIGN} **Global blacklist:**\n```\n{}```'.format(' '.join(self.blacklists['global_blacklist'])))
|
await ctx.send('\N{NO ENTRY SIGN} **Global blacklist:**\n```\n{}```'.format(' '.join(self.blacklists['global_blacklist'])))
|
||||||
|
args, lst = u.kwargs(args)
|
||||||
|
|
||||||
@_get_blacklist.command(name='channel', aliases=['ch', 'c'], brief='Get current channel blacklist', description='Get current channel blacklist\n\nThis is based on context - the channel where the command was executed\n\nExample:\{p\}bl get channel')
|
@_get_blacklist.command(name='channel', aliases=['ch', 'c'], brief='Get current channel blacklist', description='Get current channel blacklist\n\nThis is based on context - the channel where the command was executed\n\nExample:\{p\}bl get channel')
|
||||||
async def __get_channel_blacklist(self, ctx, *args):
|
async def __get_channel_blacklist(self, ctx, *args):
|
||||||
|
@ -1532,8 +1535,7 @@ class MsG:
|
||||||
@_add_tags.command(name='global', aliases=['gl', 'g'])
|
@_add_tags.command(name='global', aliases=['gl', 'g'])
|
||||||
@cmds.is_owner()
|
@cmds.is_owner()
|
||||||
async def __add_global_tags(self, ctx, *args):
|
async def __add_global_tags(self, ctx, *args):
|
||||||
kwargs = u.get_kwargs(ctx, args)
|
tags, lst = u.kwargs(args)
|
||||||
tags = kwargs['remaining']
|
|
||||||
|
|
||||||
try:
|
try:
|
||||||
async with ctx.channel.typing():
|
async with ctx.channel.typing():
|
||||||
|
@ -1552,8 +1554,7 @@ class MsG:
|
||||||
@_add_tags.command(name='channel', aliases=['ch', 'c'], brief='@manage_channel@ Add tag(s) to the current channel blacklist (requires manage_channel)', description='Add tag(s) to the current channel blacklist ')
|
@_add_tags.command(name='channel', aliases=['ch', 'c'], brief='@manage_channel@ Add tag(s) to the current channel blacklist (requires manage_channel)', description='Add tag(s) to the current channel blacklist ')
|
||||||
@cmds.has_permissions(manage_channels=True)
|
@cmds.has_permissions(manage_channels=True)
|
||||||
async def __add_channel_tags(self, ctx, *args):
|
async def __add_channel_tags(self, ctx, *args):
|
||||||
kwargs = u.get_kwargs(ctx, args)
|
tags, lst = u.kwargs(args)
|
||||||
tags = kwargs['remaining']
|
|
||||||
|
|
||||||
guild = ctx.guild if isinstance(
|
guild = ctx.guild if isinstance(
|
||||||
ctx.guild, d.Guild) else ctx.channel
|
ctx.guild, d.Guild) else ctx.channel
|
||||||
|
@ -1576,6 +1577,7 @@ class MsG:
|
||||||
async def __add_user_tags(self, ctx, *args):
|
async def __add_user_tags(self, ctx, *args):
|
||||||
kwargs = u.get_kwargs(ctx, args)
|
kwargs = u.get_kwargs(ctx, args)
|
||||||
tags = kwargs['remaining']
|
tags = kwargs['remaining']
|
||||||
|
tags, lst = u.kwargs(args)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
async with ctx.channel.typing():
|
async with ctx.channel.typing():
|
||||||
|
@ -1612,6 +1614,7 @@ class MsG:
|
||||||
raise exc.TagError(tag)
|
raise exc.TagError(tag)
|
||||||
|
|
||||||
u.dump(self.blacklists, 'cogs/blacklists.pkl')
|
u.dump(self.blacklists, 'cogs/blacklists.pkl')
|
||||||
|
tags, lst = u.kwargs(args)
|
||||||
|
|
||||||
await ctx.send('**Removed from global blacklist:**\n```\n{}```'.format(' '.join(tags)))
|
await ctx.send('**Removed from global blacklist:**\n```\n{}```'.format(' '.join(tags)))
|
||||||
|
|
||||||
|
@ -1676,7 +1679,7 @@ class MsG:
|
||||||
@_clear_blacklist.command(name='global', aliases=['gl', 'g'])
|
@_clear_blacklist.command(name='global', aliases=['gl', 'g'])
|
||||||
@cmds.is_owner()
|
@cmds.is_owner()
|
||||||
async def __clear_global_blacklist(self, ctx, *args):
|
async def __clear_global_blacklist(self, ctx, *args):
|
||||||
self.blacklists['global_blacklist'].clear()
|
args, lst = u.kwargs(args)
|
||||||
u.dump(self.blacklists, 'cogs/blacklists.pkl')
|
u.dump(self.blacklists, 'cogs/blacklists.pkl')
|
||||||
|
|
||||||
await ctx.send('**Global blacklist cleared**')
|
await ctx.send('**Global blacklist cleared**')
|
||||||
|
@ -1684,7 +1687,7 @@ class MsG:
|
||||||
@_clear_blacklist.command(name='channel', aliases=['ch', 'c'])
|
@_clear_blacklist.command(name='channel', aliases=['ch', 'c'])
|
||||||
@cmds.has_permissions(manage_channels=True)
|
@cmds.has_permissions(manage_channels=True)
|
||||||
async def __clear_channel_blacklist(self, ctx, *args):
|
async def __clear_channel_blacklist(self, ctx, *args):
|
||||||
guild = ctx.guild if isinstance(
|
args, lst = u.kwargs(args)
|
||||||
ctx.guild, d.Guild) else ctx.channel
|
ctx.guild, d.Guild) else ctx.channel
|
||||||
|
|
||||||
with suppress(KeyError):
|
with suppress(KeyError):
|
||||||
|
@ -1692,6 +1695,7 @@ class MsG:
|
||||||
u.dump(self.blacklists, 'cogs/blacklists.pkl')
|
u.dump(self.blacklists, 'cogs/blacklists.pkl')
|
||||||
|
|
||||||
await ctx.send('{} **blacklist cleared**'.format(ctx.channel.mention))
|
await ctx.send('{} **blacklist cleared**'.format(ctx.channel.mention))
|
||||||
|
args, lst = u.kwargs(args)
|
||||||
|
|
||||||
@_clear_blacklist.command(name='me', aliases=['m'])
|
@_clear_blacklist.command(name='me', aliases=['m'])
|
||||||
async def __clear_user_blacklist(self, ctx, *args):
|
async def __clear_user_blacklist(self, ctx, *args):
|
||||||
|
|
|
@ -131,6 +131,17 @@ def generate_embed(ctx, *, title=d.Embed.Empty, kind='rich', description=d.Embed
|
||||||
|
|
||||||
return embed
|
return embed
|
||||||
|
|
||||||
|
def kwargs(args):
|
||||||
|
params = list(args)
|
||||||
|
lst = 'blacklist'
|
||||||
|
|
||||||
|
for switch in ('-a', '--aliases'):
|
||||||
|
if switch in params:
|
||||||
|
lst = 'aliases'
|
||||||
|
params.remove(switch)
|
||||||
|
|
||||||
|
return params, lst
|
||||||
|
|
||||||
def get_kwargs(ctx, args, *, limit=False):
|
def get_kwargs(ctx, args, *, limit=False):
|
||||||
remaining = list(args[:])
|
remaining = list(args[:])
|
||||||
rm = False
|
rm = False
|
||||||
|
|
Loading…
Reference in a new issue