Add iOS request detection
This commit is contained in:
JelNiSław
parent
9b110f0137
commit
065b1b6d7e
6 changed files with 109 additions and 15 deletions
|
|
@ -1 +1 @@
|
|||
DISCORD_TOKEN=""
|
||||
DISCORD_TOKEN=
|
||||
|
|
|
|||
24
poetry.lock
generated
24
poetry.lock
generated
|
|
@ -85,7 +85,7 @@ unicode_backport = ["unicodedata2"]
|
|||
|
||||
[[package]]
|
||||
name = "click"
|
||||
version = "8.1.0"
|
||||
version = "8.1.2"
|
||||
description = "Composable command line interface toolkit"
|
||||
category = "dev"
|
||||
optional = false
|
||||
|
|
@ -104,7 +104,7 @@ python-versions = ">=2.7, !=3.0.*, !=3.1.*, !=3.2.*, !=3.3.*, !=3.4.*"
|
|||
|
||||
[[package]]
|
||||
name = "discord.py"
|
||||
version = "2.0.0a3989+ge5153780"
|
||||
version = "2.0.0a4171+g5c2431d5"
|
||||
description = "A Python wrapper for the Discord API"
|
||||
category = "main"
|
||||
optional = false
|
||||
|
|
@ -123,8 +123,8 @@ voice = ["PyNaCl (>=1.3.0,<1.6)"]
|
|||
[package.source]
|
||||
type = "git"
|
||||
url = "https://github.com/Rapptz/discord.py"
|
||||
reference = "HEAD"
|
||||
resolved_reference = "e515378076c64c5ab2f3a06237f2362ac8556290"
|
||||
reference = "master"
|
||||
resolved_reference = "5c2431d5435d8a921a83cc05de29968f22a777ef"
|
||||
|
||||
[[package]]
|
||||
name = "flake8"
|
||||
|
|
@ -203,15 +203,15 @@ python-versions = "!=3.0.*,!=3.1.*,!=3.2.*,!=3.3.*,!=3.4.*,>=2.7"
|
|||
|
||||
[[package]]
|
||||
name = "platformdirs"
|
||||
version = "2.5.1"
|
||||
version = "2.5.2"
|
||||
description = "A small Python module for determining appropriate platform-specific dirs, e.g. a \"user data dir\"."
|
||||
category = "dev"
|
||||
optional = false
|
||||
python-versions = ">=3.7"
|
||||
|
||||
[package.extras]
|
||||
docs = ["Sphinx (>=4)", "furo (>=2021.7.5b38)", "proselint (>=0.10.2)", "sphinx-autodoc-typehints (>=1.12)"]
|
||||
test = ["appdirs (==1.4.4)", "pytest (>=6)", "pytest-cov (>=2.7)", "pytest-mock (>=3.6)"]
|
||||
docs = ["furo (>=2021.7.5b38)", "proselint (>=0.10.2)", "sphinx-autodoc-typehints (>=1.12)", "sphinx (>=4)"]
|
||||
test = ["appdirs (==1.4.4)", "pytest-cov (>=2.7)", "pytest-mock (>=3.6)", "pytest (>=6)"]
|
||||
|
||||
[[package]]
|
||||
name = "pycodestyle"
|
||||
|
|
@ -263,7 +263,7 @@ multidict = ">=4.0"
|
|||
[metadata]
|
||||
lock-version = "1.1"
|
||||
python-versions = "^3.10"
|
||||
content-hash = "41ef34b1d795becacbfdd1efcbb925d2ccf5d89029cfad6f6087d2050ac2a905"
|
||||
content-hash = "e02794cfd8d7eed71d5c9a704f5b37186bfe918a6ac9f17f612b8d952dc7195b"
|
||||
|
||||
[metadata.files]
|
||||
aiohttp = [
|
||||
|
|
@ -382,8 +382,8 @@ charset-normalizer = [
|
|||
{file = "charset_normalizer-2.0.12-py3-none-any.whl", hash = "sha256:6881edbebdb17b39b4eaaa821b438bf6eddffb4468cf344f09f89def34a8b1df"},
|
||||
]
|
||||
click = [
|
||||
{file = "click-8.1.0-py3-none-any.whl", hash = "sha256:19a4baa64da924c5e0cd889aba8e947f280309f1a2ce0947a3e3a7bcb7cc72d6"},
|
||||
{file = "click-8.1.0.tar.gz", hash = "sha256:977c213473c7665d3aa092b41ff12063227751c41d7b17165013e10069cc5cd2"},
|
||||
{file = "click-8.1.2-py3-none-any.whl", hash = "sha256:24e1a4a9ec5bf6299411369b208c1df2188d9eb8d916302fe6bf03faed227f1e"},
|
||||
{file = "click-8.1.2.tar.gz", hash = "sha256:479707fe14d9ec9a0757618b7a100a0ae4c4e236fac5b7f80ca68028141a1a72"},
|
||||
]
|
||||
colorama = [
|
||||
{file = "colorama-0.4.4-py2.py3-none-any.whl", hash = "sha256:9f47eda37229f68eee03b24b9748937c7dc3868f906e8ba69fbcbdd3bc5dc3e2"},
|
||||
|
|
@ -537,8 +537,8 @@ pathspec = [
|
|||
{file = "pathspec-0.9.0.tar.gz", hash = "sha256:e564499435a2673d586f6b2130bb5b95f04a3ba06f81b8f895b651a3c76aabb1"},
|
||||
]
|
||||
platformdirs = [
|
||||
{file = "platformdirs-2.5.1-py3-none-any.whl", hash = "sha256:bcae7cab893c2d310a711b70b24efb93334febe65f8de776ee320b517471e227"},
|
||||
{file = "platformdirs-2.5.1.tar.gz", hash = "sha256:7535e70dfa32e84d4b34996ea99c5e432fa29a708d0f4e394bbcb2a8faa4f16d"},
|
||||
{file = "platformdirs-2.5.2-py3-none-any.whl", hash = "sha256:027d8e83a2d7de06bbac4e5ef7e023c02b863d7ea5d079477e722bb41ab25788"},
|
||||
{file = "platformdirs-2.5.2.tar.gz", hash = "sha256:58c8abb07dcb441e6ee4b11d8df0ac856038f944ab98b7be6b27b2a3c7feef19"},
|
||||
]
|
||||
pycodestyle = [
|
||||
{file = "pycodestyle-2.8.0-py2.py3-none-any.whl", hash = "sha256:720f8b39dde8b293825e7ff02c475f3077124006db4f440dcbc9a20b76548a20"},
|
||||
|
|
|
|||
|
|
@ -8,11 +8,11 @@ packages = [{ include = "bot" }]
|
|||
|
||||
[tool.poetry.dependencies]
|
||||
python = "^3.10"
|
||||
discord = { git = "https://github.com/Rapptz/discord.py" }
|
||||
"discord.py" = { git = "https://github.com/Rapptz/discord.py" }
|
||||
aiohttp = "^3.8.1"
|
||||
python-dotenv = "^0.20.0"
|
||||
|
||||
[tool.poetry.group.dev.dependencies]
|
||||
[tool.poetry.dev-dependencies]
|
||||
flake8 = "^4.0.1"
|
||||
black = "^22.3.0"
|
||||
isort = "^5.10.1"
|
||||
|
|
|
|||
42
wulkabot/cogs/automod.py
Normal file
42
wulkabot/cogs/automod.py
Normal file
|
|
@ -0,0 +1,42 @@
|
|||
"""
|
||||
Wulkabot
|
||||
Copyright (C) 2022-present Stanisław Jelnicki
|
||||
"""
|
||||
|
||||
import discord
|
||||
from discord.ext import commands
|
||||
|
||||
from .. import bot
|
||||
from ..utils.text_utils import remove_diacritics
|
||||
from ..utils.views import DeleteButton
|
||||
|
||||
IOS_REQUEST_WORDS = ("kiedy", "bedzie", "wulkanowy", "wulkanowego", "pobrac", "ogarnac")
|
||||
|
||||
|
||||
def is_ios_request(text: str, /) -> bool:
|
||||
words = remove_diacritics(text.replace("?", "").replace("!", "")).lower().split()
|
||||
|
||||
if len(words) > 10:
|
||||
# the text is longer and doesn't look like just a simple question
|
||||
return False
|
||||
|
||||
if "ios" not in words:
|
||||
return False
|
||||
|
||||
return True
|
||||
|
||||
|
||||
class Automod(commands.Cog):
|
||||
@commands.Cog.listener()
|
||||
async def on_message(self, message: discord.Message):
|
||||
if message.author.bot:
|
||||
return
|
||||
|
||||
if is_ios_request(message.content):
|
||||
view = DeleteButton()
|
||||
reply = await message.reply("Witam chyba nigdy", view=view)
|
||||
view.message = reply
|
||||
|
||||
|
||||
async def setup(bot: bot.Wulkabot):
|
||||
await bot.add_cog(Automod())
|
||||
33
wulkabot/utils/text_utils.py
Normal file
33
wulkabot/utils/text_utils.py
Normal file
|
|
@ -0,0 +1,33 @@
|
|||
POLISH_DIACRITICS = {
|
||||
"Ą": "A",
|
||||
"Ć": "C",
|
||||
"Ę": "E",
|
||||
"Ł": "L",
|
||||
"Ń": "N",
|
||||
"Ó": "O",
|
||||
"Ś": "S",
|
||||
"Ź": "Z",
|
||||
"Ż": "Z",
|
||||
"ą": "a",
|
||||
"ć": "c",
|
||||
"ę": "e",
|
||||
"ł": "l",
|
||||
"ń": "n",
|
||||
"ó": "o",
|
||||
"ś": "s",
|
||||
"ź": "z",
|
||||
"ż": "z",
|
||||
}
|
||||
|
||||
|
||||
def remove_diacritics(text: str, /) -> str:
|
||||
"""
|
||||
Replaces Polish diacritics with their ASCII counterparts
|
||||
|
||||
Żółć => Zolc
|
||||
Źdźbło => Zdzblo
|
||||
"""
|
||||
for (diacritic, ascii) in POLISH_DIACRITICS.items():
|
||||
text = text.replace(diacritic, ascii)
|
||||
|
||||
return text
|
||||
19
wulkabot/utils/views.py
Normal file
19
wulkabot/utils/views.py
Normal file
|
|
@ -0,0 +1,19 @@
|
|||
import discord
|
||||
|
||||
|
||||
class DeleteButton(discord.ui.View):
|
||||
def __init__(self, timeout: float | None = 30):
|
||||
super().__init__(timeout=timeout)
|
||||
self.message: discord.Message | None = None
|
||||
|
||||
@discord.ui.button(label="Usuń", style=discord.ButtonStyle.danger)
|
||||
async def confirm(self, interaction: discord.Interaction, button: discord.ui.Button):
|
||||
message = interaction.message
|
||||
await message.edit(
|
||||
content=f"Wiadomość usunięta przez {interaction.user}", embeds=[], attachments=[], view=None, delete_after=5
|
||||
)
|
||||
self.stop()
|
||||
|
||||
async def on_timeout(self) -> None:
|
||||
await self.message.edit(view=None)
|
||||
self.stop()
|
||||
Loading…
Reference in a new issue