server.py

from flask import Flask, render_template, request, g, url_for, redirect
from unicodedata import normalize as normal
from database import AcessoBD

app = Flask(__name__)
dba = AcessoBD()


@app.route('/', methods=['GET', 'POST'])
def index():
    cams = dba.get_entes(ente='Câmara')
    prefs = dba.get_entes(ente='Prefeitura')
    return render_template('index.html', cams=cams, prefs=prefs)


@app.route('/sobre')
def sobre():
    return render_template('sobre.html')


@app.route('/consulta-basica', methods=['POST'])
def consulta_basica():
    req = request.form
    cidade = req['municipio'].lower().replace(' ', '-')
    cidade = None
    for municipio in g.municipios:
        if(str(municipio) == req['municipio']):
            cidade = municipio.nome_tratado
            break
    ente = req['entidade'].lower()
    ente = normal('NFKD', ente).encode('ASCII', 'ignore').decode('ASCII')
    return redirect('/dados/{}/{}'.format(cidade, ente))


@app.route('/dados/', defaults={'cidade': None, 'ente': None})
@app.route('/dados/<cidade>/<ente>')
def dados_abertos(cidade, ente):
    temp = None
    for municipio in g.municipios:
        if(municipio.nome_tratado == cidade):
            temp = municipio
            break

    cidade = temp

    if(ente in ['camara', 'prefeitura']):
        temp = {'camara': 'Câmara', 'prefeitura': 'Prefeitura'}
        ente = temp[ente]
    else:
        ente = None

    if(not cidade or not ente):
        return redirect(url_for('index'))

    lics = dba.get_licitacoes(municipio, ente)
    return render_template('dados-entidades.html', ente=ente,
                           municipio=municipio, lics=lics)


@app.before_request
def before_request():
    g.brand_name = 'Dados Abertos CE'
    g.municipios = dba.get_municipios()


if(__name__ == '__main__'):
    app.run()