1 files changed, 38 insertions, 0 deletions
diff --git a/discordrpc/button.py b/discordrpc/button.py
new file mode 100644
index 0000000..b5d21ef
--- /dev/null
+++ b/discordrpc/button.py
@@ -0,0 +1,38 @@
+from .exceptions import *
+
+valid_url = ["https://", "http://"]
+
+def button_one(label:str, url:str):
+    if any(v in url for v in valid_url):
+        payloads = {"label": label, "url": url}
+        return payloads
+    else:
+        raise InvalidURL
+
+def button_two(label:str, url:str):
+    if any(v in url for v in valid_url):
+        payloads = {"label": label, "url": url}
+        return payloads
+    else:
+        raise InvalidURL
+
+def button(
+    button_one_label:str=None,
+    button_two_label:str=None,
+    button_one_url:str=None,
+    button_two_url:str=None):
+    
+    if button_one_label == None:
+        raise ButtonError('"button_one_label" cannot None')
+    if button_one_url == None:
+        raise ButtonError('"button_one_url" cannot None')
+    if button_two_label == None:
+        raise ButtonError('"button_two_label" cannot None')
+    if button_two_url == None:
+        raise ButtonError('"button_two_url" cannot None')
+    
+    btn_one = button_one(label=button_one_label, url=button_one_url)
+    btn_two = button_two(label=button_two_label, url=button_two_url)
+    payloads = [btn_one, btn_two]
+
+    return payloads