refactoring

This commit is contained in:
klemek
2021-01-08 16:31:38 +01:00
parent ea0febfb3c
commit 7702e36fa0
5 changed files with 19 additions and 13 deletions
+7 -3
View File
@@ -5,6 +5,7 @@ import discord
# Custom libs
from .scanner import Scanner
from data_types import Frequency
from logs import ChannelLogs, MessageLog
@@ -25,11 +26,12 @@ class FrequencyScanner(Scanner):
intro_context="frequency",
)
async def compute_args(self, message: discord.Message, *args: str) -> bool:
async def init(self, message: discord.Message, *args: str) -> bool:
self.freq = Frequency()
return True
def compute_message(self, channel: ChannelLogs, message: MessageLog):
return FrequencyScanner.analyse_message(message, self.raw_members)
return FrequencyScanner.analyse_message(message, self.freq, self.raw_members)
def get_results(self, intro: str) -> List[str]:
res = [intro]
@@ -37,7 +39,9 @@ class FrequencyScanner(Scanner):
return res
@staticmethod
def analyse_message(message: MessageLog, raw_members: List[int]) -> bool:
def analyse_message(
message: MessageLog, freq: Frequency, raw_members: List[int]
) -> bool:
impacted = False
# If author is included in the selection (empty list is all)
if not message.bot and (len(raw_members) == 0 or message.author in raw_members):