escape text in find scanner

This commit is contained in:
Klemek
2021-04-22 16:12:13 +02:00
parent 3f7abd9a15
commit d2cdea3db6
2 changed files with 8 additions and 3 deletions
+4 -2
View File
@@ -1,7 +1,6 @@
from typing import Dict, List
from collections import defaultdict
import discord
import re
# Custom libs
@@ -13,6 +12,7 @@ from utils import (
plural,
precise,
mention,
escape_text,
)
@@ -64,7 +64,9 @@ class FindScanner(Scanner):
res += [
self.matches[match].to_string(
matches.index(match),
f"`{match}`",
f'"{escape_text(match)}"'
if not match.startswith("<:")
else escape_text(match),
total_usage=self.msg_count,
ranking=False,
transform=lambda id: f" by {mention(id)}",
+4 -1
View File
@@ -1,4 +1,3 @@
from calendar import month
from typing import Callable, List, Dict, Union, Optional, Any
import os
import logging
@@ -81,6 +80,10 @@ def message_link(message: discord.Message) -> str:
return f"https://discord.com/channels/{message.channel.guild.id}/{message.channel.id}/{message.id}"
def escape_text(text: str) -> str:
return discord.utils.escape_markdown(discord.utils.escape_mentions(text))
class FakeMessage:
def __init__(self, id: int):
self.id = id