discord_web.py

import requests
#import json
import sys
from discord_webhook import DiscordWebhook, DiscordEmbed
from getConfig import ConfigHandler

getConfig = ConfigHandler()

def send_webhook(url, id="Unknown", status="error", message=None):
    if url is None:
        return
    
    if message is not None:
        message = remove_ansi_escape_sequences(message)
        
        
    webhook = DiscordWebhook(url, rate_limit_retry=True, timeout=30)
    if(status == "starting"):
        title="Starting"
        color="fc8803"
        embed = DiscordEmbed(title, description="Starting holo-downloader", color=color)
        embed.set_timestamp()
        webhook.add_embed(embed)
        webhook_response = webhook.execute()
        return
    elif(status=="membership-error"):
        title = "Membership error"
        color="ff0000"
        embed = DiscordEmbed(title, description=("Error checking membership streams for [{0}](https://www.youtube.com/channel/{0}). \nCheck cookies!".format(id)), color=color)
        if message:
            embed.add_embed_field(name="Error Message: {0}".format(id), value=message)
            embed.set_footer(text='Error Logger')
        embed.set_timestamp()
        #embed.set_author(name='{0}'.format(id), url='www.youtube.com/channel/{0}'.format(id))
        #embed.set_thumbnail(url=data.get('thumbnail_url'))
        webhook.add_embed(embed)
        webhook_response = webhook.execute()
        return
    
    embed_error = None
    try:
        response = requests.get("https://www.youtube.com/oembed?format=json&url=https://www.youtube.com/watch?v={0}".format(id), timeout=30)
    except Exception as e:
        embed_error = str(e)
    
    color="03b2f8"   

    # create embed object for webhook
    # you can set the color as a decimal (color=242424) or hex (color="03b2f8") number
    match status:
        case "recording":
            title = "Recording"
            color="0011ff"
        case "waiting":
            title = "Waiting"
            color="ffff00"
        case "error":
            title = "Error"
            color="ff0000"
        case "done":
            title = "Done"
            color="00ff00"

    # Check if the request was successful (status code 200)
    if embed_error is None and response.status_code == 200:
        # Parse the JSON data from the response
        data = response.json()

        embed = DiscordEmbed(title, description="[{0}](https://youtu.be/{1})".format(data.get('title'),id), color=color)

        embed.set_author(name=data.get('author_name'), url=data.get('author_url'))

        embed.set_thumbnail(url=data.get('thumbnail_url'))
        
    elif embed_error:
        embed = DiscordEmbed(title, description="Video https://youtu.be/{0} is not accessible due to error: {1}".format(id, embed_error[:250]), color=color)
    else:                
        embed = DiscordEmbed(title, description="Video https://youtu.be/{0} is not accessible, perhaps it has been privated".format(id), color=color)

    if message:
        embed.add_embed_field(name="Message: {0}".format(id), value=message)
    embed.set_timestamp()

    # add embed object to webhook
    webhook.add_embed(embed)
    


    webhook_response = webhook.execute()
    
def remove_ansi_escape_sequences(text):
    import re
    # This regular expression matches ANSI escape sequences
    ansi_escape = re.compile(r'\x1B[@-_][0-?]*[ -/]*[@-~]')
    return ansi_escape.sub('', text)

def main(id, status, message = None):
    if getConfig.get_discord_webhook() is not None:
        send_webhook(getConfig.get_discord_webhook(), id, status, message)


if __name__ == "__main__":
    main(sys.argv[1], sys.argv[2])