fix logging wrapper

setup.py

@@ -1,6 +1,6 @@
 from setuptools import setup, find_packages
 
-__version__ = "1.5.0"
+__version__ = "1.5.1"
 
 with open("README.md", "r", encoding="utf-8") as fh:
     long_description = fh.read()

src/crypto_pkg/utils/logging.py

@@ -1,3 +1,4 @@
+import inspect
 import logging
 
 from crypto_pkg import settings
@@ -9,14 +10,17 @@ def get_logger(location: str = __name__):
     return log
 
 
-def set_level(logger):
+def set_level(logger: logging.Logger, level: int = logging.DEBUG):
     def deco(func):
         def wrapper(*args, **kwargs):
             _verbose = kwargs.get('_verbose', None)
-            if _verbose is None and len(args) == 5:
-                _verbose = args[4]
+            if _verbose is None:
+                params = inspect.signature(func).parameters
+                default_verbose = params.get('_verbose', {}).default
+                _verbose = next((arg_value for i, arg_value in enumerate(args) if list(params)[i] == '_verbose'),
+                                default_verbose)
             if _verbose is True:
-                logger.setLevel(logging.DEBUG)
+                logger.setLevel(level)
             return func(*args, **kwargs)
 
         return wrapper

tests/test_logger.py

@@ -0,0 +1,49 @@
+import logging
+import unittest
+
+from crypto_pkg.utils.logging import get_logger, set_level
+
+log = get_logger()
+
+
+class TestLogger(unittest.TestCase):
+
+    def test_logger_dec_without_level(self):
+        @set_level(log)
+        def func(_a: int, _verbose=False):
+            pass
+
+        log.setLevel(logging.INFO)
+
+        func(_a=1)
+        self.assertEqual(log.level, logging.INFO)
+        func(1)
+        self.assertEqual(log.level, logging.INFO)
+
+        func(_a=1, _verbose=True)
+        self.assertEqual(log.level, logging.DEBUG)
+
+        log.setLevel(logging.INFO)
+
+        func(1, True)
+        self.assertEqual(log.level, logging.DEBUG)
+
+    def test_logger_dec_with_level(self):
+        @set_level(logger=log, level=logging.ERROR)
+        def func(_a: int, _verbose=False):
+            pass
+
+        log.setLevel(logging.INFO)
+
+        func(_a=1)
+        self.assertEqual(log.level, logging.INFO)
+        func(1)
+        self.assertEqual(log.level, logging.INFO)
+
+        func(_a=1, _verbose=True)
+        self.assertEqual(log.level, logging.ERROR)
+
+        log.setLevel(logging.INFO)
+
+        func(1, True)
+        self.assertEqual(log.level, logging.ERROR)