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()