From 6e21b7bb67a0130e2828c30bc8a1075f2e4d68cd Mon Sep 17 00:00:00 2001 From: Zevaryx Date: Fri, 2 Jul 2021 13:27:59 -0600 Subject: [PATCH] Fix on_message to not execute on DMs --- jarvis/__init__.py | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) diff --git a/jarvis/__init__.py b/jarvis/__init__.py index 004f272..99b86e7 100644 --- a/jarvis/__init__.py +++ b/jarvis/__init__.py @@ -1,7 +1,7 @@ from pathlib import Path import pymongo from datetime import datetime, timedelta -from discord import Intents, Member, Message +from discord import Intents, Member, Message, DMChannel from discord.ext import commands from discord.ext.tasks import loop from discord.utils import find, get @@ -70,13 +70,14 @@ async def on_member_join(user: Member): @jarvis.event async def on_message(message: Message): - db = DBManager(get_config().mongo).mongo - autoreact = db.jarvis.autoreact.find_one( - {"guild": message.guild.id, "channel": message.channel.id} - ) - if autoreact: - for reaction in autoreact["reactions"]: - await message.add_reaction(reaction) + if not isinstance(message.channel, DMChannel): + db = DBManager(get_config().mongo).mongo + autoreact = db.jarvis.autoreact.find_one( + {"guild": message.guild.id, "channel": message.channel.id} + ) + if autoreact: + for reaction in autoreact["reactions"]: + await message.add_reaction(reaction) await jarvis.process_commands(message)