discord_messenger.py

import requests
import os
from utils import take_screenshot_of_app, load_yaml_config

# Load webhooks from config.yaml
def load_webhooks():
    config = load_yaml_config()
    return config['webhooks']

# Send message to Discord
def send_message_to_discord(message, noimage, win, debug):
    screenshot_path = None
    if not noimage:
        # Take a screenshot and return the temporary file path
        screenshot_path = take_screenshot_of_app("Trade Automation Toolbox", win)
    
    message_ids = []
    webhooks = load_webhooks()  # Load webhooks from config.yaml

    for webhook in webhooks:
        url = webhook["url"]
        thread_id = webhook.get("thread_id")
        if thread_id:
            url += f"?thread_id={thread_id}"

        payload = {"content": message}
        response = None  # Initialize response variable

        try:
            if screenshot_path and os.path.isfile(screenshot_path):
                with open(screenshot_path, "rb") as image_file:
                    files = {"file": (os.path.basename(screenshot_path), image_file)}
                    response = requests.post(url, data=payload, files=files)
            else:
                response = requests.post(url, data=payload)

            # Check if response is successful
            if response.status_code not in [200, 204]:
                print(f"Failed to send message to webhook {url}. Status code: {response.status_code}")
                print(f"Response: {response.text}")
                message_ids.append(None)
            else:
                try:
                    response_data = response.json()
                    message_id = response_data.get('id')
                    message_ids.append(message_id)
                except ValueError:
                    message_ids.append(None)

        except Exception as e:
            print(f"Error while sending message to {url}: {e}")
            message_ids.append(None)

    # Clean up the temporary screenshot file after all webhooks have been processed
    if screenshot_path and os.path.exists(screenshot_path):
        os.remove(screenshot_path)

    return message_ids

# Delete messages from Discord
def delete_messages(message_ids):
    webhooks = load_webhooks()  # Load webhooks from config.yaml
    for msg_id in message_ids:
        url = f"{webhooks[0]['url']}/messages/{msg_id}"
        requests.delete(url)