From 8576c7473069b3433d64cc6d6ac34ef8c7b4f776 Mon Sep 17 00:00:00 2001 From: klemek Date: Thu, 14 Jan 2021 18:59:12 +0100 Subject: [PATCH] take members args in account --- src/scanners/channels_scanner.py | 6 +++++- src/scanners/messages_scanner.py | 6 +++++- 2 files changed, 10 insertions(+), 2 deletions(-) diff --git a/src/scanners/channels_scanner.py b/src/scanners/channels_scanner.py index 1b29542..6211221 100644 --- a/src/scanners/channels_scanner.py +++ b/src/scanners/channels_scanner.py @@ -81,7 +81,11 @@ class ChannelsScanner(Scanner): all_messages: bool, ) -> bool: impacted = False - if not message.bot or all_messages: + if ( + len(raw_members) == 0 + and (not message.bot or all_messages) + or message.author in raw_members + ): impacted = True messages[channel_id].update_use(1, message.created_at, message.author) return impacted diff --git a/src/scanners/messages_scanner.py b/src/scanners/messages_scanner.py index 7e05cb9..c04e825 100644 --- a/src/scanners/messages_scanner.py +++ b/src/scanners/messages_scanner.py @@ -81,7 +81,11 @@ class MessagesScanner(Scanner): all_messages: bool, ) -> bool: impacted = False - if not message.bot or all_messages: + if ( + len(raw_members) == 0 + and (not message.bot or all_messages) + or message.author in raw_members + ): impacted = True messages[message.author].update_use(1, message.created_at, channel_id) return impacted