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


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


def DisplayForm():
    root = Tk()
    root.geometry("900x500")
    root.title("Simulador da recomendação de filmes")

    global tree
    global SEARCH
    global nome, genero

    SEARCH = StringVar()
    nome = StringVar()
    genero = 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="Recomendar filmes", font=('verdana', 18), width=600, bg="cyan")
    lbl_text.pack(fill=X)

    Label(LFrom, text="Nome do Filme", 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="Género do Filme", font=("Arial", 12), bg="#15244C", fg="white").pack(side=TOP)
    genero.set("Género do Filme")
    content = {'Ação', 'Aventura', 'Animação', 'Biografia', 'Comédia', 'Comédia Romântica',
               'Comédia de Ação', 'Comédia de Terror', 'Crime', 'Documentário', 'Drama', 'Drama Histórico',
               'Drama Psicológico', 'Fantasia', 'Ficção Científica', 'Guerra', 'Histórico',
               'Musical', 'Mistério', 'Noir', 'Policial', 'Romance', 'Suspense', 'Terror', 'Thriller', 'Western'}
    OptionMenu(LFrom, genero, *content).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)
    Button(LFrom, text="Atualizar", font=("Arial", 10, "bold"), bg="#15244C", fg="white", command=actualizacao).pack(
        side=TOP, padx=10, pady=5, fill=X)

    lbl_txtsearch = Label(LeftViewForm, text="Digite Género", 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)

    # Configuração da Treeview
    scrollbarx = Scrollbar(MidViewForm, orient=HORIZONTAL)
    scrollbary = Scrollbar(MidViewForm, orient=VERTICAL)
    tree = ttk.Treeview(MidViewForm, columns=("RID", "Nome", "Género"), 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('RID', text="Id", anchor=W)
    tree.heading('Nome', text="Nome do Filme", anchor=W)
    tree.heading('Género', text="Género", anchor=W)
    tree.column('#0', stretch=NO, minwidth=0, width=0)
    tree.column('#1', stretch=NO, minwidth=0, width=150)
    tree.column('#2', stretch=NO, minwidth=0, width=150)

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

    DisplayData()


def register():
    Database()
    vnome = nome.get()
    vgenero = genero.get()

    if vnome == '' or vgenero == '':
        tkMessageBox.showinfo("Atenção", "Preencha os espaços vazios!!!")
    else:
        conn.execute('INSERT INTO REGISTRATION (NOME, GENERO) VALUES (?, ?)', (vnome, vgenero))
        conn.commit()
        tkMessageBox.showinfo("Sucesso", "Filme adicionado com sucesso")
        limpar()
        conn.close()


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 SearchRecord():
    Database()
    if SEARCH.get() != "":
        tree.delete(*tree.get_children())
        cursor = conn.execute("SELECT * FROM REGISTRATION WHERE GENERO LIKE ?", ('%' + str(SEARCH.get()) + '%',))
        fetch = cursor.fetchall()
        for data in fetch:
            tree.insert('', 'end', values=(data))
        cursor.close()
        conn.close()


def limpar():
    tree.delete(*tree.get_children())
    DisplayData()
    SEARCH.set("")
    nome.set("")
    genero.set("Género do Filme")


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


def actualizacao():
    Database()
    vnome = nome.get()
    vgenero = genero.get()

    if vnome == '' or vgenero == '':
        tkMessageBox.showinfo("Atenção", "Preencha todos os espaços!!!")
    else:
        curItem = tree.focus()
        if not curItem:
            tkMessageBox.showwarning("Atenção", "Selecione o filme a atualizar!")
            return

        contents = (tree.item(curItem))
        selecteditem = contents['values']

        conn.execute('UPDATE REGISTRATION SET NOME = ?, GENERO = ? WHERE RID = ?', (vnome, vgenero, selecteditem[0]))
        conn.commit()
        tkMessageBox.showinfo("Sucesso", "Filme atualizado com sucesso!")
        limpar()
        DisplayData()
        conn.close()


def OnDoubleClick(event):
    curItem = tree.focus()
    if not curItem:
        return

    contents = (tree.item(curItem))
    selecteditem = contents['values']

    nome.set(selecteditem[1])
    genero.set(selecteditem[2])


DisplayForm()

if __name__ == '__main__':
    mainloop()