Klemek
104 lines
3.1 KiB
Python
104 lines
3.1 KiB
Python
from abc import ABC, abstractmethod
|
|
from typing import List, Tuple, Optional
|
|
import discord
|
|
import re
|
|
|
|
# Custom libs
|
|
|
|
from .scanner import Scanner
|
|
from data_types import History
|
|
from logs import ChannelLogs, MessageLog
|
|
from utils import FilterLevel
|
|
|
|
|
|
class HistoryScanner(Scanner, ABC):
|
|
def __init__(self, *, help: str):
|
|
super().__init__(
|
|
has_digit_args=True,
|
|
valid_args=[
|
|
"all",
|
|
"everyone",
|
|
"spoiler",
|
|
"spoiler:allow",
|
|
"spoiler:only",
|
|
"image",
|
|
"img",
|
|
"gif",
|
|
],
|
|
help=help,
|
|
intro_context="",
|
|
all_args=True,
|
|
)
|
|
|
|
async def init(self, message: discord.Message, *args: str) -> bool:
|
|
self.history = History()
|
|
self.all_messages = "all" in args or "everyone" in args
|
|
self.images_only = "image" in args or "img" in args or "gif" in args
|
|
self.gif_only = "gif" in args
|
|
if "spoiler" in args or "spoiler:allow" in args:
|
|
self.spoiler = FilterLevel.ALLOW
|
|
elif "spoiler:only" in args:
|
|
self.spoiler = FilterLevel.ONLY
|
|
else:
|
|
self.spoiler = FilterLevel.NONE
|
|
if not self.images_only:
|
|
self.queries = [
|
|
(
|
|
query.lower(),
|
|
query.strip("`") if re.match(r"^`.*`$", query) else None,
|
|
)
|
|
for query in self.other_args
|
|
]
|
|
else:
|
|
self.queries = []
|
|
return True
|
|
|
|
def compute_message(self, channel: ChannelLogs, message: MessageLog):
|
|
return HistoryScanner.analyse_message(
|
|
channel,
|
|
message,
|
|
self.history,
|
|
self.raw_members,
|
|
all_messages=self.all_messages,
|
|
queries=self.queries,
|
|
images_only=self.images_only,
|
|
)
|
|
|
|
@abstractmethod
|
|
def get_results(self, intro: str):
|
|
pass
|
|
|
|
@staticmethod
|
|
def analyse_message(
|
|
channel: ChannelLogs,
|
|
message: MessageLog,
|
|
history: History,
|
|
raw_members: List[int],
|
|
*,
|
|
all_messages: bool,
|
|
queries: List[Tuple[str, Optional[str]]],
|
|
images_only: bool,
|
|
) -> bool:
|
|
impacted = False
|
|
# If author is included in the selection (empty list is all)
|
|
if (
|
|
(
|
|
(not message.bot or all_messages)
|
|
and len(raw_members) == 0
|
|
or message.author in raw_members
|
|
)
|
|
and (message.content or message.attachment)
|
|
and (not images_only or message.image)
|
|
):
|
|
if not images_only:
|
|
content = message.content.lower()
|
|
for query in queries:
|
|
if query[1] is not None:
|
|
if not re.match(query[1], message.content):
|
|
return False
|
|
elif not query[0] in content:
|
|
return False
|
|
impacted = True
|
|
history.messages += [message]
|
|
return impacted
|