duke/view.py

import os
import sys
from pprint import pprint
import json

from dotenv import load_dotenv
import roman
import requests

base_url = 'https://discordapp.com/api/webhooks/'


def send_message(status):
    name = status['name']
    its_id = status['id']
    status_text = f"[{name} [{its_id}]](https://www.torn.com/profiles.php?XID={its_id}) → " + \
                  "Loot level changed to "+roman.toRoman(status['integer'])
    status_text += "\n" + status['time_string']
    this_json = {'content': status_text}
    pprint(this_json)
    _send(this_json)


def send_alert(this_alert):
    this_json = {'content': this_alert}
    _send(this_json)


def _send(send_json):
    load_dotenv()
    webhooks_json = os.environ["DISCORD_TOKENS"]
    webhooks = json.loads(webhooks_json)
    for webhook_key in webhooks:
        webhook = webhooks[webhook_key]
        if 'Alert: 10 mins' in send_json['content'] and webhook_key == 'Unrelenting':
            send_json['content'] = '@here ' + send_json['content']
        url = base_url+webhook
        response = requests.post(url, json=send_json)
        if response.status_code in [200, 204]:
            print("Webhook executed")
        else:
            print("status code {}: {}".format(response.status_code, response.content.decode("utf-8")))
            if response.status_code == int(429):
                sys.exit(1)