Apply black formatting

This commit is contained in:
TSR Berry 2022-11-10 15:09:57 +01:00
parent b2bd3142a9
commit 3418c206d9
No known key found for this signature in database
GPG key ID: 52353C0A4CCA15E2
4 changed files with 46 additions and 27 deletions

View file

@ -262,7 +262,7 @@ class LogFileReader(Cog):
setting_value = [ setting_value = [
line.split()[-1] line.split()[-1]
for line in log_file.splitlines() for line in log_file.splitlines()
if re.search(fr"LogValueChange: ({setting_string})\s", line) if re.search(rf"LogValueChange: ({setting_string})\s", line)
][-1] ][-1]
if setting_value and setting.get(name): if setting_value and setting.get(name):
setting[name] = setting_value setting[name] = setting_value

View file

@ -8,6 +8,7 @@ from discord.ext import commands
from discord.ext.commands import Cog from discord.ext.commands import Cog
from helpers.checks import check_if_staff from helpers.checks import check_if_staff
class RyujinxReactionRoles(Cog): class RyujinxReactionRoles(Cog):
def __init__(self, bot): def __init__(self, bot):
self.bot = bot self.bot = bot
@ -26,7 +27,7 @@ class RyujinxReactionRoles(Cog):
async def register_reaction_role(self, ctx, target_role_id: int, emoji_name: str): async def register_reaction_role(self, ctx, target_role_id: int, emoji_name: str):
"""Register a reaction role, staff only.""" """Register a reaction role, staff only."""
if emoji_name[0] == '<': if emoji_name[0] == "<":
emoji_name = emoji_name[1:-1] emoji_name = emoji_name[1:-1]
if target_role_id in config.staff_role_ids: if target_role_id in config.staff_role_ids:
@ -73,18 +74,22 @@ class RyujinxReactionRoles(Cog):
async def generate_embed(self): async def generate_embed(self):
last_descrption = [] last_descrption = []
description = ["React to this message with the emojis given below to get your 'Looking for LDN game' roles. \n"] description = [
"React to this message with the emojis given below to get your 'Looking for LDN game' roles. \n"
]
for x in self.emoji_map: for x in self.emoji_map:
value = self.emoji_map[x] value = self.emoji_map[x]
emoji = x emoji = x
if len(emoji.split(':')) == 3: if len(emoji.split(":")) == 3:
emoji = f"<{emoji}>" emoji = f"<{emoji}>"
if type(value) is str: if type(value) is str:
description.append(f"{emoji} for __{self.emoji_map.get(x).split('(')[1].split(')')[0]}__") description.append(
f"{emoji} for __{self.emoji_map.get(x).split('(')[1].split(')')[0]}__"
)
else: else:
role_name = value["role"] role_name = value["role"]
line_fmt = value["fmt"] line_fmt = value["fmt"]
@ -94,7 +99,9 @@ class RyujinxReactionRoles(Cog):
description.append(line_fmt.format(emoji, role_name)) description.append(line_fmt.format(emoji, role_name))
embed = discord.Embed( embed = discord.Embed(
title="**Select your roles**", description='\n'.join(description) + '\n' + '\n'.join(last_descrption), color=420420 title="**Select your roles**",
description="\n".join(description) + "\n" + "\n".join(last_descrption),
color=420420,
) )
embed.set_footer( embed.set_footer(
text="To remove a role, simply remove the corresponding reaction." text="To remove a role, simply remove the corresponding reaction."
@ -106,7 +113,7 @@ class RyujinxReactionRoles(Cog):
for reaction in self.m.reactions: for reaction in self.m.reactions:
for user in await reaction.users().flatten(): for user in await reaction.users().flatten():
emoji_name = str(reaction.emoji) emoji_name = str(reaction.emoji)
if emoji_name[0] == '<': if emoji_name[0] == "<":
emoji_name = emoji_name[1:-1] emoji_name = emoji_name[1:-1]
if self.get_role_from_emoji(emoji_name) is not None: if self.get_role_from_emoji(emoji_name) is not None:
@ -120,7 +127,7 @@ class RyujinxReactionRoles(Cog):
for emoji in self.emoji_map: for emoji in self.emoji_map:
for reaction in self.m.reactions: for reaction in self.m.reactions:
emoji_name = str(reaction.emoji) emoji_name = str(reaction.emoji)
if emoji_name[0] == '<': if emoji_name[0] == "<":
emoji_name = emoji_name[1:-1] emoji_name = emoji_name[1:-1]
role = self.get_role(emoji_name) role = self.get_role(emoji_name)
@ -141,14 +148,20 @@ class RyujinxReactionRoles(Cog):
with open(self.file, "w") as f: with open(self.file, "w") as f:
json.dump(value, f) json.dump(value, f)
async def reload_reaction_message(self, should_handle_offline = True): async def reload_reaction_message(self, should_handle_offline=True):
self.emoji_map = collections.OrderedDict(sorted(self.reaction_config["reaction_roles_emoji_map"].items(), key=lambda x: str(x[1]))) self.emoji_map = collections.OrderedDict(
sorted(
self.reaction_config["reaction_roles_emoji_map"].items(),
key=lambda x: str(x[1]),
)
)
guild = self.bot.guilds[0] # The ryu guild in which the bot is. guild = self.bot.guilds[0] # The ryu guild in which the bot is.
channel = guild.get_channel(self.channel_id) channel = guild.get_channel(self.channel_id)
m = discord.utils.get(await channel.history().flatten(), id=self.reaction_config["id"]) m = discord.utils.get(
await channel.history().flatten(), id=self.reaction_config["id"]
)
if m is None: if m is None:
self.reaction_config["id"] = None self.reaction_config["id"] = None
@ -165,7 +178,9 @@ class RyujinxReactionRoles(Cog):
await self.handle_offline_reaction_remove() await self.handle_offline_reaction_remove()
else: else:
self.m = discord.utils.get(await channel.history().flatten(), id=self.reaction_config["id"]) self.m = discord.utils.get(
await channel.history().flatten(), id=self.reaction_config["id"]
)
self.msg_id = self.m.id self.msg_id = self.m.id
await self.m.edit(embed=await self.generate_embed()) await self.m.edit(embed=await self.generate_embed())
@ -195,11 +210,11 @@ class RyujinxReactionRoles(Cog):
target_role = self.get_role(emoji_name) target_role = self.get_role(emoji_name)
if target_role is not None: if target_role is not None:
await payload.member.add_roles( await payload.member.add_roles(target_role)
target_role
)
else: else:
self.bot.log.error(f"Role {self.emoji_map[emoji_name]} not found.") self.bot.log.error(
f"Role {self.emoji_map[emoji_name]} not found."
)
await self.m.clear_reaction(payload.emoji) await self.m.clear_reaction(payload.emoji)
else: else:
await self.m.clear_reaction(payload.emoji) await self.m.clear_reaction(payload.emoji)

View file

@ -18,15 +18,17 @@ class RyujinxVerification(Cog):
async def on_member_join(self, member): async def on_member_join(self, member):
await self.bot.wait_until_ready() await self.bot.wait_until_ready()
if (member.guild.id not in config.guild_whitelist): if member.guild.id not in config.guild_whitelist:
return return
join_channel = self.bot.get_channel(config.welcome_channel) join_channel = self.bot.get_channel(config.welcome_channel)
if join_channel is not None: if join_channel is not None:
await join_channel.send( await join_channel.send(
'Hello {0.mention}! Welcome to Ryujinx! Please read the <#411271165429022730>, and then type the verifying command here to gain access to the rest of the channels.\n\nIf you need help with basic common questions, visit the <#585288848704143371> channel after joining.\n\nIf you need help with Animal Crossing visit the <#692104087889641472> channel for common issues and solutions. If you need help that is not Animal Crossing related, please visit the <#410208610455519243> channel after verifying.'.format( "Hello {0.mention}! Welcome to Ryujinx! Please read the <#411271165429022730>, and then type the verifying command here to gain access to the rest of the channels.\n\nIf you need help with basic common questions, visit the <#585288848704143371> channel after joining.\n\nIf you need help with Animal Crossing visit the <#692104087889641472> channel for common issues and solutions. If you need help that is not Animal Crossing related, please visit the <#410208610455519243> channel after verifying.".format(
member)) member
)
)
async def process_message(self, message): async def process_message(self, message):
"""Process the verification process""" """Process the verification process"""
@ -67,7 +69,7 @@ class RyujinxVerification(Cog):
async def on_member_join(self, member): async def on_member_join(self, member):
await self.bot.wait_until_ready() await self.bot.wait_until_ready()
if (member.guild.id not in config.guild_whitelist): if member.guild.id not in config.guild_whitelist:
return return
join_channel = self.bot.get_channel(config.welcome_channel) join_channel = self.bot.get_channel(config.welcome_channel)
@ -80,8 +82,10 @@ class RyujinxVerification(Cog):
async def reset(self, ctx, limit: int = 100, force: bool = False): async def reset(self, ctx, limit: int = 100, force: bool = False):
"""Wipes messages and pastes the welcome message again. Staff only.""" """Wipes messages and pastes the welcome message again. Staff only."""
if ctx.message.channel.id != config.welcome_channel and not force: if ctx.message.channel.id != config.welcome_channel and not force:
await ctx.send(f"This command is limited to" await ctx.send(
f" <#{config.welcome_channel}>, unless forced.") f"This command is limited to"
f" <#{config.welcome_channel}>, unless forced."
)
return return
await self.do_reset(ctx.channel, ctx.author.mention, limit) await self.do_reset(ctx.channel, ctx.author.mention, limit)