from flask import jsonify
from flask_restful import Resource
from database import e
from json import dumps

class Demandas_Meta(Resource):
    def get(self):
        #Connect to database
        conn = e.connect()
        #Perform query and return JSON data
        query = conn.execute("select distinct PAIS from GENERACION_EOR_TR")
        result = query.cursor.fetchall()
        data = {'paises' : [row[0] for row in result]}
        return jsonify(data)

class Demandas(Resource):
    def get(self):
        #Connect to database
        conn = e.connect()
        #Perform query and return JSON data
        query = conn.execute("select * from DEMANDA_MER")
        columns = query.keys()
        data = []
        result = query.cursor.fetchall()
        for row in result:
            data.append(dict(zip(columns, row)))
        #List of Dicts to Dict of List (LD to DL)
        v = dict(zip(data[0], zip(*[d.values() for d in data])))
        response = jsonify({'data' : v})
        #Response is List of Dicts (LD)
        #response = jsonify({'data' : data})
        return response

class Demanda_Max(Resource):
    def get(self, pais):
        #Connect to database
        conn = e.connect()
        #Perform query and return JSON data
        query = conn.execute("SELECT fecha_hora, pais, demanda FROM generacion_eor_tr WHERE \
                            demanda = (select max(demanda) from generacion_eor_tr where pais = '{0}') \
                            and pais = '{0}'".format(pais.upper()))
        columns = query.keys()
        data = []
        result = query.cursor.fetchall()
        for row in result:
            data.append(dict(zip(columns, row)))
        response = jsonify({'data' : data})
        return response