reginaldCog/debug_stuff.py

@@ -1,15 +1,24 @@
+import functools
+import inspect
+
+
 def debug(func):
-    def wrap(*args, **kwargs):
+    if inspect.iscoroutinefunction(func):
-        # Log the function name and arguments
+
+        @functools.wraps(func)
+        async def wrap(*args, **kwargs):
             print(f"DEBUG: Calling {func.__name__} with args: {args}, kwargs: {kwargs}")
-
+            result = await func(*args, **kwargs)
-        # Call the original function
+            print(f"DEBUG: {func.__name__} returned: {result}")
-        result = func(*args, **kwargs)
+            return result
-
+
-        # Log the return value
+        return wrap
+
+    @functools.wraps(func)
+    def wrap(*args, **kwargs):
+        print(f"DEBUG: Calling {func.__name__} with args: {args}, kwargs: {kwargs}")
+        result = func(*args, **kwargs)
         print(f"DEBUG: {func.__name__} returned: {result}")
-
-        # Return the result
         return result
 
     return wrap