generated from HoloTech/holotech-plugin-template
fix: Add edit support to ThreadedWebhook
This commit is contained in:
+10
-3
@@ -17,22 +17,25 @@ def LOWER(expression: Expression) -> RawExpr:
|
|||||||
|
|
||||||
return RawExpr(final_expr, final_values)
|
return RawExpr(final_expr, final_values)
|
||||||
|
|
||||||
|
|
||||||
def asexpr(value: Any) -> RawExpr:
|
def asexpr(value: Any) -> RawExpr:
|
||||||
"""
|
"""
|
||||||
Turn a value into an expression.
|
Turn a value into an expression.
|
||||||
"""
|
"""
|
||||||
return RawExpr(sql.Placeholder(), (value,))
|
return RawExpr(sql.Placeholder(), (value,))
|
||||||
|
|
||||||
|
|
||||||
async def fire_and_forget(awaitable, do_in=1, ignorable=(discord.HTTPException)):
|
async def fire_and_forget(awaitable, do_in=1, ignorable=(discord.HTTPException)):
|
||||||
await asyncio.sleep(do_in)
|
await asyncio.sleep(do_in)
|
||||||
try:
|
try:
|
||||||
await awaitable
|
await awaitable
|
||||||
except ignorable:
|
except ignorable:
|
||||||
pass
|
pass
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
# TODO: Log unexpected exceptions
|
# TODO: Log unexpected exceptions
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
|
||||||
class ThreadedWebhook(discord.Webhook):
|
class ThreadedWebhook(discord.Webhook):
|
||||||
__slots__ = ("thread_id",)
|
__slots__ = ("thread_id",)
|
||||||
|
|
||||||
@@ -55,10 +58,14 @@ class ThreadedWebhook(discord.Webhook):
|
|||||||
kwargs.setdefault("thread", discord.Object(self.thread_id))
|
kwargs.setdefault("thread", discord.Object(self.thread_id))
|
||||||
return await super().send(*args, **kwargs)
|
return await super().send(*args, **kwargs)
|
||||||
|
|
||||||
|
async def edit_message(self, *args, **kwargs):
|
||||||
|
if self.thread_id is not None:
|
||||||
|
kwargs.setdefault("thread", discord.Object(self.thread_id))
|
||||||
|
return await super().edit_message(*args, **kwargs)
|
||||||
|
|
||||||
async def test_webhook(self):
|
async def test_webhook(self):
|
||||||
embed = discord.Embed(
|
embed = discord.Embed(
|
||||||
title="Testing",
|
title="Testing", description="Testing logging webhook, feel free to delete."
|
||||||
description="Testing logging webhook, feel free to delete."
|
|
||||||
)
|
)
|
||||||
result = await self.send(embed=embed, wait=True, silent=True)
|
result = await self.send(embed=embed, wait=True, silent=True)
|
||||||
asyncio.create_task(fire_and_forget(result.delete()))
|
asyncio.create_task(fire_and_forget(result.delete()))
|
||||||
|
|||||||
Reference in New Issue
Block a user