From 6d15e2ad298ab89e11181772b5a25c28f50930c5 Mon Sep 17 00:00:00 2001 From: KKlochko Date: Tue, 12 Dec 2023 20:52:14 +0200 Subject: [PATCH] Add a notify function to send one notification. --- notification_producer/producer_client.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/notification_producer/producer_client.py b/notification_producer/producer_client.py index a6b30c9..f06cc30 100644 --- a/notification_producer/producer_client.py +++ b/notification_producer/producer_client.py @@ -73,3 +73,15 @@ class ProducerClient: self.writer.close() await self.writer.wait_closed() + async def notify(self, title, message): + await self.connect() + + try: + await self.send_notification(title, message) + await self.send_notification('', '') + + except asyncio.CancelledError: + pass + finally: + self.writer.close() + await self.writer.wait_closed()