from tkinter import *
import tkinter.ttk as ttk
import tkinter.messagebox as tkMessageBox
import sqlite3

def Database():
    global conn, cursor
    conn = sqlite3.connect("contactos.db")
    cursor = conn.cursor()
    cursor.execute(
        "CREATE TABLE IF NOT EXISTS REGISTRATION (RID INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL, NOME TEXT, NUMEROTELEFONICO TEXT)"
    )
    conn.commit()

def DisplayForm():
    root = Tk()
    root.geometry("900x500")
    root.title("Agenda de Contactos")
    global tree
    global SEARCH
    global nome, numero_telefonico
    SEARCH = StringVar()
    nome = StringVar()
    numero_telefonico = StringVar()
    TopViewForm = Frame(root, width=600, bd=1, relief=SOLID)
    TopViewForm.pack(side=TOP, fill=X)
    LFrom = Frame(root, width="350", bg="#15244C")
    LFrom.pack(side=LEFT, fill=Y)
    LeftViewForm = Frame(root, width=500, bg="#0B4670")
    LeftViewForm.pack(side=LEFT, fill=Y)
    MidViewForm = Frame(root, width=600)
    MidViewForm.pack(side=RIGHT)
    lbl_text = Label(TopViewForm, text="Agenda de Contactos", font=('verdana', 18), width=600, bg="cyan")
    lbl_text.pack(fill=X)
    Label(LFrom, text="Nome  ", font=("Arial", 12), bg="#15244C", fg="white").pack(side=TOP)
    Entry(LFrom, font=("Arial", 10, "bold"), textvariable=nome).pack(side=TOP, padx=10, fill=X)
    Label(LFrom, text="Número de Telefónico ", font=("Arial", 12), bg="#15244C", fg="white").pack(side=TOP)
    Entry(LFrom, font=("Arial", 10, "bold"), textvariable=numero_telefonico).pack(side=TOP, padx=10, fill=X)
    Button(LFrom, text="Submeter", font=("Arial", 10, "bold"), bg="#15244C", fg="white", command=register).pack(
        side=TOP, padx=10, pady=5, fill=X)

    lbl_txtsearch = Label(LeftViewForm, text="Digite o Nome", font=('verdana', 10), bg="#0B4670")
    lbl_txtsearch.pack()
    search = Entry(LeftViewForm, textvariable=SEARCH, font=('verdana', 15), width=10)
    search.pack(side=TOP, padx=10, fill=X)
    btn_search = Button(LeftViewForm, text="Pesquisa", bg="cyan", command=SearchRecord)
    btn_search.pack(side=TOP, padx=10, pady=10, fill=X)
    btn_view = Button(LeftViewForm, text="Ver tudo", bg="cyan", command=DisplayData)
    btn_view.pack(side=TOP, padx=10, pady=10, fill=X)
    btn_Limpar = Button(LeftViewForm, text="Limpar", bg="cyan", command=limpar)
    btn_Limpar.pack(side=TOP, padx=10, pady=10, fill=X)
    btn_apagar = Button(LeftViewForm, text="Apagar", bg="cyan", command=apagar)
    btn_apagar.pack(side=TOP, padx=10, pady=10, fill=X)
    btn_actualizar = Button(LeftViewForm, text="Actualizar", bg="cyan", command=actualizacao)
    btn_actualizar.pack(side=TOP, padx=10, pady=10, fill=X)
    scrollbarx = Scrollbar(MidViewForm, orient=HORIZONTAL)
    scrollbary = Scrollbar(MidViewForm, orient=VERTICAL)
    tree = ttk.Treeview(MidViewForm, columns=("Id", "Nome", "Numerotelefonico"),
                        selectmode="extended", height=100, yscrollcommand=scrollbary.set, xscrollcommand=scrollbarx.set)
    scrollbary.config(command=tree.yview)
    scrollbary.pack(side=RIGHT, fill=Y)
    scrollbarx.config(command=tree.xview)
    scrollbarx.pack(side=BOTTOM, fill=X)
    tree.heading('Id', text="Id", anchor=W)
    tree.heading('Nome', text="Nome", anchor=W)
    tree.heading('Numerotelefonico', text="Número telefónico", anchor=W)
    tree.column('#0', stretch=NO, minwidth=0, width=0)
    tree.column('#1', stretch=NO, minwidth=0, width=100)
    tree.column('#2', stretch=NO, minwidth=0, width=150)
    tree.pack()

    tree.bind("<Double-1>", OnDoubleClick)
    DisplayData()

def SearchRecord():
    Database()
    if SEARCH.get() != "":
        tree.delete(*tree.get_children())
        cursor = conn.execute("SELECT * FROM REGISTRATION WHERE NOME LIKE ?", ('%' + str(SEARCH.get()) + '%',))
        fetch = cursor.fetchall()
        for data in fetch:
            tree.insert('', 'end', values=(data))
        cursor.close()
        conn.close()

def register():
    Database()
    vnome = nome.get()
    vnumerotelefonico = numero_telefonico.get()
    if vnome == '' or vnumerotelefonico == '':
        tkMessageBox.showinfo("Atenção", "Preencha os espaços vazios!!!")
    else:
        conn.execute('INSERT INTO REGISTRATION (NOME, NUMEROTELEFONICO) VALUES (?, ?)', (vnome, vnumerotelefonico))
        conn.commit()
        tkMessageBox.showinfo("Sucesso", "Novo contacto adicionado com Sucesso")
        DisplayData()
        conn.close()

def actualizacao():
    Database()
    vnome = nome.get()
    vnumerotelefonico = numero_telefonico.get()
    if vnome == '' or vnumerotelefonico == '':
        tkMessageBox.showinfo("Atenção", "Preencha todos os espaços!!!")
    else:
        curItem = tree.focus()
        contents = (tree.item(curItem))
        selecteditem = contents['values']
        conn.execute('UPDATE REGISTRATION SET NOME=?, NUMEROTELEFONICO=? WHERE RID = ?', (vnome, vnumerotelefonico, selecteditem[0]))
        conn.commit()
        tkMessageBox.showinfo("Mensagem", "Actualizado com sucesso")
        limpar()
        DisplayData()
        conn.close()

def apagar():
    Database()
    if not tree.selection():
        tkMessageBox.showwarning("Atenção", "Selecione o contacto a eliminar")
    else:
        result = tkMessageBox.askquestion('Confirmar', 'Deseja apagar o registo do contacto?', icon="warning")
        if result == 'yes':
            curItem = tree.focus()
            contents = (tree.item(curItem))
            selecteditem = contents['values']
            tree.delete(curItem)
            cursor = conn.execute("DELETE FROM REGISTRATION WHERE RID = %d" % selecteditem[0])
            conn.commit()
            cursor.close()
            conn.close()

def limpar():
    tree.delete(*tree.get_children())
    DisplayData()
    SEARCH.set("")
    nome.set("")
    numero_telefonico.set("")

def DisplayData():
    Database()
    tree.delete(*tree.get_children())
    cursor = conn.execute("SELECT * FROM REGISTRATION")
    fetch = cursor.fetchall()
    for data in fetch:
        tree.insert('', 'end', values=(data))
    cursor.close()
    conn.close()

def OnDoubleClick(event):
    curItem = tree.focus()
    contents = (tree.item(curItem))
    selecteditem = contents['values']
    nome.set(selecteditem[1])
    numero_telefonico.set(selecteditem[2])

if __name__ == '__main__':
    DisplayForm()
    mainloop()