haldis/app/notification.py

68 lines
2.2 KiB
Python

"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