haldis/app/notification.py

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

import requests
from flask import current_app as app
from flask import url_for
from models.order import Order


def webhook_text(order_item: Order) -> typing.Optional[str]:
    "Function that makes the text for the notification"
    if "Testlocation" in order_item.location.name:
        return None

    if order_item.courier is not None:
        # pylint: disable=C0301
        return "<!channel|@channel> {3} is going to {1}, order <{0}|here>! Deadline in {2} minutes!".format(
            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.courier.username.title(),
        )

    return "<!channel|@channel> New order for {}. Deadline in {} minutes. <{}|Open here.>".format(
        order_item.location.name,
        remaining_minutes(order_item.stoptime),
        url_for("order_bp.order_from_id", order_id=order_item.id, _external=True),
    )


def post_order_to_webhook(order_item: Order) -> None:
    "Function that sends the notification for the order"
    message = webhook_text(order_item)
    if message:
        webhookthread = WebhookSenderThread(
            message, app.config["SLACK_WEBHOOK"])
        webhookthread.start()


class WebhookSenderThread(Thread):
    "Extension of the Thread class, which sends a webhook for the notification"

    def __init__(self, message: str, url: str) -> None:
        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(self.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