path: root/discordrpc/button.py

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