Remove guild event handler, fix other handlers

This commit is contained in:
Zeva Rose 2022-02-04 09:47:56 -07:00
parent 61e4d4d497
commit 490c8fc142
3 changed files with 15 additions and 48 deletions

View file

@ -1,36 +0,0 @@
"""J.A.R.V.I.S. guild event handler."""
import asyncio
from dis_snek import Snake
from dis_snek.models.discord.guild import Guild
from jarvis.db.models import Setting
from jarvis.utils import find
class GuildEventHandler(object):
"""J.A.R.V.I.S. guild event handler."""
def __init__(self, bot: Snake):
self.bot = bot
self.bot.add_listener(self.on_guild_join)
async def on_guild_join(self, guild: Guild) -> None:
"""Handle on_guild_join event."""
general = find(lambda x: x.name == "general", guild.channels)
if general and general.permissions_for(guild.me).send_messages:
user = self.bot.user
await general.send(
f"Allow me to introduce myself. I am {user.mention}, a virtual "
"artificial intelligence, and I'm here to assist you with a "
"variety of tasks as best I can, "
"24 hours a day, seven days a week."
)
await asyncio.sleep(1)
await general.send("Importing all preferences from home interface...")
# Set some default settings
_ = Setting(guild=guild.id, setting="massmention", value=5).save()
_ = Setting(guild=guild.id, setting="noinvite", value=True).save()
await general.send("Systems are now fully operational")

View file

@ -1,5 +1,5 @@
"""J.A.R.V.I.S. Member event handler."""
from dis_snek import Snake
from dis_snek import Snake, listen
from dis_snek.models.discord.user import Member
from jarvis.db.models import Mute, Setting
@ -12,6 +12,7 @@ class MemberEventHandler(object):
self.bot = bot
self.bot.add_listener(self.on_member_join)
@listen()
async def on_member_join(self, user: Member) -> None:
"""Handle on_member_join event."""
guild = user.guild

View file

@ -1,14 +1,14 @@
"""J.A.R.V.I.S. Message event handler."""
import re
from dis_snek import Snake
from dis_snek import Snake, listen
from dis_snek.models.discord.channel import DMChannel
from dis_snek.models.discord.embed import EmbedField
from dis_snek.models.discord.message import Message
from jarvis.config import get_config
from jarvis.db.models import Autopurge, Autoreact, Roleping, Setting, Warning
from jarvis.utils import build_embed, find
from jarvis.utils.field import Field
invites = re.compile(
r"(?:https?://)?(?:www.)?(?:discord.(?:gg|io|me|li)|discord(?:app)?.com/invite)/([^\s/]+?)(?=\b)", # noqa: E501
@ -72,10 +72,10 @@ class MessageEventHandler(object):
user=message.author.id,
).save()
fields = [
Field(
"Reason",
"Sent an invite link",
False,
EmbedField(
name="Reason",
value="Sent an invite link",
inline=False,
)
]
embed = build_embed(
@ -113,7 +113,7 @@ class MessageEventHandler(object):
reason="Mass Mention",
user=message.author.id,
).save()
fields = [Field("Reason", "Mass Mention", False)]
fields = [EmbedField(name="Reason", value="Mass Mention", inline=False)]
embed = build_embed(
title="Warning",
description=f"{message.author.mention} has been warned",
@ -178,10 +178,10 @@ class MessageEventHandler(object):
user=message.author.id,
).save()
fields = [
Field(
"Reason",
"Pinged a blocked role/user with a blocked role",
False,
EmbedField(
name="Reason",
value="Pinged a blocked role/user with a blocked role",
inline=False,
)
]
embed = build_embed(
@ -198,6 +198,7 @@ class MessageEventHandler(object):
)
await message.channel.send(embed=embed)
@listen()
async def on_message(self, message: Message) -> None:
"""Handle on_message event. Calls other event handlers."""
if not isinstance(message.channel, DMChannel) and not message.author.bot:
@ -207,6 +208,7 @@ class MessageEventHandler(object):
await self.autopurge(message)
await self.checks(message)
@listen()
async def on_message_edit(self, before: Message, after: Message) -> None:
"""Handle on_message_edit event. Calls other event handlers."""
if not isinstance(after.channel, DMChannel) and not after.author.bot: