# -*- coding: utf-8 -*-

import logging

import os

import codecs

import vobject

import StringIO

import qrcode

from docutils.core import publish_parts

from flask import Flask

from flask import render_template

from flask import request

from flask.ext.babel import gettext as _

from flask.ext.babel import Babel

from config import LANGUAGES

import base64

LANGUAGE_SELECTED = "de"

#ToDo after engelish is implemented set LANGUAGE_SELECTED = None

# We need the path of this file to find templates to translate

ESKP_PATH = os.path.dirname(os.path.abspath(__file__))

logging.basicConfig(filename=os.path.join(ESKP_PATH, 'eskp-app.log'),level=logging.DEBUG)

app = Flask(__name__)

babel = Babel(app)

app.config['BABEL_DEFAULT_LOCALE'] = 'de'

def get_vcard(filename):

    filename = os.path.join(ESKP_PATH, filename)

    with codecs.open(filename, 'r', 'utf-8') as f:

            vcard = f.read()

            f.close()

    return vobject.readOne(vcard)

def get_content(filename, overrides=None):

    content = u""

    filename = os.path.join(ESKP_PATH, filename)

    if os.path.isfile(filename):

        with codecs.open(filename, 'r', 'utf-8') as f:

            rst_data = f.read()

        f.close()

        content = publish_parts(rst_data, writer_name='html', settings_overrides=overrides)['html_body']

    return content

def get_topmenue():

    menue = [

             ('/ozoneloss', _(u'Ozoneloss'),

             (('/ozoneloss', _(u'overview')),

             ('/ozoneloss/clams/2016', _(u'calculations')),

             ('/ozoneloss/vpsc/2016', _(u'estimations')),

             ('/ozoneloss/uvi', _(u'uv increase')))),

             ('/eskp', _(u'ESKP'),(None, None)),

             ('/iek-7', _(u'IEK-7'),(None, None))

            ]

    return menue

app.jinja_env.globals.update(get_topmenue=get_topmenue)

def get_o3lossclams_dates():

    menue = [

        ('/ozoneloss/clams/2016', _(u'2016')),

        ('/ozoneloss/clams/2015', _(u'2015')),

        ('/ozoneloss/clams/2012', _(u'2012')),

        ('/ozoneloss/clams/2011', _(u'2011')),

        ('/ozoneloss/clams/2010', _(u'2010')),

        ]

    return menue

def get_vpsc_dates():

    menue = [

        ('/ozoneloss/vpsc/2016', _(u'2016')),

        ('/ozoneloss/vpsc/2015', _(u'2015')),

        ('/ozoneloss/vpsc/2014', _(u'2014')),

        ('/ozoneloss/vpsc/2013', _(u'2013')),

        ('/ozoneloss/vpsc/2012', _(u'2012')),

        ('/ozoneloss/vpsc/2011', _(u'2011')),

        ('/ozoneloss/vpsc/2010', _(u'2010')),

        ]

    return menue

app.jinja_env.globals.update(get_o3lossclams_dates=get_o3lossclams_dates)

app.jinja_env.globals.update(get_vpsc_dates=get_vpsc_dates)

def modal_info(template, act, title, filename):

    content = get_content(filename)

    html = render_template(template, act=act, title=title, content=content, exit=_(u"Close"))

    return html

@babel.localeselector

def get_locale():

    """ToDo: if translation is completed, switch to en """

    return LANGUAGE_SELECTED or request.accept_languages.best_match(LANGUAGES.keys()) or 'de'

@app.route("/")

@app.route("/index")

def index():

    return render_template("/index.html",

                           eskp_info=_(u'About ESKP'),

                           )

@app.route('/ozoneloss/clams/<year>')

def ozoneloss_clams_year(year):

    filename = os.path.join("templates", get_locale(), "rst", "ozoneloss_clams.rst")

    content = get_content(filename)

    return render_template("/ozoneloss_clams.html", act="ozoneloss/clams/%s" % year, content=content, year=year)

@app.route('/ozoneloss/vpsc/<year>')

def ozoneloss_vspc_year(year):

    filename = os.path.join("templates", get_locale(), "rst", "ozoneloss_vpsc.rst")

    content = get_content(filename)

    filename = os.path.join("templates", get_locale(), "rst", "explanation_vpsc.rst")

    explanation = get_content(filename)

    return render_template("/ozoneloss_vpsc.html", act="ozoneloss/vpsc/%s" % year, content=content,

                           content_explanation=explanation, year=year)

@app.route('/de')

def de():

    global LANGUAGE_SELECTED

    LANGUAGE_SELECTED = "de"

    return render_template("/index.html",

                           eskp_info=_(u'About ESKP'),

                           )

@app.route('/en')

def en():

    global LANGUAGE_SELECTED

    LANGUAGE_SELECTED = "en"

    return render_template("/index.html",

                           eskp_info=_(u'About ESKP'),

                           )

@app.route("/eskp")

def eskp():

    filename = os.path.join("templates", get_locale(), "rst", "eskp.rst")

    content = get_content(filename)

    filename = os.path.join("templates", get_locale(), "rst", "eskp_title.rst")

    headline = get_content(filename)

    return render_template("/eskp.html", act="eskp", content=content, headline=headline)

def qr_image_data(card):

    buf= StringIO.StringIO()

    qr = qrcode.QRCode(

        version=1,

        error_correction=qrcode.constants.ERROR_CORRECT_L,

        box_size=2,

        border=2,

        )

    qr.add_data(card.serialize())

    qr.make(fit=True)

    img = qr.make_image()

    img.save(buf)

    image = buf.getvalue()

    return base64.b64encode(image)

@app.route("/ozoneloss")

def ozoneloss():

    filename = os.path.join("templates", get_locale(), "rst", "ozoneloss.rst")

    content = get_content(filename)

    vcard_file = os.path.join("vcards", "jug.vcf")

    author = u""

    try:

        card = get_vcard(vcard_file)

    except IOError:

        card = None

    if card is not None:

        qr_image = qr_image_data(card)

        author = render_template("/author_info.html", act="author", title=_(u"Ozoneloss"),

                                  card=card, image=qr_image, contact=_(u"Contact"), exit=_(u"Close"))

    filename = os.path.join("templates", get_locale(), "rst", "ozoneloss_publications.rst")

    publications = modal_info("/publications_info.html", "publications", _(u"Ozoneloss"), filename)

    return render_template("/ozoneloss.html", act="ozoneloss", content=content,

                           author=author,card=card, publications=publications )

@app.route("/ozoneloss/clams")

def ozoneloss_clams():

    filename = os.path.join("templates", get_locale(), "rst", "ozoneloss_clams.rst")

    content = get_content(filename)

    return render_template("/ozoneloss_clams.html", act="ozoneloss/clams", content=content)

@app.route("/ozoneloss/vpsc")

def ozoneloss_vspc():

    filename = os.path.join("templates", get_locale(), "rst", "ozoneloss_vpsc.rst")

    content = get_content(filename)

    return render_template("/ozoneloss_vpsc.html", act="ozoneloss/vpsc", content=content)

@app.route("/ozoneloss/uvi", methods=['GET'])

def ozoneloss_uvi():

    # XXX check 'POST' does not work

    filename = os.path.join("templates", get_locale(), "rst", "ozoneloss_uvi.rst")

    content = get_content(filename)

    lat = 50.

    o3offset = 50.

    figname = "uvincr_lat%0.3i_do3%0.3i.svg" % (lat, o3offset)

    return render_template('ozoneloss_uvi.html', act="ozoneloss/uvi", content=content, figname=figname,

                           alt=_(u"UV increase at {{lat}} degrees N for {{o3offset}} DU ozone depletion"))

@app.route("/ozoneloss/uvi_graph", methods=['POST'])

def ozoneloss_uvi_graph():

    filename = os.path.join("templates", get_locale(), "rst", "ozoneloss_uvi.rst")

    content = get_content(filename)

    latstr = request.form['Gradzahl']

    o3offsetstr = request.form['Dobson-Unit']

    latstr2 = latstr.replace(u'\xb0', '')

    if latstr2.endswith(u'N'):

        latstr2 = latstr2.replace(u'N', '')

        lat = float(latstr2)

    if latstr2.endswith(u'S'):

        latstr2 = latstr2.replace(u'S', '')

        lat = -float(latstr2)

    figname = "uvincr_lat%0.3i_do3%0.3i.svg" % (lat, float(o3offsetstr))

    return render_template('graph.html', act="ozoneloss_uvi_graph", content=content, figname=figname,

                           o3offsetstr=o3offsetstr, latstr=latstr)

@app.route("/iek-7")

def institute():

    filename = os.path.join("templates", get_locale(), "rst", "iek-7.rst")

    content = get_content(filename)

    vcard_file = os.path.join("vcards", "sas.vcf")

    author = u""

    try:

        card = get_vcard(vcard_file)

    except IOError:

        card = None

    if card is not None:

        qr_image = qr_image_data(card)

        author = render_template("/author_info.html", act="author", title=_(u"IEK-7"),

                                  card=card, image=qr_image, contact=_(u"Contact"), exit=_(u"Close"))

    return render_template("/iek-7.html", act="iek-7", content=content,

                           author=author, card=card, contact=u"IEK-7")

@app.route("/imprint")

def imprint():

    filename = os.path.join("templates", get_locale(), "rst", "imprint.rst")

    content = get_content(filename)

    return render_template("/content.html", act="imprint", content=content)

@app.errorhandler(404)

def page_not_found(e):

    msg = _(u"Url: %(url)s not found", url=request.url)

    info = _(u"This information is not available!")

    return render_template("404.html", msg=msg, info=info)

if __name__ == "__main__":

    app.run(host='localhost', port=5014)