From d4cee86e2b18a8541e97f29a56fe3397af5d6a07 Mon Sep 17 00:00:00 2001
From: Levi Vasconcelos <levi.vasconcelos@printbear.net>
Date: Mon, 26 Feb 2024 16:51:42 -0300
Subject: [PATCH] reviews: add custom message option to @deprecated_parameter;
 add docstring

---
 supervision/utils/internal.py | 43 +++++++++++++++++++++++++++++++----
 1 file changed, 38 insertions(+), 5 deletions(-)

diff --git a/supervision/utils/internal.py b/supervision/utils/internal.py
index 290addfa..e9abac52 100644
--- a/supervision/utils/internal.py
+++ b/supervision/utils/internal.py
@@ -4,12 +4,43 @@ from typing import Callable
 
 
 def deprecated_parameter(
-    old_param: str, new_param: str, map_func: Callable = lambda x: x
+    old_parameter: str, 
+    new_parameter: str, 
+    map_function: Callable = lambda x: x, 
+    warn_message: str = "Warning: '{old_parameter}' in '{function_name}' is deprecated: " \
+                        "use '{new_parameter}' instead.",
+    **message_kwargs
 ):
+    """
+    A decorator to mark a function's parameter as deprecated and issue a warning when used.
+
+    Parameters:
+    - old_parameter (str): The name of the deprecated parameter.
+    - new_parameter (str): The name of the parameter that should be used instead.
+    - map_function (Callable, optional): A function used to map the value of the old parameter to the new parameter.
+      Defaults to the identity function.
+    - warn_message (str, optional): The warning message to be displayed when the deprecated parameter is used.
+      Defaults to a generic warning message with placeholders for the old parameter, new parameter, and function name.
+    - **message_kwargs: Additional keyword arguments that can be used to customize the warning message.
+
+    Returns:
+    Callable: A decorator function that can be applied to mark a function's parameter as deprecated.
+
+    Usage Example:
+    ```python
+    @deprecated_parameter(old_parameter="old_param", new_parameter="new_param")
+    def example_function(new_param):
+        print(f"Function called with new_param: {new_param}")
+
+    # When calling the function with the deprecated parameter:
+    example_function(old_param="deprecated_value")
+    ```
+    This will trigger a deprecation warning and execute the function with the mapped value of the deprecated parameter.
+    """
     def decorator(func):
         @functools.wraps(func)
         def wrapper(*args, **kwargs):
-            if old_param in kwargs:
+            if old_parameter in kwargs:
                 # In case of a method, display also the class name.
                 if args and hasattr(args[0], "__class__"):
                     class_name = args[0].__class__.__name__
@@ -19,13 +50,15 @@ def deprecated_parameter(
 
                 # Display deprecation warning
                 warnings.warn(
-                    f"Warning: '{old_param}' in '{function_name}' is deprecated: "
-                    f"use '{new_param}' instead.",
+                    message=warn_message.format(function_name=function_name, 
+                                                old_parameter=old_parameter, 
+                                                new_parameter=new_parameter, 
+                                                **message_kwargs),
                     category=DeprecationWarning,
                     stacklevel=2,
                 )
                 # Map old_param to new_param
-                kwargs[new_param] = map_func(kwargs.pop(old_param))
+                kwargs[new_parameter] = map_function(kwargs.pop(old_parameter))
 
             return func(*args, **kwargs)