From fa6cddc2f5e1bdfc886926a41145f2d470872a77 Mon Sep 17 00:00:00 2001 From: Andrei Date: Wed, 12 Apr 2017 14:37:44 -0700 Subject: [PATCH] Add support for constructing webhook from a URL --- disco/types/webhook.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/disco/types/webhook.py b/disco/types/webhook.py index 4a630d3..e9c6080 100644 --- a/disco/types/webhook.py +++ b/disco/types/webhook.py @@ -1,8 +1,13 @@ +import re + from disco.types.base import SlottedModel, Field, snowflake from disco.types.user import User from disco.util.functional import cached_property +WEBHOOK_URL_RE = re.compile(r'\/api\/webhooks\/(\d+)\/(.[^/]+)') + + class Webhook(SlottedModel): id = Field(snowflake) guild_id = Field(snowflake) @@ -12,6 +17,14 @@ class Webhook(SlottedModel): avatar = Field(str) token = Field(str) + @classmethod + def from_url(cls, url): + results = WEBHOOK_URL_RE.findall(url) + if len(results) != 1: + return Exception('Invalid Webhook URL') + + return cls(id=results[0][0], token=results[0][1]) + @cached_property def guild(self): return self.client.state.guilds.get(self.guild_id)