From bf2e0ac826bfb81f7559cd44e956c6a4aad982f1 Mon Sep 17 00:00:00 2001
From: Guillaume Leclerc <leclerc@mit.edu>
Date: Fri, 14 May 2021 00:04:13 -0400
Subject: [PATCH] Add support for decoration of constructors

Resolves #9
---
 fastargs/decorators.py | 24 +++++++++++++++++-------
 1 file changed, 17 insertions(+), 7 deletions(-)

diff --git a/fastargs/decorators.py b/fastargs/decorators.py
index 119b2f0..38bc5b9 100644
--- a/fastargs/decorators.py
+++ b/fastargs/decorators.py
@@ -40,20 +40,31 @@ def __call__(self, *args, **kwargs):
             else:
                 raise e
 
-
+def extract_function(func):
+    if hasattr(func, '__fastarg_wrapper'):
+        return getattr(func, '__fastarg_wrapper')
+    else:
+        return WrappedFunction(func)
 
 def param(parameter, alias=None):
     if isinstance(parameter, str):
         parameter = tuple(parameter.split('.'))
 
-    if alias==None:
+    if alias is None:
         alias = parameter[-1]
 
     def wrapper(func):
-        if not isinstance(func, WrappedFunction):
-            func = WrappedFunction(func)
+
+        func = extract_function(func)
+
         func.add_arg(parameter, alias)
-        return func
+
+        def result(*args, **kwargs):
+            return func(*args, **kwargs)
+
+        setattr(result, '__fastarg_wrapper', func)
+        return result
+
     return wrapper
 
 def section(section):
@@ -61,8 +72,7 @@ def section(section):
         section = tuple(section.split('.'))
 
     def wrapper(func):
-        if not isinstance(func, WrappedFunction):
-            func = WrappedFunction(func)
+        func = extract_function(func)
         func.set_section(section)
         return func
     return wrapper