add additional handling logic to template decorator

elastic · Jun 5, 2020 · 2f8210c · 2f8210c
1 parent c6ba0ea
commit 2f8210c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/scripts/generators/asciidoc_fields.py b/scripts/generators/asciidoc_fields.py
@@ -28,6 +28,10 @@ def decorator(func):
         @wraps(func)
         def decorated_function(*args, **kwargs):
             ctx = func(*args, **kwargs)
+            if ctx is None:
+                ctx = {}
+            elif not isinstance(ctx, dict):
+                return ctx
             return render_template(template_name, **ctx)
         return decorated_function
     return decorator