import psycopg2
import termcolor
import sys

if len(sys.argv) > 1:
    option = sys.argv[1]
else:
    print("1 - Resumo\n2 - Cigarros em falta/acabando\n3 - Reposicao")
    option = raw_input()

conn = psycopg2.connect('dbname=cigarro host=10.40.140.25 user=postgres password=postgres')
cur = conn.cursor()

connmatriz = psycopg2.connect('dbname=matrizpitstop host=10.40.140.25 user=postgres password=postgres')
curmatriz = connmatriz.cursor()

curmatriz.execute("SELECT codigo_barra, nome, preco_unit, grid FROM produto WHERE grupo = 12912 "
                  " order by nome")

table_produto = {}

for a in curmatriz:
    table_produto[a[0]] = a[1:]

cur.execute('SELECT produto, estoque, quant, '
            '   last_turn, '
            '   (select empresa from estoque where estoque = id) '
            'FROM saldos '
            'WHERE estoque > 99')

table_saldos = {}

for a in cur:
    table_saldos[a[0]] = a[1:]

for produto in table_saldos.items():
    curmatriz.execute("SELECT quantidade FROM lancto WHERE "
                      " hora > '%s' and operacao in ('C', 'V', 'DF') and "
                      " produto = %d " %
                      (produto[1][2], table_produto[produto[0]][2]))

    lbuffer = curmatriz.fetchone()

    if lbuffer is None:
        venda = 0
    else:
        venda = lbuffer[0]

    saldo = produto[1][1] - venda

    if produto[0] in table_produto:
        nome_produto = table_produto[produto[0]][0]
    else:
        nome_produto = "Produto invalido"

    outp = "(%s @ %d) %s\tAnterior = %d\tVendas = %d\tAtual = %d" % \
           (produto[0].ljust(8), produto[1][0], nome_produto.ljust(60), produto[1][1], venda, saldo)

    if option == '1':
        if saldo < 5:
            outp = termcolor.colored(outp, 'red')

        print(outp)

    if option == '2':
        if saldo < 10:
            print(outp)


    if option == '3':
        if saldo < 10:
            cur.execute("SELECT quant FROM saldos WHERE "
                        " produto = '%s' and estoque = %d" %
                        (produto[0], table_saldos[produto[0]][0] / 100))

            lbuffer = cur.fetchone()

            if lbuffer is not None:
                outp += termcolor.colored("/%d" % lbuffer[0], 'yellow')
            else:
                outp += termcolor.colored("/?", 'red')

            curmatriz.execute("SELECT sum(quantidade) FROM lancto WHERE "
                              " produto = (select grid from produto where codigo_barra = '%s') "
                              " AND data >= current_timestamp - interval '15' day "
                              " AND operacao in ('C', 'V', 'DF') " % (produto[0]))

            lbuffer = curmatriz.fetchone()

            outp += termcolor.colored("\t%.1f/dia" % (float(lbuffer[0]) / 15), 'green')

            print(outp)