Exctract decorators to separate file

Author: insspb

flask_mongoengine/db_fields.py

@@ -41,8 +41,6 @@
     "URLField",
     "UUIDField",
 ]
-import functools
-import logging
 from typing import Callable, List, Optional, Union
 
 from mongoengine import fields
@@ -54,22 +52,6 @@
     wtf_fields = None
     wtf_validators = None
 
-logger = logging.getLogger("flask_mongoengine")
-
-
-def wtf_required(func):
-    """Special decorator to warn user on incorrect installation."""
-
-    @functools.wraps(func)
-    def wrapped(*args, **kwargs):
-        if wtf_validators is None or wtf_fields is None:
-            logger.error(f"WTForms not installed. Function '{func.__name__}' aborted.")
-            return None
-
-        return func(*args, **kwargs)
-
-    return wrapped
-
 
 class WtfFieldMixin:
     """

flask_mongoengine/decorators.py

@@ -0,0 +1,25 @@
+import functools
+import logging
+
+try:
+    import wtforms  # noqa
+
+    wtf_installed = True
+except ImportError:  # pragma: no cover
+    wtf_installed = False
+
+logger = logging.getLogger("flask_mongoengine")
+
+
+def wtf_required(func):
+    """Special decorator to warn user on incorrect installation."""
+
+    @functools.wraps(func)
+    def wrapped(*args, **kwargs):
+        if not wtf_installed:
+            logger.error(f"WTForms not installed. Function '{func.__name__}' aborted.")
+            return None
+
+        return func(*args, **kwargs)
+
+    return wrapped