myned/modufur
1
0
Fork 0
mirror of https://github.com/myned/modufur.git synced 2024-12-24 14:27:27 +00:00
Code Issues Activity

Added reaction paginator, WIP .format conversion, del blacklist

Browse source
This commit is contained in:
Myned 2017-10-11 03:12:13 -04:00
parent 79c53fd8d0
commit 25c28a509f
1 changed files with 188 additions and 120 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

308
src/main/cogs/booru.py
View file

@ -65,23 +65,23 @@ class MsG:
except exc.ImageError: except exc.ImageError:
await ctx.send('❌ ' + ctx.message.author.mention + ' **No image found.**') await ctx.send('❌ ' + ctx.message.author.mention + ' **No image found.**')
except Exception: except Exception:
await ctx.send(exc.base + '\n```python' + traceback.format_exc(limit=1) + '```') await ctx.send('{}\n```{}```'.format(exc.base, tb.format_exc(limit=1)))
traceback.print_exc(limit=1) tb.print_exc()
# Tag aliases # Tag aliases
@commands.command(name='aliases', aliases=['a'], brief='e621 Tag aliases', description='e621 | NSFW\nSearch aliases for given tag') @commands.command(aliases=['alias', 'a'], brief='e621 Tag aliases', description='e621 | NSFW\nSearch aliases for given tag')
@checks.del_ctx() @checks.del_ctx()
async def alias(self, ctx, tag): async def aliases(self, ctx, tag):
global headers global HEADERS
aliases = [] aliases = []
try: try:
alias_request = requests.get('https://e621.net/tag_alias/index.json?aliased_to=' + tag + '&approved=true', headers=headers).json() alias_request = requests.get('https://e621.net/tag_alias/index.json?aliased_to=' + tag + '&approved=true', headers=HEADERS).json()
for dic in alias_request: for dic in alias_request:
aliases.append(dic['name']) aliases.append(dic['name'])
await ctx.send('✅ `' + tag + '` **aliases:**\n```' + formatter.tostring(aliases) + '```') await ctx.send('✅ `' + tag + '` **aliases:**\n```' + formatter.tostring(aliases) + '```')
except Exception: except Exception:
await ctx.send(exc.base + '\n```python' + traceback.format_exc(limit=1) + '```') await ctx.send('{}\n```{}```'.format(exc.base, tb.format_exc(limit=1)))
traceback.print_exc(limit=1) tb.print_exc()
# Reverse image searches a linked image using the public iqdb # Reverse image searches a linked image using the public iqdb
@commands.command(name='reverse', aliases=['rev', 'ris'], brief='e621 Reverse image search', description='e621 | NSFW\nReverse-search an image with given URL') @commands.command(name='reverse', aliases=['rev', 'ris'], brief='e621 Reverse image search', description='e621 | NSFW\nReverse-search an image with given URL')
@ -89,133 +89,201 @@ class MsG:
async def reverse_image_search(self, ctx, url): async def reverse_image_search(self, ctx, url):
try: try:
await ctx.trigger_typing() await ctx.trigger_typing()
await ctx.send('✅ ' + ctx.message.author.mention + ' **Probable match:**\n' + scraper.check_match('http://iqdb.harry.lu/?url=' + url)) await ctx.send('✅ ' + ctx.message.author.mention + ' **Probable match:**\n' + scraper.check_match('http://iqdb.harry.lu/?url={}'.format(url)))
except exc.MatchError: except exc.MatchError:
await ctx.send('❌ ' + ctx.message.author.mention + ' **No probable match.**', delete_after=10) await ctx.send('❌ ' + ctx.message.author.mention + ' **No probable match.**', delete_after=10)
except Exception: except Exception:
await ctx.send(exc.base + '\n```python' + traceback.format_exc(limit=1) + '```') await ctx.send('{}\n```{}```'.format(exc.base, tb.format_exc(limit=1)))
traceback.print_exc(limit=1) tb.print_exc()
@commands.command(name='e621p', aliases=['e6p', '6p'])
@checks.del_ctx()
@checks.is_nsfw()
async def e621_paginator(self, ctx, *args):
def react(reaction, user):
if reaction.emoji == '⬅' and reaction.message.content == paginator.content and user is ctx.message.author: raise exc.Left
elif reaction.emoji == '🚫' and reaction.message.content == paginator.content and user is ctx.message.author: raise exc.Abort
elif reaction.emoji == '📁' and reaction.message.content == paginator.content and user is ctx.message.author: raise exc.Save
elif reaction.emoji == '➡' and reaction.message.content == paginator.content and user is ctx.message.author: raise exc.Right
else: return False
user = ctx.message.author
args = list(args)
limit = 100
try:
await ctx.trigger_typing()
c = 1
posts = self.check_return_posts(ctx=ctx, booru='e621', tags=args, limit=limit)
starred = []
embed = d.Embed(title='/post/{}'.format(list(posts.keys())[c-1]), url='https://e621.net/post/show/{}'.format(list(posts.keys())[c-1]), color=ctx.me.color).set_image(url=list(posts.values())[c-1])
embed.set_author(name=formatter.tostring(args, random=True), url='https://e621.net/post?tags={}'.format(','.join(args)), icon_url=user.avatar_url)
embed.set_footer(text='e621', icon_url='http://ndl.mgccw.com/mu3/app/20141013/18/1413204353554/icon/icon_xl.png')
paginator = await ctx.send(embed=embed)
await paginator.add_reaction('⬅')
await paginator.add_reaction('🚫')
await paginator.add_reaction('📁')
await paginator.add_reaction('➡')
await asyncio.sleep(1)
while True:
try:
await self.bot.wait_for('reaction_add', check=react, timeout=5*60)
except exc.Left:
if c > 1:
c -= 1
embed.title = '/post/{}'.format(list(posts.keys())[c-1])
embed.url = 'https://e621.net/post/show/{}'.format(list(posts.keys())[c-1])
embed.set_image(url=list(posts.values())[c-1])
await paginator.edit(embed=embed)
except exc.Save:
if list(posts.values())[c-1] not in starred:
starred.append(list(posts.values())[c-1])
except exc.Right:
if c % limit == 0:
await ctx.trigger_typing()
try: posts.update(self.check_return_posts(ctx=ctx, booru='e621', tags=args, limit=limit, previous=posts))
except exc.NotFound:
await paginator.edit(content='❌ **No more images found.**')
c += 1
embed.title = '/post/{}'.format(list(posts.keys())[c-1])
embed.url = 'https://e621.net/post/show/{}'.format(list(posts.keys())[c-1])
embed.set_image(url=list(posts.values())[c-1])
await paginator.edit(embed=embed)
except exc.Abort: await paginator.edit(content='🚫 **Exited paginator.**')
except exc.TagBlacklisted as e: await ctx.send('❌ `{}` **blacklisted.**'.format(e), delete_after=10)
except exc.TagBoundsError as e: await ctx.send('❌ `{}` **out of bounds.** Tags limited to 5, currently.'.format(e), delete_after=10)
except exc.Timeout: await ctx.send('❌ **Request timed out.**')
except asyncio.TimeoutError: await paginator.edit(content='❌ **Paginator timed out.**')
except Exception:
await ctx.send('{}\n```{}```'.format(exc.base, tb.format_exc(limit=1)))
tb.print_exc()
finally:
if starred:
for url in starred: await user.send(url)
@e621_paginator.error
async def e621_paginator_error(self, ctx, error):
if isinstance(error, errext.CheckFailure):
return await ctx.send('❌ <#' + str(ctx.message.channel.id) + '> **is not an NSFW channel.**', delete_after=10)
# Searches for and returns images from e621.net given tags when not blacklisted # Searches for and returns images from e621.net given tags when not blacklisted
@commands.command(aliases=['e6', '6'], brief='e621 | NSFW', description='e621 | NSFW\nTag-based search for e621.net\n\nYou can only search 5 tags and 6 images at once for now.\ne6 [tags...] ([# of images])') @commands.command(aliases=['e6', '6'], brief='e621 | NSFW', description='e621 | NSFW\nTag-based search for e621.net\n\nYou can only search 5 tags and 6 images at once for now.\ne6 [tags...] ([# of images])')
@checks.del_ctx() @checks.del_ctx()
@checks.is_nsfw() @checks.is_nsfw()
async def e621(self, ctx, *args): async def e621(self, ctx, *args):
global blacklists
# global temp_urls # global temp_urls
args = list(args) args = list(args)
limit = 1
try: try:
await ctx.trigger_typing() await ctx.trigger_typing()
await self.check_send_urls(ctx, 'e621', args) # Checks for, defines, and removes limit from args
except exc.TagBlacklisted as e: for arg in args:
await ctx.send('❌ `' + str(e) + '` **blacklisted.**', delete_after=10) if len(arg) == 1:
except exc.BoundsError as e: if int(arg) <= 6 and int(arg) >= 1:
await ctx.send('❌ `' + str(e) + '` **out of bounds.**', delete_after=10) limit = int(arg)
except exc.TagBoundsError as e: args.remove(arg)
await ctx.send('❌ `' + str(e) + '` **out of bounds.** Tags limited to 5, currently.', delete_after=10) else: raise exc.BoundsError(arg)
except ValueError:
await ctx.send('❌ `' + args[-1] + '` **not a valid limit.**', delete_after=10)
except exc.NotFound:
await ctx.send('❌ **Post not found.**', delete_after=10)
except exc.Timeout:
await ctx.send('❌ **Request timed out.**')
posts = self.check_return_posts(ctx=ctx, booru='e621', tags=args, limit=limit)#, previous=temp_urls.get(ctx.message.author.id, [])) posts = self.check_return_posts(ctx=ctx, booru='e621', tags=args, limit=limit)#, previous=temp_urls.get(ctx.message.author.id, []))
for ident, url in posts.items():
embed = d.Embed(title='/post/{}'.format(ident), url='https://e621.net/post/show/{}'.format(ident), color=ctx.me.color).set_image(url=url)
embed.set_author(name=formatter.tostring(args, random=True), url='https://e621.net/post?tags={}'.format(','.join(args)), icon_url=ctx.message.author.avatar_url)
embed.set_footer(text='e621', icon_url='http://ndl.mgccw.com/mu3/app/20141013/18/1413204353554/icon/icon_xl.png')
await ctx.send(embed=embed)
# temp_urls.setdefault(ctx.message.author.id, []).extend(posts.values()) # temp_urls.setdefault(ctx.message.author.id, []).extend(posts.values())
except exc.TagBlacklisted as e: await ctx.send('❌ `' + str(e) + '` **blacklisted.**', delete_after=10)
except exc.BoundsError as e: await ctx.send('❌ `' + str(e) + '` **out of bounds.**', delete_after=10)
except exc.TagBoundsError as e: await ctx.send('❌ `' + str(e) + '` **out of bounds.** Tags limited to 5, currently.', delete_after=10)
except exc.NotFound as e: await ctx.send('❌ `' + str(e) + '` **not found.**', delete_after=10)
except exc.Timeout: await ctx.send('❌ **Request timed out.**')
except Exception: except Exception:
await ctx.send(exc.base + '\n```python' + traceback.format_exc(limit=1) + '```') await ctx.send('{}\n```{}```'.format(exc.base, tb.format_exc(limit=1)))
traceback.print_exc() tb.print_exc()
tools.command_dict.setdefault(str(ctx.message.author.id), {}).update({'command': ctx.command, 'args': ctx.args}) tools.command_dict.setdefault(str(ctx.message.author.id), {}).update({'command': ctx.command, 'args': ctx.args})
@e621.error @e621.error
async def e621_error(self, ctx, error): async def e621_error(self, ctx, error):
if isinstance(error, errors.CheckFailure): if isinstance(error, errext.CheckFailure):
return await ctx.send('❌ <#' + str(ctx.message.channel.id) + '> **is not an NSFW channel.**', delete_after=10) return await ctx.send('❌ <#' + str(ctx.message.channel.id) + '> **is not an NSFW channel.**', delete_after=10)
# Searches for and returns images from e926.net given tags when not blacklisted # Searches for and returns images from e926.net given tags when not blacklisted
@commands.command(aliases=['e9', '9'], brief='e926 | SFW', description='e926 | SFW\nTag-based search for e926.net\n\nYou can only search 5 tags and 6 images at once for now.\ne9 [tags...] ([# of images])') @commands.command(aliases=['e9', '9'], brief='e926 | SFW', description='e926 | SFW\nTag-based search for e926.net\n\nYou can only search 5 tags and 6 images at once for now.\ne9 [tags...] ([# of images])')
@checks.del_ctx() @checks.del_ctx()
async def e926(self, ctx, *args): async def e926(self, ctx, *args):
global blacklists
# global temp_urls # global temp_urls
args = list(args) args = list(args)
limit = 1
try: try:
await ctx.trigger_typing() await ctx.trigger_typing()
await self.check_send_urls(ctx, 'e926', args) # Checks for, defines, and removes limit from args
except exc.TagBlacklisted as e: for arg in args:
await ctx.send('❌ `' + str(e) + '` **blacklisted.**', delete_after=10) if len(arg) == 1:
except exc.BoundsError as e: if int(arg) <= 6 and int(arg) >= 1:
await ctx.send('❌ `' + str(e) + '` **out of bounds.**', delete_after=10) limit = int(arg)
except exc.TagBoundsError as e: args.remove(arg)
await ctx.send('❌ `' + str(e) + '` **out of bounds.** Tags limited to 5, currently.', delete_after=10) else: raise exc.BoundsError(arg)
except ValueError:
await ctx.send('❌ `' + args[-1] + '` **not a valid limit.**', delete_after=10)
except exc.NotFound:
await ctx.send('❌ **Post not found.**', delete_after=10)
except exc.Timeout:
await ctx.send('❌ **Request timed out.**')
posts = self.check_return_posts(ctx=ctx, booru='e926', tags=args, limit=limit)#, previous=temp_urls.get(ctx.message.author.id, [])) posts = self.check_return_posts(ctx=ctx, booru='e926', tags=args, limit=limit)#, previous=temp_urls.get(ctx.message.author.id, []))
for ident, url in posts.items():
embed = d.Embed(title='/post/{}'.format(ident), url='https://e926.net/post/show/{}'.format(ident), color=ctx.me.color).set_image(url=url)
embed.set_author(name=formatter.tostring(args, random=True), url='https://e621.net/post?tags={}'.format(','.join(args)), icon_url=ctx.message.author.avatar_url)
embed.set_footer(text='e926', icon_url='http://ndl.mgccw.com/mu3/app/20141013/18/1413204353554/icon/icon_xl.png')
await ctx.send(embed=embed)
# temp_urls.setdefault(ctx.message.author.id, []).extend(posts.values()) # temp_urls.setdefault(ctx.message.author.id, []).extend(posts.values())
except exc.TagBlacklisted as e: await ctx.send('❌ `' + str(e) + '` **blacklisted.**', delete_after=10)
except exc.BoundsError as e: await ctx.send('❌ `' + str(e) + '` **out of bounds.**', delete_after=10)
except exc.TagBoundsError as e: await ctx.send('❌ `' + str(e) + '` **out of bounds.** Tags limited to 5, currently.', delete_after=10)
except exc.NotFound as e: await ctx.send('❌ `' + str(e) + '` **not found.**', delete_after=10)
except exc.Timeout: await ctx.send('❌ **Request timed out.**')
except Exception: except Exception:
await ctx.send(exc.base + '\n```python' + traceback.format_exc(limit=1) + '```') await ctx.send('{}\n```{}```'.format(exc.base, tb.format_exc(limit=1)))
traceback.print_exc(limit=1) tb.print_exc()
# Messy code that checks image limit and tags in blacklists # Messy code that checks image limit and tags in blacklists
async def check_send_urls(self, ctx, booru, args): def check_return_posts(self, *, ctx, booru='e621', tags=[], limit=1, previous=[]):
global blacklists, aliases, headers global blacklists, aliases, HEADERS
if isinstance(ctx.message.guild, discord.Guild):
guild = ctx.message.guild if isinstance(ctx.message.guild, d.Guild): guild = ctx.message.guild
else: else: guild = ctx.message.channel
guild = ctx.message.channel
channel = ctx.message.channel channel = ctx.message.channel
user = ctx.message.author user = ctx.message.author
blacklist = [] blacklist = []
urls = []
limit = 1
c = 0
if len(args) > 5:
raise exc.TagBoundsError(formatter.tostring(args[5:]))
# Checks for, defines, and removes limit from end of args
if args and len(args[-1]) == 1:
if int(args[-1]) <= 5 and int(args[-1]) >= 1:
limit = int(args[-1])
args.pop()
else:
raise exc.BoundsError(args[-1])
# Creates temp blacklist based on context # Creates temp blacklist based on context
for k, v in aliases['global_blacklist'].items(): for k, v in aliases['global_blacklist'].items(): blacklist.extend([k] + v)
blacklist.extend([k] + v) for k, v in aliases['guild_blacklist'].get(str(guild.id), {}).get(str(channel.id), {}).items(): blacklist.extend([k] + v)
for k, v in aliases['guild_blacklist'].get(str(guild.id), {}).get(str(channel.id), {}).items(): for k, v in aliases['user_blacklist'].get(str(user.id), {}).items(): blacklist.extend([k] + v)
blacklist.extend([k] + v)
for k, v in aliases['user_blacklist'].get(str(user.id), {}).items():
blacklist.extend([k] + v)
# Checks if tags are in local blacklists # Checks if tags are in local blacklists
if args: if tags:
for tag in args: if len(tags) > 5: raise exc.TagBoundsError(formatter.tostring(tags[5:]))
if tag == 'swf' or tag == 'webm' or tag in blacklist: for tag in tags:
raise exc.TagBlacklisted(tag) if tag == 'swf' or tag == 'webm' or tag in blacklist: raise exc.TagBlacklisted(tag)
# Checks for blacklisted tags in endpoint blacklists - try/except is for continuing the parent loop # Checks for blacklisted tags in endpoint blacklists - try/except is for continuing the parent loop
while len(urls) < limit: posts = {}
request = requests.get('https://' + booru + '.net/post/index.json?limit=6&tags=order:random' + formatter.tostring_commas(args), headers=headers).json() c = 0
if not request: while len(posts) < limit:
raise exc.NotFound if c == 50 + limit: raise exc.Timeout
request = requests.get('https://{}.net/post/index.json?tags={}'.format(booru, ','.join(['order:random'] + tags)), headers=HEADERS).json()
if len(request) == 0: raise exc.NotFound(formatter.tostring(tags))
if len(request) < limit: limit = len(request)
for post in request: for post in request:
if 'swf' in post['file_ext'] or 'webm' in post['file_ext']: if 'swf' in post['file_ext'] or 'webm' in post['file_ext']: continue
continue
try: try:
for tag in blacklist: for tag in blacklist:
if tag in post['tags']: if tag in post['tags']: raise exc.Continue
raise exc.Continue except exc.Continue: continue
except exc.Continue: if post['file_url'] not in posts.values() and post['file_url'] not in previous: posts[post['id']] = post['file_url']
continue if len(posts) == limit: break
if post['file_url'] not in urls:
urls.append(post['file_url'])
if len(urls) == limit:
break
c += 1 c += 1
if c == 50 + limit: return posts
raise exc.Timeout
for url in urls:
await ctx.send('`' + formatter.tostring(args) + '`\n' + url)
# Umbrella command structure to manage global, channel, and user blacklists # Umbrella command structure to manage global, channel, and user blacklists
@commands.group(aliases=['bl', 'b'], brief='Manage blacklists', description='Blacklist base command for managing blacklists\n\n`bl get [blacklist]` to show a blacklist\n`bl set [blacklist] [tags]` to replace a blacklist\n`bl clear [blacklist]` to clear a blacklist\n`bl add [blacklist] [tags]` to add tags to a blacklist\n`bl remove [blacklist] [tags]` to remove tags from a blacklist', usage='[flag] [blacklist] ([tags])') @commands.group(aliases=['bl', 'b'], brief='Manage blacklists', description='Blacklist base command for managing blacklists\n\n`bl get [blacklist]` to show a blacklist\n`bl set [blacklist] [tags]` to replace a blacklist\n`bl clear [blacklist]` to clear a blacklist\n`bl add [blacklist] [tags]` to add tags to a blacklist\n`bl remove [blacklist] [tags]` to remove tags from a blacklist', usage='[flag] [blacklist] ([tags])')
@ -241,7 +309,7 @@ class MsG:
@_get_blacklist.command(name='channel', aliases=['ch', 'c']) @_get_blacklist.command(name='channel', aliases=['ch', 'c'])
async def __get_channel_blacklist(self, ctx): async def __get_channel_blacklist(self, ctx):
global blacklists global blacklists
if isinstance(ctx.message.guild, discord.Guild): if isinstance(ctx.message.guild, d.Guild):
guild = ctx.message.guild guild = ctx.message.guild
else: else:
guild = ctx.message.channel guild = ctx.message.channel
@ -255,7 +323,7 @@ class MsG:
@_get_blacklist.command(name='here', aliases=['h']) @_get_blacklist.command(name='here', aliases=['h'])
async def __get_here_blacklists(self, ctx): async def __get_here_blacklists(self, ctx):
global blacklists global blacklists
if isinstance(ctx.message.guild, discord.Guild): if isinstance(ctx.message.guild, d.Guild):
guild = ctx.message.guild guild = ctx.message.guild
else: else:
guild = ctx.message.channel guild = ctx.message.channel
@ -269,7 +337,7 @@ class MsG:
@commands.has_permissions(manage_channels=True) @commands.has_permissions(manage_channels=True)
async def ___get_all_guild_blacklists(self, ctx): async def ___get_all_guild_blacklists(self, ctx):
global blacklists global blacklists
if isinstance(ctx.message.guild, discord.Guild): if isinstance(ctx.message.guild, d.Guild):
guild = ctx.message.guild guild = ctx.message.guild
else: else:
guild = ctx.message.channel guild = ctx.message.channel
@ -287,14 +355,14 @@ class MsG:
@_add_tags.command(name='global', aliases=['gl', 'g']) @_add_tags.command(name='global', aliases=['gl', 'g'])
@commands.is_owner() @commands.is_owner()
async def __add_global_tags(self, ctx, *tags): async def __add_global_tags(self, ctx, *tags):
global blacklists, aliases, headers global blacklists, aliases, HEADERS
try: try:
for tag in tags: for tag in tags:
if tag in blacklists['global_blacklist']: if tag in blacklists['global_blacklist']:
raise exc.TagExists(tag) raise exc.TagExists(tag)
blacklists['global_blacklist'].extend(tags) blacklists['global_blacklist'].extend(tags)
for tag in tags: for tag in tags:
alias_request = requests.get('https://e621.net/tag_alias/index.json?aliased_to=' + tag + '&approved=true', headers=headers).json() alias_request = requests.get('https://e621.net/tag_alias/index.json?aliased_to=' + tag + '&approved=true', headers=HEADERS).json()
for dic in alias_request: for dic in alias_request:
aliases['global_blacklist'].setdefault(tag, []).append(dic['name']) aliases['global_blacklist'].setdefault(tag, []).append(dic['name'])
with open('blacklists.json', 'w') as outfile: with open('blacklists.json', 'w') as outfile:
@ -305,13 +373,13 @@ class MsG:
except exc.TagExists as e: except exc.TagExists as e:
await ctx.send('❌ `' + str(e) + '` **already in blacklist.**', delete_after=10) await ctx.send('❌ `' + str(e) + '` **already in blacklist.**', delete_after=10)
except Exception: except Exception:
await ctx.send(exc.base + '\n```python' + traceback.format_exc(limit=1) + '```') await ctx.send('{}\n```{}```'.format(exc.base, tb.format_exc(limit=1)))
traceback.print_exc(limit=1) tb.print_exc()
@_add_tags.command(name='channel', aliases=['ch', 'c']) @_add_tags.command(name='channel', aliases=['ch', 'c'])
@commands.has_permissions(manage_channels=True) @commands.has_permissions(manage_channels=True)
async def __add_channel_tags(self, ctx, *tags): async def __add_channel_tags(self, ctx, *tags):
global blacklists, aliases, headers global blacklists, aliases, HEADERS
if isinstance(ctx.message.guild, discord.Guild): if isinstance(ctx.message.guild, d.Guild):
guild = ctx.message.guild guild = ctx.message.guild
else: else:
guild = ctx.message.channel guild = ctx.message.channel
@ -322,7 +390,7 @@ class MsG:
raise exc.TagExists(tag) raise exc.TagExists(tag)
blacklists['guild_blacklist'].setdefault(str(guild.id), {}).setdefault(str(channel.id), []).extend(tags) blacklists['guild_blacklist'].setdefault(str(guild.id), {}).setdefault(str(channel.id), []).extend(tags)
for tag in tags: for tag in tags:
alias_request = requests.get('https://e621.net/tag_alias/index.json?aliased_to=' + tag + '&approved=true', headers=headers).json() alias_request = requests.get('https://e621.net/tag_alias/index.json?aliased_to=' + tag + '&approved=true', headers=HEADERS).json()
for dic in alias_request: for dic in alias_request:
aliases['guild_blacklist'].setdefault(str(guild.id), {}).setdefault(str(channel.id), {}).setdefault(tag, []).append(dic['name']) aliases['guild_blacklist'].setdefault(str(guild.id), {}).setdefault(str(channel.id), {}).setdefault(tag, []).append(dic['name'])
with open('blacklists.json', 'w') as outfile: with open('blacklists.json', 'w') as outfile:
@ -333,11 +401,11 @@ class MsG:
except exc.TagExists as e: except exc.TagExists as e:
await ctx.send('❌ `' + str(e) + '` **already in blacklist.**', delete_after=10) await ctx.send('❌ `' + str(e) + '` **already in blacklist.**', delete_after=10)
except Exception: except Exception:
await ctx.send(exc.base + '\n```python' + traceback.format_exc(limit=1) + '```') await ctx.send('{}\n```{}```'.format(exc.base, tb.format_exc(limit=1)))
traceback.print_exc(limit=1) tb.print_exc()
@_add_tags.command(name='me', aliases=['m']) @_add_tags.command(name='me', aliases=['m'])
async def __add_user_tags(self, ctx, *tags): async def __add_user_tags(self, ctx, *tags):
global blacklists, aliases, headers global blacklists, aliases, HEADERS
user = ctx.message.author user = ctx.message.author
try: try:
for tag in tags: for tag in tags:
@ -345,7 +413,7 @@ class MsG:
raise exc.TagExists(tag) raise exc.TagExists(tag)
blacklists['user_blacklist'].setdefault(str(user.id), []).extend(tags) blacklists['user_blacklist'].setdefault(str(user.id), []).extend(tags)
for tag in tags: for tag in tags:
alias_request = requests.get('https://e621.net/tag_alias/index.json?aliased_to=' + tag + '&approved=true', headers=headers).json() alias_request = requests.get('https://e621.net/tag_alias/index.json?aliased_to=' + tag + '&approved=true', headers=HEADERS).json()
for dic in alias_request: for dic in alias_request:
aliases['user_blacklist'].setdefault(str(user.id), {}).setdefault(tag, []).append(dic['name']) aliases['user_blacklist'].setdefault(str(user.id), {}).setdefault(tag, []).append(dic['name'])
with open('blacklists.json', 'w') as outfile: with open('blacklists.json', 'w') as outfile:
@ -356,8 +424,8 @@ class MsG:
except exc.TagExists as e: except exc.TagExists as e:
await ctx.send('❌ `' + str(e) + '` **already in blacklist.**', delete_after=10) await ctx.send('❌ `' + str(e) + '` **already in blacklist.**', delete_after=10)
except Exception: except Exception:
await ctx.send(exc.base + '\n```python' + traceback.format_exc(limit=1) + '```') await ctx.send('{}\n```{}```'.format(exc.base, tb.format_exc(limit=1)))
traceback.print_exc(limit=1) tb.print_exc()
@blacklist.group(name='remove', aliases=['rm', 'r']) @blacklist.group(name='remove', aliases=['rm', 'r'])
async def _remove_tags(self, ctx): async def _remove_tags(self, ctx):
@ -382,13 +450,13 @@ class MsG:
except exc.TagError as e: except exc.TagError as e:
await ctx.send('❌ `' + str(e) + '` **not in blacklist.**', delete_after=10) await ctx.send('❌ `' + str(e) + '` **not in blacklist.**', delete_after=10)
except Exception: except Exception:
await ctx.send(exc.base + '\n```python' + traceback.format_exc(limit=1) + '```') await ctx.send('{}\n```{}```'.format(exc.base, tb.format_exc(limit=1)))
traceback.print_exc(limit=1) tb.print_exc()
@_remove_tags.command(name='channel', aliases=['ch', 'c']) @_remove_tags.command(name='channel', aliases=['ch', 'c'])
@commands.has_permissions(manage_channels=True) @commands.has_permissions(manage_channels=True)
async def __remove_channel_tags(self, ctx, *tags): async def __remove_channel_tags(self, ctx, *tags):
global blacklists, aliases global blacklists, aliases
if isinstance(ctx.message.guild, discord.Guild): if isinstance(ctx.message.guild, d.Guild):
guild = ctx.message.guild guild = ctx.message.guild
else: else:
guild = ctx.message.channel guild = ctx.message.channel
@ -408,8 +476,8 @@ class MsG:
except exc.TagError as e: except exc.TagError as e:
await ctx.send('❌ `' + str(e) + '` **not in blacklist.**', delete_after=10) await ctx.send('❌ `' + str(e) + '` **not in blacklist.**', delete_after=10)
except Exception: except Exception:
await ctx.send(exc.base + '\n```python' + traceback.format_exc(limit=1) + '```') await ctx.send('{}\n```{}```'.format(exc.base, tb.format_exc(limit=1)))
traceback.print_exc(limit=1) tb.print_exc()
@_remove_tags.command(name='me', aliases=['m']) @_remove_tags.command(name='me', aliases=['m'])
async def __remove_user_tags(self, ctx, *tags): async def __remove_user_tags(self, ctx, *tags):
global blacklists, aliases global blacklists, aliases
@ -429,8 +497,8 @@ class MsG:
except exc.TagError as e: except exc.TagError as e:
await ctx.send('❌ `' + str(e) + '` **not in blacklist.**', delete_after=10) await ctx.send('❌ `' + str(e) + '` **not in blacklist.**', delete_after=10)
except Exception: except Exception:
await ctx.send(exc.base + '\n```python' + traceback.format_exc(limit=1) + '```') await ctx.send('{}\n```{}```'.format(exc.base, tb.format_exc(limit=1)))
traceback.print_exc(limit=1) tb.print_exc()
@blacklist.group(name='clear', aliases=['cl', 'c']) @blacklist.group(name='clear', aliases=['cl', 'c'])
async def _clear_blacklist(self, ctx): async def _clear_blacklist(self, ctx):
@ -440,8 +508,8 @@ class MsG:
@commands.is_owner() @commands.is_owner()
async def __clear_global_blacklist(self, ctx): async def __clear_global_blacklist(self, ctx):
global blacklists, aliases global blacklists, aliases
blacklists['global_blacklist'] = [] del blacklists['global_blacklist']
aliases['global_blacklist'] = {} del aliases['global_blacklist']
with open('blacklists.json', 'w') as outfile: with open('blacklists.json', 'w') as outfile:
json.dump(blacklists, outfile, indent=4, sort_keys=True) json.dump(blacklists, outfile, indent=4, sort_keys=True)
with open('aliases.json', 'w') as outfile: with open('aliases.json', 'w') as outfile:
@ -451,13 +519,13 @@ class MsG:
@commands.has_permissions(manage_channels=True) @commands.has_permissions(manage_channels=True)
async def __clear_channel_blacklist(self, ctx): async def __clear_channel_blacklist(self, ctx):
global blacklists, aliases global blacklists, aliases
if isinstance(ctx.message.guild, discord.Guild): if isinstance(ctx.message.guild, d.Guild):
guild = ctx.message.guild guild = ctx.message.guild
else: else:
guild = ctx.message.channel guild = ctx.message.channel
channel = ctx.message.channel channel = ctx.message.channel
blacklists['guild_blacklist'][str(guild.id)][str(channel.id)] = [] del blacklists['guild_blacklist'][str(guild.id)][str(channel.id)]
aliases['guild_blacklist'][str(guild.id)][str(channel.id)] = {} del aliases['guild_blacklist'][str(guild.id)][str(channel.id)]
with open('blacklists.json', 'w') as outfile: with open('blacklists.json', 'w') as outfile:
json.dump(blacklists, outfile, indent=4, sort_keys=True) json.dump(blacklists, outfile, indent=4, sort_keys=True)
with open('aliases.json', 'w') as outfile: with open('aliases.json', 'w') as outfile:
@ -467,8 +535,8 @@ class MsG:
async def __clear_user_blacklist(self, ctx): async def __clear_user_blacklist(self, ctx):
global blacklists, aliases global blacklists, aliases
user = ctx.message.author user = ctx.message.author
blacklists['user_blacklist'][str(user.id)] = [] del blacklists['user_blacklist'][str(user.id)]
aliases['user_blacklist'][str(user.id)] = {} del aliases['user_blacklist'][str(user.id)]
with open('blacklists.json', 'w') as outfile: with open('blacklists.json', 'w') as outfile:
json.dump(blacklists, outfile, indent=4, sort_keys=True) json.dump(blacklists, outfile, indent=4, sort_keys=True)
with open('aliases.json', 'w') as outfile: with open('aliases.json', 'w') as outfile: