geocode

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

from configparser import ConfigParser
from os import path, environ, getcwd, chdir


# Altera para diretório de trabalho
chdir(path.dirname(path.realpath(__file__)))

if path.exists('./config/wsgi.cfg'):
    config = ConfigParser()
    config.read('./config/wsgi.cfg')

    for key in config.options('env'):
        environ[key.upper()] = config.get('env', key)


# Adiciona diretório atual ao path.
import sys
sys.path.insert(0, path.dirname(path.realpath(__file__)))


# Carregar ambiente virtual se disponível.
if path.exists('./venv'):
    activate_this = path.join(getcwd(), './venv/bin/activate_this.py')
    with open(activate_this) as file_:
        exec(file_.read(), dict(__file__ = activate_this))


# Apaga base se executar da linha de comando.
if __name__ == '__main__':
    import click

    @click.command(name = 'geocode', add_help_option = False)
    @click.option('--arquivo', type=str, help = 'Executa em modo terminal com o arquivo especificado.')
    @click.option('--delimitador', default=';', help = 'Determina o limitador no arquivo CSV (modo terminal).')
    @click.option('--campos', default='*', help = 'Determina os campos do arquivo CSV a serem usados (modo terminal).')
    @click.option('--rebuild', is_flag = True, help = 'Reconstroi base de dados de busca.')
    @click.option('--server', is_flag = True, help = 'Executa servidor em modo DEBUG.')
    def comando(arquivo, delimitador, campos, rebuild, server):
        """Executar o GEOCODE em linha de comando para realizar diversas funções."""
        if arquivo:
            import terminal
            delimitador_ = delimitador if not delimitador is None else ';';
            campos_ = campos if not campos is None else '*';
            terminal.run(arquivo, delimitador_, campos_)
            click.get_current_context().exit()

        if rebuild:
            from modelos import apagar_base, criar_base
            apagar_base()
            criar_base()

        elif server:
            click.echo('Iniciando servidor localmente em modo DEBUG...')
            from app import create_app
            app = create_app()
            app.run(debug = True)

        else:
            click.echo(click.get_current_context().get_help())

    comando()

else:
    from app import create_app
    application = create_app()