From 692830f36c3d6f37f39ab3ca0b4e4a6d70a9052d Mon Sep 17 00:00:00 2001
From: Kavish Devar <mail@kavishdevar.me>
Date: Fri, 27 Sep 2024 14:00:20 +0530
Subject: [PATCH] add custom callback when notification received

---
 aln/__init__.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/aln/__init__.py b/aln/__init__.py
index cd56f28..883babc 100644
--- a/aln/__init__.py
+++ b/aln/__init__.py
@@ -19,8 +19,10 @@ class Connection:
             return False
         return True
     
-    def initialize_notifications(self):
-        self.notifications = Notifications(self.socket, self.notification_callback)
+    def initialize_notifications(self, callback: callable = None):
+        if callback is None:
+            callback = self.notification_callback
+        self.notifications = Notifications(self.socket, callback)
         self.notificationListener = self.notifications.notificationListener
         self.BatteryNotification = self.notifications.BatteryNotification
         self.notifications.initialize()