main.py

import tkinter as tk
from tkinter import ttk
import sqlite3

# класс главного окна
class Main(tk.Frame):
    def __init__(self, root):
        super().__init__(root)
        self.init_main()
        self.db = db
        self.view_records()


    # поиск записи
    def search_records(self, name):
        name = ('%' + name + '%',)
        self.db.cur.execute(
            '''SELECT * FROM db WHERE name LIKE ?''', name)
        [self.tree.delete(i) for i in self.tree.get_children()]
        [self.tree.insert('', 'end', values=row)
         for row in self.db.cur.fetchall()]


    # метод отвечающий за вызов дочернего окна
    def open_dialog(self):
        Child()

    # метод отвечающий за вызов окна для изменения данных
    def open_update_dialog(self):
        Update()

    # метод отвечающий за вызов окна для поиска
    def open_search_dialog(self):
        Search()

    # вывод данных в виджет таблицы 
    def view_records(self): 
        # выбираем информацию из БД 
        self.db.cur.execute('''SELECT * FROM db''') 
        # удаляем все из виджета таблицы 
        [self.tree.delete(i) for i in self.tree.get_children()] 
        # добавляем в виджет таблицы всю информацию из БД 
        [self.tree.insert('', 'end', values=row) 
         for row in self.db.cur.fetchall()]

    # изменение (обновление) данных
    def update_record(self, name, tel, email, salary):
        self.db.cur.execute("""UPDATE db SET name=?, tel=?, email=?, salary=? WHERE id=?""", (name, tel, email, salary, self.tree.set(self.tree.selection()[0], '#1')))
        self.db.conn.commit()
        self.view_records()

    # удаление записей
    def delete_records(self):
        for selection_item in self.tree.selection():
            self.db.cur.execute("""DELETE FROM db WHERE
                                id=?""", (self.tree.set(selection_item, '#1'),))
            self.db.conn.commit()
            self.view_records()
            
    # добавление данных
    def records(self, name, tel, email, salary):
        self.db.insert_data(name, tel, email, salary)
        self.view_records()

    # хранение и инициализация объектов GUI
    def init_main(self):
        toolbar = tk.Frame(bg='#fff0f5', bd=2)
        toolbar.pack(side=tk.TOP, fill=tk.X)


        self.add_img = tk.PhotoImage(file='img/add.png') 
        # создание кнопки добавления 
        # command - функция по нажатию 
        # bg - фон 
        # bd - граница 
        # compound - ориентация текста (tk.CENTER , tk.LEFT , tk.RIGHT , tk.TOP или tk.BOTTOM.) 
        # image - иконка кнопки 
        btn_open_dialog = tk.Button(toolbar, bg='#d7d8e0', bd=1, image=self.add_img, command=self.open_dialog) 
        # упаковка и выравнивание по левому краю 
        btn_open_dialog.pack(side=tk.LEFT)


        # Добавляем Treeview 
        # columns - столбцы 
        # height - высота таблицы 
        # show='headings' скрываем нулевую (пустую) колонку таблицы 
        self.tree = ttk.Treeview(self, columns=('ID', 'name', 'tel', 'email', 'salary'), height=45, show='headings') 
 
 
        # добавляем параметры колонкам 
        # width - ширина 
        # anchor - выравнивание текста в ячейке 
        self.tree.column("ID", width=30, anchor=tk.CENTER) 
        self.tree.column("name", width=300, anchor=tk.CENTER) 
        self.tree.column("tel", width=150, anchor=tk.CENTER)
        self.tree.column("email", width=150, anchor=tk.CENTER)
        self.tree.column("salary", width=150, anchor=tk.CENTER) 
 
 
        # подписи колонок 
        self.tree.heading("ID", text='ID') 
        self.tree.heading("name", text='ФИО') 
        self.tree.heading("tel", text='Номер телефона') 
        self.tree.heading("email", text='E-mail') 
        self.tree.heading("salary", text='Заработная плата')
 
 
        # упаковка 
        self.tree.pack(side=tk.LEFT)

        # создание кнопки изменения данных
        self.update_img = tk.PhotoImage(file='img/update.png')
        btn_edit_dialog = tk.Button(toolbar, bg='#d7d8e0', bd=1, image=self.update_img, command=self.open_update_dialog)
        btn_edit_dialog.pack(side=tk.LEFT)

        # создание кнопки удаления записи
        self.delete_img = tk.PhotoImage(file='img/delete.png')
        btn_delete = tk.Button(toolbar, bg='#d7d8e0', bd=1, image=self.delete_img, command=self.delete_records)
        btn_delete.pack(side=tk.LEFT)

        # кнопка поиска
        self.search_img = tk.PhotoImage(file='img/search.png')
        btn_search = tk.Button(toolbar, bg='#d7d8e0', bd=1,
                               image=self.search_img, command=self.open_search_dialog)
        btn_search.pack(side=tk.LEFT)


        # кнопка обновления 
        self.refresh_img = tk.PhotoImage(file='img/refresh.png')
        btn_refresh = tk.Button(toolbar, bg='#d7d8e0', bd=1,
                               image=self.refresh_img, command=self.view_records)
        btn_refresh.pack(side=tk.LEFT)



# класс дочерних окон
# Toplevel - окно верхнего уровня
class Child(tk.Toplevel):
    def __init__(self):
        super().__init__(root)
        self.init_child()
        self.view = app

    def init_child(self):
        # заголовок окна
        self.title('Добавить')
        # размер окна
        self.geometry('400x200')
        # ограничение изменения размеров окна
        self.resizable(False, False)
        # перехватываем все события происходящие в приложении
        self.grab_set()
        # захватываем фокус
        self.focus_set()


        # подписи 
        label_name = tk.Label(self, text='ФИО:') 
        label_name.place(x=50, y=50) 
        label_select = tk.Label(self, text='Телефон') 
        label_select.place(x=50, y=80) 
        label_sum = tk.Label(self, text='E-mail') 
        label_sum.place(x=50, y=110)
        label_sal = tk.Label(self, text='Заработная плата') 
        label_sal.place(x=50, y=140) 
 
 
        # добавляем строку ввода для наименования 
        self.entry_name = ttk.Entry(self) 
        # меняем координаты объекта 
        self.entry_name.place(x=200, y=50) 
 
 
        # добавляем строку ввода для email 
        self.entry_email = ttk.Entry(self) 
        self.entry_email.place(x=200, y=80) 
 
 
        # добавляем строку ввода для телефона 
        self.entry_tel = ttk.Entry(self) 
        self.entry_tel.place(x=200, y=110) 
 
        # добавляем строку ввода для зарплаты 
        self.entry_salary = ttk.Entry(self) 
        self.entry_salary.place(x=200, y=140)
 
        # кнопка закрытия дочернего окна 
        self.btn_cancel = ttk.Button(self, text='Закрыть', command=self.destroy) 
        self.btn_cancel.place(x=300, y=170) 
 
 
        # кнопка добавления 
        self.btn_ok = ttk.Button(self, text='Добавить') 
        self.btn_ok.place(x=220, y=170) 
        # срабатывание по ЛКМ 
        self.btn_ok.bind('<Button-1>', lambda event:
                            self.view.records(
                                self.entry_name.get(),                                                                     
                                self.entry_email.get(),                                                       
                                self.entry_tel.get(),
                                self.entry_salary.get())) 

# класс окна для обновления, наследуемый от класса дочернего окна
class Update(Child):
    def __init__(self):
        super().__init__()
        self.init_edit()
        self.view = app
        self.db = db
        self.default_data()

    def init_edit(self):
        self.title('Редактировать позицию')
        btn_edit = ttk.Button(self, text='Редактировать')
        btn_edit.place(x=205, y=170)
        btn_edit.bind('<Button-1>', lambda event: self.view.update_record(self.entry_name.get(),
                                                                          self.entry_email.get(),
                                                                          self.entry_tel.get(),
                                                                          self.entry_salary.get()))

        # закрываем окно редактирования
        # add='+' позваляет на одну кнопку вешать более одного события
        btn_edit.bind('<Button-1>', lambda event: self.destroy(), add='+')
        self.btn_ok.destroy()

    def default_data(self):
        self.db.cur.execute('''SELECT * FROM db WHERE id=?''',
                          (self.view.tree.set(self.view.tree.selection()[0], '#1'),))
        # получаем доступ к первой записи из выборки
        row = self.db.cur.fetchone()
        self.entry_name.insert(0, row[1])
        self.entry_email.insert(0, row[2])
        self.entry_tel.insert(0, row[3])
        self.entry_salary.insert(0, row[4])



# класс поиска записи
class Search(tk.Toplevel):
    def __init__(self):
        super().__init__()
        self.init_search()
        self.view = app
    
    def init_search(self):
        self.title('Поиск')
        self.geometry('300x100')
        self.resizable(False, False)

        label_search = ttk.Label(self, text='Поиск')
        label_search.place(x=50, y=20)
        
        self.entry_search = ttk.Entry(self)
        self.entry_search.place(x=105, y=20, width=150)
        btn_cancel = ttk.Button(self, text='Закрыть', command=self.destroy)
        btn_cancel.place(x=185, y=50)

        btn_search = ttk.Button(self, text='Поиск')
        btn_search.place(x=105, y=50)
        btn_search.bind('<Button-1>', lambda event: self.view.search_records(self.entry_search.get()))
        btn_search.bind('<Button-1>', lambda event: self.destroy(), add='+')

# класс БД
class DB:
    def __init__(self):
        # создаем соединение с БД
        self.conn = sqlite3.connect("EmployeesList.db")
        # создание объекта класса cursor, используемый для взаимодействия с БД
        self.cur = self.conn.cursor()
        # выполнение запроса к БД
        self.cur.execute("""CREATE TABLE IF NOT EXISTS db(
        id INTEGER PRIMARY KEY,
        name TEXT,
        tel TEXT,
        email TEXT,
        salary TEXT);
        """)
        # сохранение изменений БД
        self.conn.commit()

    # метод добавления в БД
    def insert_data(self, name, tel, email, salary):
        self.cur.execute("""INSERT INTO db(name, tel, email, salary)
                         VALUES (?, ?, ?, ?)
                         """, (name, tel, email, salary))
        self.conn.commit()


if __name__ == '__main__':
    root = tk.Tk()
    # экземпляр класса DB
    db = DB()
    app = Main(root)
    app.pack()
    # заголовок окна
    root.title('Список сотрудников компании')
    # размер окна
    root.geometry('665x450')
    # ограничение изменения размеров окна
    root.resizable(False, False)
    root.mainloop()