diff --git a/freenit/__init__.py b/freenit/__init__.py index 818ad96..d91d094 100644 --- a/freenit/__init__.py +++ b/freenit/__init__.py @@ -1 +1 @@ -__version__ = "0.3.28" +__version__ = "0.3.29" diff --git a/freenit/decorators.py b/freenit/decorators.py index cbeac34..eef5ced 100644 --- a/freenit/decorators.py +++ b/freenit/decorators.py @@ -11,6 +11,7 @@ def __init__(self, route, tags=["object"]): def __call__(self, cls): origGet = getattr(cls, "get", None) origPost = getattr(cls, "post", None) + origPut = getattr(cls, "put", None) origPatch = getattr(cls, "patch", None) origDelete = getattr(cls, "delete", None) @@ -31,6 +32,13 @@ class Wrapped(cls): tags=tags, ) post = _deco(origPost) + if callable(origPut): + _deco = self.app.put( + self.route, + summary=getattr(origPut, "description", f"Replace {tag}"), + tags=tags, + ) + put = _deco(origPut) if callable(origPatch): _deco = self.app.patch( self.route,