from django.shortcuts import render
from formtools.wizard.views import SessionWizard
from django import forms
from django.core.exceptions import ValidationError
 
class PersonalForm(forms.Form):
    full_name = forms.CharField(max_length=100)
    email = forms.EmailField()
    account_type = forms.ChoiceField(choices=[('individual', 'Individual'), ('corporate', 'Corporate')])
 
 
class CorporateForm(forms.Form):
    company_name = forms.CharField(max_length=200)
    tax_id = forms.CharField(max_length=50)
    employee_count = forms.IntegerField(min_value=1)
 
 
class PreferencesForm(forms.Form):
    newsletter = forms.BooleanField(required=False)
    notifications = forms.BooleanField(required=False)
 
FORMS = [('personal', PersonalForm), ('corporate', CorporateForm), ('preferences', PreferencesForm)]
 
 
TEMPLATES = {'personal': 'wizard/personal.html', 'corporate': 'wizard/corporate.html', 'preferences': 'wizard/prefs.html'}
 
class MultiStepWizard(SessionWizardView):
    def get_template_names(self):
        return [TEMPLATES[self.steps.current]]
 
    def get_form_instance(self, step):
        if step == 'corporate':
            return self.instance_dict.get('corporate') or {}
        return {}
 
    def get_form_initial(self, step):
        initial = super().get_form_initial(step)
        if step == 'preferences':
            initial['newsletter'] = True
        return initial
 
    def done(self, form_list, **kwargs):
        data = {k: v for form in form_list for k, v in form.cleaned_data.items()}
        return render(self.request, 'wizard/done.html', {'data': data})