Adding salt/utils/decorators/extension_deprecation.py

salt/utils/decorators/extension_deprecation.py

@@ -0,0 +1,33 @@
+"""
+Decorators for deprecation of modules to Salt extensions
+"""
+import logging
+from functools import wraps
+
+import salt.utils.args
+import salt.utils.versions
+
+log = logging.getLogger(__name__)
+
+
+def extension_deprecation_message(version, extension_name, extension_repo):
+    """
+    Decorator wrapper to warn about deprecation
+    """
+
+    def decorator(function):
+        @wraps(function)
+        def wrapper(*args, **kwargs):
+            salt.utils.versions.warn_until(
+                version,
+                f"The '{extension_name}' functionality in Salt has been deprecated and its "
+                "functionality will be removed in version {version} in favor of the "
+                f"saltext.{extension_name} Salt Extension. "
+                f"({extension_repo})",
+                category=FutureWarning,
+            )
+            return function(*args, **salt.utils.args.clean_kwargs(**kwargs))
+
+        return wrapper
+
+    return decorator