ryujinx-archive/ryuko-ng
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

lockdown: reduce code repetition

Browse source
This commit is contained in:
Ave Ozkal 2019-02-21 01:20:03 +03:00
parent f2e210b9e3
commit 807dd882af
No known key found for this signature in database
GPG key ID: 09356ABAA42C842B
1 changed files with 15 additions and 21 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

36
cogs/lockdown.py
View file

@ -8,17 +8,21 @@ class Lockdown:
def __init__(self, bot):
self.bot = bot
async def set_sendmessgae(self, channel: discord.TextChannel,
role, allow_send, issuer):
try:
roleobj = channel.guild.get_role(role)
overrides = channel.overwrites_for(roleobj)
overrides.send_messages = allow_send
await channel.set_permissions(roleobj,
overwrite=overrides,
reason=str(issuer))
except:
pass
async def unlock_for_staff(self, channel: discord.TextChannel, issuer):
for role in config.staff_role_ids:
try:
roleobj = channel.guild.get_role(role)
overrides = channel.overwrites_for(roleobj)
overrides.send_messages = True
await channel.set_permissions(roleobj,
overwrite=overrides,
reason=str(issuer))
except:
pass
await self.set_sendmessage(channel, role, True, issuer)
@commands.guild_only()
@commands.check(check_if_staff)
@ -39,12 +43,7 @@ class Lockdown:
roles = [config.named_roles["participant"]]
for role in roles:
roleobj = channel.guild.get_role(role)
overrides = channel.overwrites_for(roleobj)
overrides.send_messages = False
await channel.set_permissions(roleobj,
overwrite=overrides,
reason=str(ctx.author))
await self.set_sendmessage(channel, role, False, ctx.author)
await self.unlock_for_staff(channel, ctx.author)
@ -78,12 +77,7 @@ class Lockdown:
await self.unlock_for_staff(channel, ctx.author)
for role in roles:
roleobj = channel.guild.get_role(role)
overrides = channel.overwrites_for(roleobj)
overrides.send_messages = True
await channel.set_permissions(roleobj,
overwrite=overrides,
reason=str(ctx.author))
await self.set_sendmessage(channel, role, True, ctx.author)
safe_name = await commands.clean_content().convert(ctx, str(ctx.author))
await ctx.send("🔓 Channel unlocked.")