haldis/app/notification.py

"Script that handles Haldis notifications on chat platforms"
import json
from datetime import datetime
from threading import Thread

import requests
from flask import current_app as app
from flask import url_for


def post_order_to_webhook(order_item) -> None:
    "Function that sends the notification for the order"
    message = ""
    if order_item.courrier is not None:
        message = "<!channel|@channel> {3} is going to {1}, order <{0}|here>! Deadline in {2} minutes!".format(  # pylint: disable=C0301
            url_for("order_bp.order_from_id", order_id=order_item.id,
                    _external=True),
            order_item.location.name,
            remaining_minutes(order_item.stoptime),
            order_item.courrier.username.title(),
        )
    else:
        message = "<!channel|@channel> New order for {}. Deadline in {} minutes. <{}|Open here.>".format(  # pylint: disable=C0301
            order_item.location.name,
            remaining_minutes(order_item.stoptime),
            url_for("order_bp.order_from_id",
                    order_id=order_item.id, _external=True),
        )
    webhookthread = WebhookSenderThread(message, app.config["SLACK_WEBHOOK"])
    webhookthread.start()


class WebhookSenderThread(Thread):
    def __init__(self, message: str, url: str) -> None:
        "Extension of the Thread class, which sends a webhook for the notification"
        super(WebhookSenderThread, self).__init__()
        self.message = message
        self.url = url

    def run(self) -> None:
        self.slack_webhook()

    def slack_webhook(self) -> None:
        "The webhook for the specified chat platform"
        if self.url:
            requests.post(url, json={"text": self.message})
        else:
            print(self.message)


def remaining_minutes(value) -> str:
    "Return the remaining minutes until the deadline of and order"
    delta = value - datetime.now()
    if delta.total_seconds() < 0:
        return "0"
    minutes = delta.total_seconds() // 60
    return "%02d" % minutes