relatorio cigarro atualizado

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 = {}

alternate = True

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),'
            '   (select nome 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 sum(quantidade) FROM lancto WHERE "
                      " hora > '%s' and operacao in ('C', 'V', 'DF') and "
                      " produto = %d "
                      " AND empresa = %d" %
                      (produto[1][2], table_produto[produto[0]][2], table_saldos[produto[0]][3]))

    lbuffer = curmatriz.fetchone()

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

    if venda is None:
        venda = 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] %s %s\tAnterior = %d\tVendas = %d\tAtual = %d" % \
           (table_saldos[produto[0]][4], produto[0].ljust(8), nome_produto.ljust(55), 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':
        curmatriz.execute("SELECT sum(quantidade) FROM lancto WHERE "
                          " produto = (select grid from produto where codigo_barra = '%s') "
                          " AND data > current_timestamp - interval '7' day "
                          " AND operacao in ('C', 'V', 'DF') "
                          " AND empresa = %d" % (produto[0], table_saldos[produto[0]][3]))

        lbuffer = curmatriz.fetchone()

        if lbuffer[0] is None:
            media = 0

            if saldo > 5:
                continue
        else:
            media = lbuffer[0] / 7

            if saldo > media:
                continue

        if alternate:
            outp = termcolor.colored(outp, 'white', attrs=['dark'])
            alternate = not alternate
        else:
            alternate = not alternate

        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')

        outp += termcolor.colored("\t%.1f/dia" % (media), 'green')

        print(outp)