Fix Handler unregister

Add checking for handler_obj.handler
This commit is contained in:
Butvin Mihail 2022-10-31 09:43:03 +03:00 committed by GitHub
parent 8604ab7cd0
commit 8a4ecc7b96
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -74,11 +74,12 @@ class Handler:
:param handler: callback
:return:
"""
for handler_obj in self.handlers:
registered = handler_obj.handler
if handler in self.handlers:
self.handlers.remove(handler)
if handler == handler_obj or handler == handler_obj.handler:
self.handlers.remove(handler_obj)
return True
raise ValueError('This handler is not registered!')
async def notify(self, *args):