pygal/demo/cabaret/__init__.py

# -*- coding: utf-8 -*-
# This file is part of pygal
#
# A python svg graph plotting library
# Copyright © 2012-2016 Kozea
#
# This library is free software: you can redistribute it and/or modify it under
# the terms of the GNU Lesser General Public License as published by the Free
# Software Foundation, either version 3 of the License, or (at your option) any
# later version.
#
# This library 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 Lesser General Public License for more
# details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with pygal. If not, see <http://www.gnu.org/licenses/>.
from flask import Flask, render_template, request, jsonify
from pygal import CHARTS_BY_NAME
from pygal.graph import CHARTS_NAMES
from pygal.config import CONFIG_ITEMS
from pygal.interpolate import INTERPOLATIONS
from pygal.style import styles
from json import loads
from time import time


def create_app():
    """Creates the pygal test web app"""

    app = Flask(__name__)

    @app.route("/")
    def index():
        return render_template(
            'index.jinja2', charts_names=CHARTS_NAMES, configs=CONFIG_ITEMS,
            interpolations=INTERPOLATIONS, styles_names=styles.keys())

    @app.route("/svg", methods=('POST',))
    def svg():
        values = request.values
        config = loads(values['opts'])
        config['disable_xml_declaration'] = True
        config['style'] = styles[values['style']]
        config['js'] = []
        for item in CONFIG_ITEMS:
            value = config.get(item.name, None)
            if value:
                config[item.name] = item.coerce(value)
        chart = CHARTS_BY_NAME[values['type']](**config)
        for serie in loads(values['vals'])['vals']:
            chart.add(serie[0], serie[1])
        t = time()
        svg = chart.render()
        tt = int((time() - t) * 1000)
        return jsonify(svg=svg, time=tt)

    return app