habsi/authentik
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

stages/consent: add FlowPlan context variable for template name

Browse Source
This commit is contained in:
Jens Langhammer
2020-06-19 20:19:18 +02:00
parent b83aa44c4f
commit fa5c2bd85c
1 changed files with 11 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
passbook/stages/consent/stage.py
View File

@ -1,25 +1,30 @@
"""passbook consent stage""" """passbook consent stage"""
from typing import Any, Dict from typing import List, Dict, Any
from django.views.generic import FormView from django.views.generic import FormView
from passbook.flows.stage import StageView from passbook.flows.stage import StageView
from passbook.lib.utils.template import render_to_string
from passbook.stages.consent.forms import ConsentForm from passbook.stages.consent.forms import ConsentForm
PLAN_CONTEXT_CONSENT_TEMPLATE = "consent_template"
class ConsentStage(FormView, StageView): class ConsentStage(FormView, StageView):
"""Simple consent checker.""" """Simple consent checker."""
body_template_name: str
form_class = ConsentForm form_class = ConsentForm
def get_context_data(self, **kwargs: Dict[str, Any]) -> Dict[str, Any]: def get_context_data(self, **kwargs: Dict[str, Any]) -> Dict[str, Any]:
kwargs = super().get_context_data(**kwargs) kwargs = super().get_context_data(**kwargs)
if self.body_template_name: kwargs['current_stage'] = self.executor.current_stage
kwargs["body"] = render_to_string(self.body_template_name, kwargs) kwargs['context'] = self.executor.plan.context
return kwargs return kwargs
def get_template_names(self) -> List[str]:
if PLAN_CONTEXT_CONSENT_TEMPLATE in self.executor.plan.context:
template_name = self.executor.plan.context[PLAN_CONTEXT_CONSENT_TEMPLATE]
return [template_name]
return super().get_template_names()
def form_valid(self, form): def form_valid(self, form):
return self.executor.stage_ok() return self.executor.stage_ok()