add simple_tag 'stream_render' for posibility to add global context to blocks

Author: raagin

streamfield/base.py

@@ -60,7 +60,7 @@ def __str__(self):
     def __repr__(self):
         return "<%s: %s>" % (self.__class__.__name__, self or "None")
 
-    def _iterate_over_models(self, callback):
+    def _iterate_over_models(self, callback, tmpl_ctx=None):
         # iterate over models and apply callback function
         data = []
         for m in self.from_json():
@@ -87,13 +87,17 @@ def _iterate_over_models(self, callback):
                     as_list=as_list,
                     options=m['options']
                 )
+
+                # add tmpl_ctx if exists. tmpl_ctx: additional context from templates
+                if tmpl_ctx:
+                    ctx.update(tmpl_ctx)
                 res = callback(model_class, model_str, content, ctx)
                 data.append(res)
 
         return data
 
-    def _render(self):
-        data = self._iterate_over_models(_get_render_data)
+    def _render(self, tmpl_ctx=None):
+        data = self._iterate_over_models(_get_render_data, tmpl_ctx)
         return mark_safe("".join(data))
 
     @cached_property

streamfield/templatetags/streamfield_tags.py

@@ -38,8 +38,11 @@ def format_field(field):
         field=field
         )))
 
-
 def get_widget_name(field):
     return get_valid_filename(
                 camel_case_to_spaces(field.field.widget.__class__.__name__)
-                )
+                )
+
+@register.simple_tag
+def stream_render(stream_obj, **kwargs):
+    return stream_obj._render(kwargs)