#!/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
"""
import datetime

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 Wizard(SessionWizard):
    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,
                                'current_step':stp})
            previous_steps.append(stp)
        return Context({'previous_steps':previous_steps})

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

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

class FileForm1(forms.Form):
    in_charge = forms.IntegerField(label=_("Person in charge"))
    year = forms.IntegerField(label=_("Year"),
                              initial=lambda:datetime.datetime.now().year)
    internal_reference = forms.CharField(label=_(u"Internal reference"),
                                         max_length=60)
    creation_date = forms.DateField(label=_(u"Creation date"),
                                     initial=datetime.datetime.now)
    file_type = forms.ChoiceField(label=_("File type"),
                    choices=models.FileType.objects.filter(available=True))
    comment = forms.CharField(label=_(u"Comment"), widget=forms.Textarea)


class FileForm2(forms.Form):
    town = forms.IntegerField(label=_(u"Town"))