From 4df226be82e0553fe6e6e43e23a3d944414a2d1a Mon Sep 17 00:00:00 2001 From: Hayden <48267247+hayden-fr@users.noreply.github.com> Date: Thu, 30 Jan 2025 10:04:56 +0800 Subject: [PATCH] feat: add deprecated decorator (#105) --- __init__.py | 2 +- py/utils.py | 17 +++++++++++++++++ 2 files changed, 18 insertions(+), 1 deletion(-) diff --git a/__init__.py b/__init__.py index 0484eab..2c4cb23 100644 --- a/__init__.py +++ b/__init__.py @@ -89,8 +89,8 @@ async def delete_model_download_task(request): return web.json_response({"success": False, "error": error_msg}) -# @deprecated @routes.get("/model-manager/base-folders") +@utils.deprecated(reason="Use `/model-manager/models` instead.") async def get_model_paths(request): """ Returns the base folders for models. diff --git a/py/utils.py b/py/utils.py index 3a5163a..5ae9b01 100644 --- a/py/utils.py +++ b/py/utils.py @@ -7,6 +7,7 @@ import logging import requests import traceback import configparser +import functools import comfy.utils import folder_paths @@ -20,6 +21,10 @@ def print_info(msg, *args, **kwargs): logging.info(f"[{config.extension_tag}] {msg}", *args, **kwargs) +def print_warning(msg, *args, **kwargs): + logging.warning(f"[{config.extension_tag}][WARNING] {msg}", *args, **kwargs) + + def print_error(msg, *args, **kwargs): logging.error(f"[{config.extension_tag}] {msg}", *args, **kwargs) logging.debug(traceback.format_exc()) @@ -29,6 +34,18 @@ def print_debug(msg, *args, **kwargs): logging.debug(f"[{config.extension_tag}] {msg}", *args, **kwargs) +def deprecated(reason: str): + def decorator(func): + @functools.wraps(func) + def wrapper(*args, **kwargs): + print_warning(f"{func.__name__} is deprecated: {reason}") + return func(*args, **kwargs) + + return wrapper + + return decorator + + def _matches(predicate: dict): def _filter(obj: dict): return all(obj.get(key, None) == value for key, value in predicate.items())