#!/usr/bin/env python
# -*- coding: utf-8 -*-
# Copyright (C) 2010  Étienne Loks  <etienne.loks_AT_peacefrogsDOTnet>

# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as
# published by the Free Software Foundation, either version 3 of the
# License, or (at your option) any later version.

# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.

# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

# See the file COPYING for details.

"""
Forms definition
"""
from django.utils.translation import ugettext_lazy as _
from django.template import Context
from django.shortcuts import render_to_response
from django import forms

from merlin.wizards.utils import Step as BasicStep
from merlin.wizards.session import SessionWizard

import models

class Step(BasicStep):
    """
    Add a label to steps
    """
    def __init__(self, idx, label, form):
        self.label = label
        super(Step, self).__init__(idx, form)

class FileForm1(forms.Form):
    subject = forms.CharField(max_length=100)
    sender = forms.EmailField()

class FileForm2(forms.Form):
    subject2 = forms.CharField(max_length=100)
    sender2 = forms.EmailField()

class FileWizard(SessionWizard):
    def get_template(self, request, step, form):
        return ['templates/file_wizard_%s.html' % step,
                'file_wizard.html']

    def process_show_form(self, request, step, form):
        previous_steps = []
        for stp in self.base_steps:
            if stp == step:
                return Context({'previous_steps':previous_steps})
            previous_steps.append(stp)
        return Context({'previous_steps':previous_steps})

    def done(self, request, form_list):
        return render_to_response('done.html', {
                        'form_data': [form.cleaned_data for form in form_list],
         })