flifloo/Administrator-py
Archived
1
0
Fork 0
Code Activity

Add support of disabled extension for invite

Browse source
This commit is contained in:
Ethanell 2020-11-05 14:47:11 +01:00
parent 4c114aef6e
commit a10ff7daf0
1 changed files with 9 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
extensions/invite.py
View file

@ -6,7 +6,9 @@ from discord.ext import commands
from discord.ext.commands import BadArgument from discord.ext.commands import BadArgument
import db import db
from administrator.check import is_enabled
from administrator.logger import logger from administrator.logger import logger
from administrator.utils import event_is_enabled
extension_name = "invite" extension_name = "invite"
logger = logger.getChild(extension_name) logger = logger.getChild(extension_name)
@ -24,6 +26,7 @@ class Invite(commands.Cog):
return "Get role from a special invite link" return "Get role from a special invite link"
@commands.group("invite", pass_context=True) @commands.group("invite", pass_context=True)
@is_enabled()
@commands.guild_only() @commands.guild_only()
@commands.has_guild_permissions(administrator=True) @commands.has_guild_permissions(administrator=True)
async def invite(self, ctx: commands.Context): async def invite(self, ctx: commands.Context):
@ -77,6 +80,8 @@ class Invite(commands.Cog):
@commands.Cog.listener() @commands.Cog.listener()
async def on_member_join(self, member: Member): async def on_member_join(self, member: Member):
if not event_is_enabled(self.qualified_name, member.guild.id):
return
user_invites = await member.guild.invites() user_invites = await member.guild.invites()
for i in self.invites[member.guild.id]: for i in self.invites[member.guild.id]:
for ui in user_invites: for ui in user_invites:
@ -93,11 +98,15 @@ class Invite(commands.Cog):
@commands.Cog.listener() @commands.Cog.listener()
async def on_invite_create(self, invite): async def on_invite_create(self, invite):
if not event_is_enabled(self.qualified_name, invite.guild.id):
return
self.invites[invite.guild.id] = await invite.guild.invites() self.invites[invite.guild.id] = await invite.guild.invites()
@commands.Cog.listener() @commands.Cog.listener()
async def on_invite_delete(self, invite): async def on_invite_delete(self, invite):
s = db.Session() s = db.Session()
if not event_is_enabled(self.qualified_name, invite.guild.id, s):
return
invite_role = s.query(db.InviteRole).get({"guild_id": invite.guild.id, "invite_code": invite.code}) invite_role = s.query(db.InviteRole).get({"guild_id": invite.guild.id, "invite_code": invite.code})
if invite_role: if invite_role:
s.delete(invite_role) s.delete(invite_role)