Minor Fixes

Author: HMohammad2520

classmods/_decorators.py

@@ -2,7 +2,7 @@
 from functools import wraps
 from typing import Any, Callable, Literal, Optional, ParamSpec, Tuple, TypeAlias, TypeVar, Union, overload
 
-LOG_LEVEL: TypeAlias = Literal['CRITICAL', 'ERROR', 'WARNING', 'INFO', 'DEBUG', 'NOTSET', 'DEFAULT']
+LOG_LEVEL: TypeAlias = Literal['CRITICAL', 'ERROR', 'WARNING', 'INFO', 'DEBUG', 'NOTSET']
 
 P = ParamSpec("P")
 R = TypeVar("R")
@@ -17,7 +17,7 @@ def logwrap(
     A simple dynamic decorator to log function calls using the `logging` module with your current project configurations.
     Use the `LOG_LEVEL` literal to specify standard log levels.
 
-    LOG_LEVEL = ['CRITICAL', 'ERROR', 'WARNING', 'INFO', 'DEBUG', 'NOTSET', 'DEFAULT']
+    LOG_LEVEL = ['CRITICAL', 'ERROR', 'WARNING', 'INFO', 'DEBUG', 'NOTSET']
 
     - The messages are formatted dynamically using templating.
         - Available variables:
@@ -88,8 +88,8 @@ def normalize(
             return option
 
     before = normalize('DEBUG', 'Calling {func} - kwargs={kwargs}', before)
-    after = normalize('INFO', 'Function {func} ended. result={result}', after)
     on_exception = normalize('ERROR', 'Error in {func}: {e}', on_exception)
+    after = normalize('INFO', 'Function {func} ended. result={result}', after)
 
     def decorator(func: Callable[P, R]) -> Callable[P, R]:
         sig = inspect.signature(func)

test/test_decorators.py

@@ -1,4 +1,4 @@
-from classmods import suppress_errors
+from classmods import logwrap, suppress_errors
 
 @suppress_errors(Exception)
 def return_exception():
@@ -27,4 +27,18 @@ def test_supress_error():
     assert result is False, f"Expected False, got {result}"
 
     result = return_any()
-    assert result == 'Failed', f"Expected 'Failed', got {result}"
+    assert result == 'Failed', f"Expected 'Failed', got {result}"
+
+def test_standard_use():
+    # Example with defaults.
+    @logwrap(before='Starting', after='Ended')
+    def my_new_func():
+        ...
+    my_new_func()
+
+def test_with_custom_leveling():
+    # Example with Custom Levels
+    @logwrap(before=('INFO', 'Function starting'), after=('INFO', 'Function ended'))
+    def my_func(my_arg, my_kwarg=None):
+        ...
+    my_func('hello', my_kwarg=123) # calling the function