from tkinter import *
import tkinter as tk
from tkinter import ttk
from tkinter import scrolledtext
import sqlite3
import time
import threading

class Main(tk.Frame):
    def __init__(self, root):
        super().__init__(root)
        self.init_main()
        self.db = db
        self.view_records()
        self.flag = False


    def init_main(self):
        toolbar = tk.Frame(bg='#d7d8e0')
        toolbar.pack(side=tk.TOP, fill=tk.X)
        mainmenu = Menu(root)
        root.config(menu=mainmenu)
        self.add_img = tk.PhotoImage(file='add.png')
        btn_open_dialog = tk.Button(toolbar, text='Отправить заявку', command=self.open_dialog, bg='#d7d8e0', bd=0, compound=tk.TOP, image=self.add_img)
        btn_open_dialog.pack(side=tk.LEFT)
        self.update_img = tk.PhotoImage(file='update.png')
        btn_edit_dialog = tk.Button(toolbar, text='Редактировать', bg='#d7d8e0', bd=0, image=self.update_img, compound=tk.TOP, command=self.open_update_dialog)
        btn_edit_dialog.pack(side=tk.LEFT)
        self.delete_img = tk.PhotoImage(file='delete.png')
        btn_delete = tk.Button(toolbar, text='Удалить заявку', bg='#d7d8e0', bd=0, image=self.delete_img, compound=tk.TOP, command=self.delete_records)
        btn_delete.pack(side=tk.LEFT)
        self.refrash_img = tk.PhotoImage(file='refrash.png')
        btn_ref = tk.Button(toolbar, text='Автообновление', bg='#d7d8e0', bd=0, image=self.refrash_img, compound=tk.TOP, command=self.start_autorefrash)
        btn_ref.pack(side=tk.LEFT)
        label_info = tk.Label(toolbar, text='Тел: 62-20-42 (Влад, Никита)', font=('Arial Bold', 18), bg='#d7d8e0')
        label_info.pack(side=tk.LEFT)
        btn_delete = tk.Button(toolbar, text='Обновить', bg='#d7d8e0', bd=0, compound=tk.TOP, command=self.refrash, image=self.refrash_img)
        btn_delete.pack(side=tk.RIGHT)
        self.tree = ttk.Treeview(self, columns=('id', 'description', 'kabinet', 'data'), height=15, show='headings')
        self.tree.column('id', width=30, )
        self.tree.column('description', width=500, )
        self.tree.column('kabinet', width=100, )
        self.tree.column('data', width=150, )
        self.tree.heading('id', text='id')
        self.tree.heading('description', text='Текст обращения')
        self.tree.heading('kabinet', text='№ Кабинета')
        self.tree.heading('data', text='Время записи')
        self.tree.pack()

    def records(self, description, kabinet, data):
        self.db.insert_data(description, kabinet, data)
        self.view_records()

    def update_record(self, description, kabinet, data):
        self.db.c.execute('''UPDATE tiket SET description=?, kabinet=?, data=? WHERE id=?''', (description, kabinet, data, self.tree.set(self.tree.selection()[0], '#1'),))
        self.db.conn.commit()
        self.view_records()

    def start_autorefrash(self):
        t1 = threading.Thread(target=self.autorefrash)
        t1.start()


    def autorefrash(self):
        db = DB()
        self.flag = not self.flag
        while True:
            if self.flag:
                db.c.execute('''SELECT * FROM tiket''')
                [self.tree.delete(i) for i in self.tree.get_children()]
                [self.tree.insert('', 'end', values=row) for row in db.c.fetchall()]
                time.sleep(10)
            else:
                break


    def refrash(self):
        self.db.c.execute('''SELECT * FROM tiket''')
        self.db.conn.commit()
        self.view_records()


    def view_records(self):
        self.db.c.execute('''SELECT * FROM tiket''')
        [self.tree.delete(i) for i in self.tree.get_children()]
        [self.tree.insert('', 'end', values=row) for row in self.db.c.fetchall()]

    def delete_records(self):
        for selection_item in self.tree.selection():
            self.db.c.execute('''DELETE FROM tiket WHERE id=?''', (self.tree.set(selection_item, '#1'),))
        self.db.conn.commit()
        self.view_records()
    def open_dialog(self):
        Child()
    def open_update_dialog(self):
        Update()

class Child(tk.Toplevel):
    def __init__(self):
        super().__init__(root)
        self.init_child()
        self.view = app

    def init_child(self):
        self.title('Создание обращения')
        self.geometry('700x300')
        label_description = tk.Label(self, text='Описание проблемы:')
        label_description.place(x=50, y=50)
        label_kabinet = tk.Label(self, text='Кабинет:')
        label_kabinet.place(x=50, y=170)
        self.entry_description = scrolledtext.ScrolledText(self)
        self.entry_description.place(x=200, y=50, width=400, height=100)
        self.entry_kabinet = ttk.Entry(self)
        self.entry_kabinet.place(x=200, y=170)
        btn_cancel = ttk.Button(self, text='Закрыть', command=self.destroy)
        btn_cancel.place(x=300, y=200)
        self.btn_ok = ttk.Button(self, text='Отправить',command=self.destroy)
        self.btn_ok.place(x=220, y=200)
        self.btn_ok.bind('<Button-1>', lambda event: self.view.records(self.entry_description.get(1.0, 300.0),self.entry_kabinet.get(),time.strftime('%d-%m-%Y-%H:%M:%S')))
        self.grab_set()
        self.focus_set()


class Update(Child):
    def __init__(self):
        super().__init__()
        self.init_edit()
        self.view = app
    def init_edit(self):
        self.title('Редактировать')
        btn_edit = ttk.Button(self, text='Редактировать')
        btn_edit.place(x=200, y=200)
        btn_edit.bind('<Button-1>', lambda event: self.view.update_record(self.entry_description.get(1.0, 300.0),self.entry_kabinet.get(),time.strftime('%d-%m-%Y-%H:%M:%S')))
        self.btn_ok.destroy()


class DB:
    def __init__(self):
        self.conn = sqlite3.connect('database2.db')
        self.c = self.conn.cursor()
        self.c.execute('''CREATE TABLE IF NOT EXISTS tiket (id integer primary key, description text, kabinet text, data TIME)''')
        self.conn.commit()
    def insert_data(self, description, kabinet, data):
        self.c.execute('''INSERT INTO tiket (description, kabinet, data) VALUES (?, ?, ?)''', (description, kabinet, data))
        self.conn.commit()

if __name__ == '__main__':
    root = tk.Tk()
    db = DB()
    app = Main(root)
    app.pack()
    root.title('Тикеты на обращение')
    root.geometry('1000x400')
    root.mainloop()