import psycopg2
import os
from tkinter import *

janela = Tk()
janela.title("Concluir Produto")
janela.geometry("350x230+1000+470")

def update():
    if(str(numero_codigo.get()).isnumeric()):
        codigo = (numero_codigo.get())
        #Abrir conexão com o bd
        conexao = psycopg2.connect(user = "postgres", password = "password", port = "5432", database = "autosystem", host = "10.0.8.91")
        cursor = conexao.cursor()
        sql = "update comanda_produto set orcamento='1' where codigo= %s"
        codigo = str(codigo)
        cursor.execute(sql, (codigo,))
        conexao.commit()
        lb1["text"] = "Executado"
        #Fechar conexão com o bd
        cursor.close()
        conexao.close()

    else:
        lb1["text"] = "Valor inválido."

lb0 = Label(janela, text="Digite o código", font='Arial 15')
lb0.place(x=200, y=10)

numero_codigo = Entry(janela, font='Arial 20', width=10)
numero_codigo.grid(row=1, column=3)

lb1 = Label(janela, text="Aguardando", font='Arial 15')
lb1.place(x=200, y=100)

def numero0():
    numero_codigo.insert(INSERT, 0)
def numero1():
    numero_codigo.insert(INSERT, 1)
def numero2():
    numero_codigo.insert(INSERT, 2)
def numero3():
    numero_codigo.insert(INSERT, 3)
def numero4():
    numero_codigo.insert(INSERT, 4)
def numero5():
    numero_codigo.insert(INSERT, 5)
def numero6():
    numero_codigo.insert(INSERT, 6)
def numero7():
    numero_codigo.insert(INSERT, 7)
def numero8():
    numero_codigo.insert(INSERT, 8)
def numero9():
    numero_codigo.insert(INSERT, 9)
def delete():
    numero_codigo.delete(0, END)

lb_num0 = Button(janela, text="0", font='Arial 20', command=numero0)
lb_num1 = Button(janela, text="1", font='Arial 20', command=numero1)
lb_num2 = Button(janela, text="2", font='Arial 20', command=numero2)
lb_num3 = Button(janela, text="3", font='Arial 20', command=numero3)
lb_num4 = Button(janela, text="4", font='Arial 20', command=numero4)
lb_num5 = Button(janela, text="5", font='Arial 20', command=numero5)
lb_num6 = Button(janela, text="6", font='Arial 20', command=numero6)
lb_num7 = Button(janela, text="7", font='Arial 20', command=numero7)
lb_num8 = Button(janela, text="8", font='Arial 20', command=numero8)
lb_num9 = Button(janela, text="9", font='Arial 20', command=numero9)
lb_del = Button(janela, text="DEL", bg='red', font='Arial 20', command=delete)
lb_enter = Button(janela, text="ENT", bg='green', font='Arial 20', command=update)

lb_num0.grid(row=3, column=1)
lb_num1.grid(row=0, column=0)
lb_num2.grid(row=0, column=1)
lb_num3.grid(row=0, column=2)
lb_num4.grid(row=1, column=0)
lb_num5.grid(row=1, column=1)
lb_num6.grid(row=1, column=2)
lb_num7.grid(row=2, column=0)
lb_num8.grid(row=2, column=1)
lb_num9.grid(row=2, column=2)
lb_del.grid(row=3, column=0)
lb_enter.grid(row=3, column=2)

janela.mainloop()