|
10 | 10 | from django.template.loader import render_to_string |
11 | 11 | from django.utils.safestring import SafeText |
12 | 12 |
|
| 13 | +from unfold.components import ComponentRegistry |
| 14 | + |
13 | 15 | register = Library() |
14 | 16 |
|
15 | 17 |
|
@@ -154,24 +156,37 @@ def __init__( |
154 | 156 | template_name: str, |
155 | 157 | nodelist: NodeList, |
156 | 158 | extra_context: Optional[Dict] = None, |
| 159 | + include_context: bool = False, |
157 | 160 | *args, |
158 | 161 | **kwargs, |
159 | 162 | ): |
160 | 163 | self.template_name = template_name |
161 | 164 | self.nodelist = nodelist |
162 | 165 | self.extra_context = extra_context or {} |
| 166 | + self.include_context = include_context |
163 | 167 | super().__init__(*args, **kwargs) |
164 | 168 |
|
165 | 169 | def render(self, context: RequestContext) -> str: |
166 | | - result = self.nodelist.render(context) |
| 170 | + values = { |
| 171 | + name: var.resolve(context) for name, var in self.extra_context.items() |
| 172 | + } |
| 173 | + |
| 174 | + values.update( |
| 175 | + { |
| 176 | + "children": self.nodelist.render(context), |
| 177 | + } |
| 178 | + ) |
167 | 179 |
|
168 | | - ctx = {name: var.resolve(context) for name, var in self.extra_context.items()} |
169 | | - ctx.update({"children": result}) |
| 180 | + if "component_class" in values: |
| 181 | + values = ComponentRegistry.create_instance( |
| 182 | + values["component_class"], request=context.request |
| 183 | + ).get_context_data(**values) |
| 184 | + |
| 185 | + if self.include_context: |
| 186 | + values.update(context.flatten()) |
170 | 187 |
|
171 | 188 | return render_to_string( |
172 | | - self.template_name, |
173 | | - request=context.request, |
174 | | - context=ctx, |
| 189 | + self.template_name, request=context.request, context=values |
175 | 190 | ) |
176 | 191 |
|
177 | 192 |
|
@@ -202,17 +217,21 @@ def do_component(parser: Parser, token: Token) -> str: |
202 | 217 | raise TemplateSyntaxError( |
203 | 218 | '"with" in {bits[0]} tag needs at least one keyword argument.' |
204 | 219 | ) |
| 220 | + elif option == "include_context": |
| 221 | + value = True |
205 | 222 | else: |
206 | 223 | raise TemplateSyntaxError(f"Unknown argument for {bits[0]} tag: {option}.") |
207 | 224 |
|
208 | 225 | options[option] = value |
209 | 226 |
|
| 227 | + include_context = options.get("include_context", False) |
210 | 228 | nodelist = parser.parse(("endcomponent",)) |
211 | 229 | template_name = bits[1][1:-1] |
| 230 | + |
212 | 231 | extra_context = options.get("with", {}) |
213 | 232 | parser.next_token() |
214 | 233 |
|
215 | | - return RenderComponentNode(template_name, nodelist, extra_context) |
| 234 | + return RenderComponentNode(template_name, nodelist, extra_context, include_context) |
216 | 235 |
|
217 | 236 |
|
218 | 237 | @register.filter |
|
0 commit comments