from flask import (Blueprint, render_template, request,

                   redirect, url_for, current_app)

from ennstatus.donate.forms import DateForm

from ennstatus.donate.functions import load_csv, get_choices

from ennstatus.root.forms import BPMForm

from ennstatus.root.constants import BPM_ADDRESSES

donate_page = Blueprint('donate', __name__)

def index():

    current_app.logger.info('Handling snailmail')

    form = BPMForm()

    country_choices = [choice[0] for choice in form.country.choices]

    if request.method == 'POST':

        current_app.logger.debug('Validating form')

        if form.validate_on_submit():

            country = form.country.data

            return redirect(url_for('donate.snailmail', country=country))

    else:

        if 'country' in request.args:

            country = request.args['country']

            if country in country_choices:

                current_app.logger.info('Showing country %s' % country)

            else:

                current_app.logger.warn('Country %s not found' % country)

                country = 'luxembourg'

        else:

            current_app.logger.info('Using default country')

            country = 'luxembourg'

    form.country.data = country

    address = BPM_ADDRESSES[country]

@donate_page.route('/received',

                   methods=('GET', 'POST'))

def received():

    current_app.logger.info('Handling received')

    form = DateForm()

    current_app.logger.debug('Creating choices')

    files = [name for name in get_choices()]

    files.sort()

    year_choices = list({name.split('-')[0] for name in files})

    year_choices.sort()

    form.year.choices = [(name, name) for name in year_choices]

    if not year_choices:

        current_app.logger.warn('No donations found!')

        return render_template('donate/received.html',

                               form=form, csv_file=None,

                               year=None, month=None)