import json

from flask import render_template, make_response

from . import bp

from app.models import Agent, Project, Service, Capacity, Period

from .. import db_mgr

@bp.route('/')
def index():

    return render_template('index.html', subtitle="Page d'accueil")


@bp.route('/services')
def services():
    # get services list
    all_services = Service.query.order_by(Service.name).all()
    num_services = len(all_services)
    # pass to template

    return render_template('services.html', subtitle="Liste des services ({})".format(num_services),

                           services=all_services)

@bp.route('/projects')
def projects():
    # get projects list
    all_projects = Project.query.order_by(Project.name).all()
    num_projects = len(all_projects)
    # pass to template

    return render_template('projects.html', subtitle="Liste des projets ({})".format(num_projects),

                           projects=all_projects)

@bp.route('/agents')
def agents():
    # get agents list
    all_agents = Agent.query.order_by(Agent.firstname).all()
    num_agents = len(all_agents)
    # pass to template

    return render_template('agents.html', subtitle="Liste des agents ({})".format(num_agents),

                           agents=all_agents)

@bp.route('/capacities')
def capacities():
    # get capacities list
    all_capacities = Capacity.query.order_by(Capacity.name).all()
    num_capacities = len(all_capacities)
    # pass to template
    return render_template('capacities.html', subtitle="Liste des fonctions ({})".format(num_capacities),
                           capacities=all_capacities)

@bp.route('/periods')
def periods():
    # get capacities list
    all_periods = Period.query.order_by(Period.name).all()
    num_periods = len(all_periods)
    # pass to template
    return render_template('periods.html', subtitle="Liste des périodes ({})".format(num_periods),
                           periods=all_periods)

@bp.route('/charge/add')
def charge_add():

    return render_template('charge.html', subtitle="Affecter un agent")


@bp.route('/charge/agent/<agent_id>')

def charge_agent(agent_id):

    agent_charges = []
    for [period, charge] in db_mgr.charges_by_agent(agent_id):
        agent_charges.append({"charge": charge, "periode": period})
    # agent_charges = [["charge", "periode"]]+db_mgr.charges_by_agent(agent_id)#agent_charges
    resp = make_response(json.dumps(agent_charges))

    resp.headers['Content-Type'] = 'application/json'
    return resp

@bp.route('/agent/<agent_id>')
def agent(agent_id):
    agent = Agent.query.get(int(agent_id))
    return render_template('agent.html',
                           agent=agent,

                           subtitle="{} {}".format(agent.firstname, agent.secondname))