Ntfy headers

src/ntfy.py

@@ -1,16 +1,35 @@
 import requests
 
+from typing import TypedDict, Optional
 from print_t import print_t
+from enum import Enum
+
+class HeaderPriority(Enum):
+	Urgent = "5"
+	High   = "4"
+	Medium = "3"
+	Low    = "2"
+	Min    = "1"
+
+class Headers(TypedDict):
+	Priority: HeaderPriority
+	Title: str
+	Tags: str
 
 class Ntfy:
 	def __init__(self, server: str, logging_disabled: bool):
 		self.server = server
 		self.logging_disabled = logging_disabled
 
-	def send(self, message: str):
+	def send(self, message: str, title: Optional[str] = None, extra_headers: Optional[Headers] = None):
+		headers = {}
+		if title:
+			headers["Title"] = title.encode(encoding="utf-8")
+		if extra_headers:
+			headers.update(extra_headers)
 		if not self.logging_disabled:
 			print_t("Ntfy OUT: " + message)
 		try:
-			requests.post(self.server, data=message.encode(encoding="utf-8"))
+			requests.post(self.server, data=message.encode(encoding="utf-8"), headers=headers)
 		except Exception as err:
 			print_t(f"\033[31m{err}\033[0m")