Merge pull request #1260 from gyorb/fix-timer

fix timeit profiler
Ericsson · Dec 14, 2017 · bac5ab8 · bac5ab8
2 parents 116d1e6 + 224142b
commit bac5ab8
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/libcodechecker/profiler.py b/libcodechecker/profiler.py
@@ -8,6 +8,8 @@
 import cProfile
 import pstats
 
+from functools import wraps
+
 try:
     from StringIO import StringIO
 except ImportError:
@@ -61,10 +63,14 @@ def release_wrapper(*args, **kwargs):
         res = function(*args, **kwargs)
         return res
 
-    if LoggerFactory.get_log_level() == logger.DEBUG:
-        return debug_wrapper
-    else:
-        return release_wrapper
+    @wraps(function)
+    def wrapper(*args, **kwargs):
+        if LoggerFactory.get_log_level() == logger.DEBUG:
+            return debug_wrapper(*args, **kwargs)
+        else:
+            return release_wrapper(*args, **kwargs)
+
+    return wrapper
 
 
 def profileit(function):