from flask import (Blueprint, render_template, request, current_app,

                   redirect, url_for)

stats_page = Blueprint('stats', __name__)

@stats_page.route('/')

def index():

    return render_template('stats/index.html')

@stats_page.route('/worldmap', methods=('GET', 'POST'))

def worldmap():

    current_app.logger.info('Handling worldmap')

    form = WorldmapStyleForm()

    style_choices = [choice[0] for choice in form.style.choices]

    server_choices = [choice[0] for choice in form.server_type.choices]

    if request.method == 'POST':

        current_app.logger.debug('Validating form')

        if form.validate_on_submit():

            style = form.style.data

            server_type = form.server_type.data

            return redirect(url_for('stats.worldmap', server_type=server_type,

                                    style=style))

    else:

        if 'style' in request.args:

            style = request.args['style']

            if style in style_choices:

                current_app.logger.info('Using style %s' % style)

            else:

                current_app.logger.warn('Style %s not found' % style)

                style = 'default'

        else:

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

            style = 'default'

        if 'server_type' in request.args:

            server_type = request.args['server_type']

            if server_type in server_choices:

                current_app.logger.info('Showing %s nodes' % server_type)

            else:

                current_app.logger.warn('Server type %s not found' % style)

                server_type = 'all'

        else:

            current_app.logger.info('Showing all servers')

            server_type = 'all'