Add functools.wraps to BGAPI decorator

We had this on the gatttool backend, but not bgapi. Using this preserves the docstring of the original method.
peplin · Feb 21, 2021 · 0a8edec · 0a8edec
1 parent 538c24f
commit 0a8edec
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pygatt/backends/bgapi/device.py b/pygatt/backends/bgapi/device.py
@@ -1,3 +1,4 @@
+import functools
 import logging
 import time
 
@@ -15,6 +16,7 @@ def connection_required(func):
     """Raise an exception if the device is not connected before calling the
     actual function.
     """
+    @functools.wraps(func)
     def wrapper(self, *args, **kwargs):
         if self._handle is None:
             raise exceptions.NotConnectedError()