add custom callback when notification received

This commit is contained in:
Kavish Devar
2024-09-27 14:00:20 +05:30
parent 654b2291a9
commit 692830f36c

View File

@@ -19,8 +19,10 @@ class Connection:
return False return False
return True return True
def initialize_notifications(self): def initialize_notifications(self, callback: callable = None):
self.notifications = Notifications(self.socket, self.notification_callback) if callback is None:
callback = self.notification_callback
self.notifications = Notifications(self.socket, callback)
self.notificationListener = self.notifications.notificationListener self.notificationListener = self.notifications.notificationListener
self.BatteryNotification = self.notifications.BatteryNotification self.BatteryNotification = self.notifications.BatteryNotification
self.notifications.initialize() self.notifications.initialize()