Infinidoge/nomen
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

implement plaintext notifications

Browse source
This commit is contained in:
Infinidoge 2024-12-23 00:49:14 -05:00
parent ec54e31b5c
commit 0df744c946
Signed by: Infinidoge
SSH key fingerprint: SHA256:GT2StvPQMMfFHyiiFJymQxfTG/z6EWLJ6NWItf5K5sA
1 changed files with 14 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
nomen/notifications.py
View file

@ -1,5 +1,6 @@
import logging
from asyncio import TaskGroup
from textwrap import indent
from disnake import Embed
from disnake.ext.commands import Cog, group, guild_only
@ -10,7 +11,7 @@ log = logging.getLogger("nomen.notifications")
log.setLevel(logging.INFO)
async def handle_notification(db_updates, ctx, message, keyword, user_id):
async def handle_notification(db_updates, ctx, message, keyword, user_id, use_embed):
"""
Async task to dispatch a notification
"""
@ -26,17 +27,19 @@ async def handle_notification(db_updates, ctx, message, keyword, user_id):
log.debug(f"- - Notifying {user_id}")
db_updates.append((ctx.guild.id, keyword, user_id))
footer = f"\n\n<t:{int(message.created_at.timestamp())}:R> | [Show]({message.jump_url}) | <#{ctx.channel.id}>"
header = f"🔔 `{message.author.display_name}` mentioned `{keyword}` on `{ctx.guild}`:"
footer = f"\n<t:{int(message.created_at.timestamp())}:R> | [Show]({message.jump_url}) | <#{ctx.channel.id}>"
embed = Embed(
description=message.content + footer,
)
embed.set_author(name=message.author, icon_url=message.author.display_avatar)
if use_embed:
embed = Embed(description=message.content + "\n" + footer)
embed.set_author(
name=f"{message.author.display_name} ({message.author})",
icon_url=message.author.display_avatar,
)
await member.send(
f":bell: `{message.author}` mentioned `{keyword}` on `{ctx.guild}`:",
embed=embed,
)
await member.send(header, embed=embed)
else:
await member.send("\n".join((header, indent(message.content, "> ", lambda line: True).strip(), footer)))
async def handle_triggers(ctx, message):
@ -88,7 +91,7 @@ async def handle_triggers(ctx, message):
async for keyword, user_id, use_embed in cur:
log.debug(f"- Creating notification task: '{keyword}' for {user_id}")
tg.create_task(handle_notification(db_updates, ctx, message, keyword, user_id))
tg.create_task(handle_notification(db_updates, ctx, message, keyword, user_id, use_embed))
await ctx.bot.db.executemany(
"UPDATE keywords SET count = count + 1 WHERE guild_id=? AND keyword=? AND user_id=?",