Use @decorator in @time_limit decorator to remove function nesting

plextraktsync/decorators/time_limit.py

@@ -1,22 +1,16 @@
-from functools import wraps
-
 from plextraktsync.config import TRAKT_POST_DELAY
 from plextraktsync.util.Timer import Timer
+from decorator import decorator
+
 
 timer = Timer(TRAKT_POST_DELAY)
 
 
-def time_limit():
+@decorator
+def time_limit(fn, *args, **kwargs):
     """
     Throttles calls not to be called more often than TRAKT_POST_DELAY
     """
+    timer.wait_if_needed()
 
-    def decorator(fn):
-        @wraps(fn)
-        def wrapper(*args, **kwargs):
-            timer.wait_if_needed()
-            return fn(*args, **kwargs)
-
-        return wrapper
-
-    return decorator
+    return fn(*args, **kwargs)