refactoring
This commit is contained in:
@@ -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):
|
||||
|
||||
Reference in New Issue
Block a user