26 lines
805 B
Python
26 lines
805 B
Python
import discord
|
|
from urllib.parse import urlparse, parse_qs
|
|
|
|
|
|
class ThreadedWebhook(discord.Webhook):
|
|
__slots__ = ("thread_id",)
|
|
|
|
def __init__(self, *args, thread_id=None, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.thread_id = thread_id
|
|
|
|
@classmethod
|
|
def from_url(cls, url: str, *args, **kwargs):
|
|
self = super().from_url(url, *args, **kwargs)
|
|
parse = urlparse(url)
|
|
if parse.query:
|
|
args = parse_qs(parse.query)
|
|
if "thread_id" in args:
|
|
self.thread_id = int(args["thread_id"][0])
|
|
return self
|
|
|
|
async def send(self, *args, **kwargs):
|
|
if self.thread_id is not None:
|
|
kwargs.setdefault("thread", discord.Object(self.thread_id))
|
|
return await super().send(*args, **kwargs)
|