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

from app import app
from forms import *

import psycopg2
import psycopg2.extras

from flask import render_template, request, redirect, url_for

try:
    conn = psycopg2.connect(dbname="transporte", host="localhost", user="postgres", password="123456")
    print "conectado"
except:
    print "falha ao conectar com o banco"

@app.route('/', methods=['POST', 'GET'])
@app.route('/login', methods=['POST', 'GET'])
def login():

    form = LoginForm()

    #POST
    if request.method == 'POST':
        Matricula = form['Matricula']
        Senha = form['Senha']

        cur = conn.cursor(cursor_factory=psycopg2.extras.DictCursor)

        cur.execute("SELECT * FROM usuario U WHERE '%s'=U.matricula and '%s'=U.senha"%(request.form['Matricula'],request.form['Senha']))

        usuarios = cur.fetchone()

        if usuarios:
            return render_template('index.html', title="IFPB Transportes")
        else:
            return render_template('usuario/login.html', title="Login", form=form, result="Senha ou usuario invalidos")

    # GET

    return render_template('usuario/login.html', title="Login", form=form)

@app.route('/ifpb_transportes')
def menu():
    return render_template('index.html', title='IFPB Transportes')

#usuario
@app.route('/add_usuario', methods=['GET', 'POST'])
def add_usuario():

    form = UsuarioForm()

    #POST
    if request.method == 'POST':
        Id_usuario = form['Id_usuario']
        Matricula = form['Matricula']
        Nome = form['Nome']
        CPF = form['CPF']
        Email = form['Email']
        Senha = form['Senha']

        cur = conn.cursor(cursor_factory=psycopg2.extras.DictCursor)
        cur.execute("INSERT INTO usuario VALUES ('%s', '%s', '%s', '%s', '%s', '%s')"%(request.form['Id_usuario'],request.form['Matricula'], request.form['Nome'], request.form['CPF'], request.form['Email'], request.form['Senha']))

        conn.commit()

        return render_template('usuario/create_usuario.html', title="Criar Conta", form=form)

    #GET
    return render_template('usuario/create_usuario.html', title="Criar Conta", form=form)

@app.route('/detal_usuario/<Id_usuario>')
def update_usuario(Id_usuario):

    cur = conn.cursor(cursor_factory=psycopg2.extras.DictCursor)
    cur.execute("""SELECT * FROM usuario WHERE Id_usuario = '%s'""" % Id_usuario)
    usuario = cur.fetchone()
    return render_template('usuario/update_usuario.html', usuario=usuario, title='Editar Usuario')

@app.route('/list_usuario')
def list_usuario():
    cur = conn.cursor(cursor_factory=psycopg2.extras.DictCursor)
    cur.execute("""SELECT * FROM Usuario WHERE status != 'Removido'""")
    usuarios = cur.fetchall()

    return render_template('usuario/list_usuario.html', title="Lista de Usuarios", usuarios=usuarios)

##ERRO
@app.route('/del_usuario/<Id_usuario>', methods = ['POST', 'GET'])
def delete(Id_usuario):

    if request.method == 'POST':
        cur = conn.cursor(cursor_factory=psycopg2.extras.DictCursor)
        cur.execute("""UPDATE usuario SET status = 'Removido' WHERE Id_usuario = '%s'""" % Id_usuario)
        conn.commit()

        return redirect(url_for('list_usuario'))

    return redirect(url_for('list_usuario'))
#veiculo
@app.route('/veiculo')
def veiculo():
    cur = conn.cursor(cursor_factory=psycopg2.extras.DictCursor)
    cur.execute("""SELECT * FROM Veiculo WHERE status != 'Removido'""")
    veiculos = cur.fetchall()

    return render_template('veiculo/list_veiculo.html', title="Lista de Veiculos", veiculos=veiculos)


@app.route('/add_veiculo', methods=['GET', 'POST'])
def add_veiculo():

    form = VeiculoForm()

    #POST
    if request.method == 'POST':
        Id_veiculo = form['Id_veiculo']
        Placa = form['Placa']
        Modelo = form['Modelo']
        Status = form['Status']


        cur = conn.cursor(cursor_factory=psycopg2.extras.DictCursor)
        cur.execute("INSERT INTO veiculo VALUES ('%s', '%s', '%s', '%s')"%(request.form['Id_veiculo'],
            request.form['Placa'], request.form['Modelo'], request.form['Status']))

        conn.commit()

        return render_template('veiculo/create_veiculo.html', title="Adicionar Veiculo", form=form)

    #GET
    return render_template('veiculo/create_veiculo.html', title="Adicionar Veiculo", form=form)


# @app.route('/detal_veiculo/<Id_veiculo>')
# def update_veiculo(Id_veiculo):

#     cur = conn.cursor(cursor_factory=psycopg2.extras.DictCursor)
#     cur.execute("""SELECT * FROM veiculo WHERE Id_veiculo = '%s'"""% Id_veiculo)
#     veiculo = cur.fetchone()
#     return render_template('veiculo/update_veiculo.html', veiculo=veiculo, title='Editar Veiculo')


##ERRO
################################
@app.route('/editar_veiculo/', methods=['POST'])
@app.route('/editar_veiculo/<id_veiculo>', methods=['GET'])
#################################
def editar_veiculo(id_veiculo):

    form = VeiculoForm()

    if request.method == 'POST':
        Id_veiculo = request.form['Id_veiculo']
        Placa = request.form['Placa']
        Modelo = request.form['Modelo']
        Status = request.form['Status']

        cur = conn.cursor(cursor_factory=psycopg2.extras.DictCursor)
        cur.execute("UPDATE veiculo SET id_veiculo = '%s', placa = '%s', modelo = '%s',  status = '%s' WHERE id_veiculo = '%s'"%(Id_veiculo, Placa, Modelo, Status, Id_veiculo))

        conn.commit()

        return redirect(url_for('veiculo'))

    cur = conn.cursor(cursor_factory=psycopg2.extras.DictCursor)
    cur.execute("""SELECT * FROM veiculo WHERE Id_veiculo = '%s'"""% id_veiculo)
    veiculo = cur.fetchone()
    return render_template('veiculo/update_veiculo.html', veiculo=veiculo, title='Editar Veiculo')

@app.route('/del_veiculo/<Id_veiculo>', methods = ['POST', 'GET'])
def delete_veiculo(Id_veiculo):

    if request.method == 'POST':
        cur = conn.cursor(cursor_factory=psycopg2.extras.DictCursor)
        cur.execute("""UPDATE veiculo SET status = 'Removido' WHERE Id_veiculo = '%s'""" % Id_veiculo)
        conn.commit()

        return redirect(url_for('veiculo'))

    return redirect(url_for('veiculo'))

#motorista
@app.route('/motorista')
def motorista():
    cur = conn.cursor(cursor_factory=psycopg2.extras.DictCursor)
    cur.execute("""SELECT * FROM Motorista""")
    motoristas = cur.fetchall()

    return render_template('motorista/list_motorista.html', title="Lista de Motoristas", motoristas=motoristas)

@app.route('/add_motorista', methods=['GET', 'POST'])
def add_motorista():

    form = MotoristaForm()

    #POST
    if request.method == 'POST':
        Id_motorista = form['Id_motorista']
        Matricula = form['Matricula']
        Nome = form['Nome']
        CNH = form['CNH']


        cur = conn.cursor(cursor_factory=psycopg2.extras.DictCursor)
        cur.execute("INSERT INTO motorista VALUES ('%s', '%s', '%s', '%s')"%(request.form['Id_motorista'],
            request.form['Matricula'], request.form['Nome'], request.form['CNH']))

        conn.commit()

        return render_template('motorista/create_motorista.html', title="Adicionar Motorista", form=form)

    #GET
    return render_template('motorista/create_motorista.html', title="Adicionar Motorista", form=form)

@app.route('/detal_motorista/<Id_motorista>')
def update_motorista(Id_motorista):

    cur = conn.cursor(cursor_factory=psycopg2.extras.DictCursor)
    cur.execute("""SELECT * FROM motorista WHERE Id_motorista = '%s'""" % Id_motorista)
    motorista = cur.fetchone()
    return render_template('motorista/update_motorista.html', motorista=motorista, title='Editar Motorista')


@app.route('/editar_motorista', methods=['GET', 'POST'])
def atualizar_motorista():
    return 'motorista atualizado'


@app.route('/delete_motorista')
def delete_motorista():
    cur = conn.cursor(cursor_factory=psycopg2.extras.DictCursor)
    cur.execute("""UPDATE veiculo SET status = 'Removido' WHERE Id_motorista = '%s'""" % Id_motorista)
    return redirect(url_for('motorista'))

#viagem
@app.route('/viagem')
def viagem():
    cur = conn.cursor(cursor_factory=psycopg2.extras.DictCursor)
    cur.execute("""SELECT * FROM viagem""")
    viagens = cur.fetchall()

    return render_template('viagem/list_viagem.html', title="Lista de Viagens", viagens=viagens)

##ERRO (não add)
@app.route('/add_viagem', methods=['GET', 'POST'])
def add_viagem():

    form = ViagemForm()

    #POST
    if request.method == 'POST':
        Id_viagem = form['Id_viagem']
        Cod_veiculo = form['Cod_veiculo']
        Cod_motorista = form['Cod_motorista']
        DATA_SAIDA = form['DATA_SAIDA']
        DATA_CHEGADA = form['DATA_CHEGADA']
        KM_SAIDA = form['KM_SAIDA']
        KM_CHEGADA = form['KM_CHEGADA']
        Soliciatante = form['Solicitante']
        Descricao = form['Descricao']

        cur = conn.cursor(cursor_factory=psycopg2.extras.DictCursor)
        cur.execute("INSERT INTO viagem VALUES ('%s', '%s', '%s', '%s', '%s', '%s','%s', '%s', '%s')"%
            (request.form['Id_viagem'], request.form['Cod_veiculo'], request.form['Cod_motorista'],
                request.form['DATA_SAIDA'], request.form['DATA_CHEGADA'],
                request.form['KM_SAIDA'], request.form['KM_CHEGADA'], request.form['Solicitante'],
                request.form['Descricao']))

        conn.commit()

        return render_template('viagem/create_viagem.html', title="Adicionar Viagem", form=form)

    return render_template('viagem/create_viagem.html', title="Adicionar Viagem", form=form)

@app.route('/detal_viagem/<Id_viagem>')
def update_viagem(Id_viagem):

    cur = conn.cursor(cursor_factory=psycopg2.extras.DictCursor)
    cur.execute("""SELECT * FROM viagem WHERE Id_viagem = '%s'""" % Id_viagem)
    viagem = cur.fetchone()
    return render_template('viagem/update_viagem.html', viagem=viagem, title='Editar Viagem')


@app.route('/editar_viagem', methods=['GET', 'POST'])
def atualizar_viagem():
    return 'viagem atualizada'


@app.route('/delete_viagem')
def delete_viagem():
    return 'viagens deletadas'

#revisao
@app.route('/revisao')
def revisao():
    cur = conn.cursor(cursor_factory=psycopg2.extras.DictCursor)
    cur.execute("""SELECT * FROM Revisao""")
    revisoes = cur.fetchall()

    return render_template('revisao/list_revisao.html', title="Lista de Revisoes", revisoes=revisoes)


@app.route('/add_revisao', methods=['GET', 'POST'])
def add_revisao():
    form = RevisaoForm()

    #POST
    if request.method == 'POST':
        Id_revisao = form['Id_revisao']
        Codigo = form['Codigo']
        Cod_veiculo = form['Cod_veiculo']
        Descricao = form['Descricao']
        DATA_REVISAO = form['DATA_REVISAO']


        cur = conn.cursor(cursor_factory=psycopg2.extras.DictCursor)
        cur.execute("INSERT INTO revisao VALUES ('%s', '%s', '%s', '%s', '%s')"%(request.form['Id_revisao'], request.form['Codigo'],
            request.form['Cod_veiculo'], request.form['Descricao'], request.form['DATA_REVISAO']))

        conn.commit()

        return render_template('revisao/create_revisao.html', title="Adicionar Revisao", form=form)

    #GET
    return render_template('revisao/create_revisao.html', title="Adicionar Revisao", form=form)

@app.route('/detal_revisao/<Id_revisao>')
def update_revisao(Id_revisao):

    cur = conn.cursor(cursor_factory=psycopg2.extras.DictCursor)
    cur.execute("""SELECT * FROM revisao WHERE Id_revisao = '%s'""" % Id_revisao)
    revisao = cur.fetchone()
    return render_template('revisao/update_revisao.html', revisao=revisao, title='Editar Revisao')


@app.route('/editar_revisao', methods=['GET', 'POST'])
def atualizar_revisao():
    return 'revisao atualizada'


@app.route('/del_revisao')
def delete_revisao():
    return 'revisao deletada'