6jv4EVEN

1import tkinter as tk
2from tkinter import ttk  # библиотека с таблицами
3import sqlite3  # модуль базы данных
4
5
6class Main(tk.Frame):
7
8    def __init__(self, root):
9        super().__init__(root)
10        self.init_main()
11        self.db = db
12        self.view_records()
13
14    def init_main(self):
15        toolbar = tk.Frame(bg='#d7d8e0', bd=2)  # создание тул бара(меню, файл)
16        toolbar.pack(side=tk.TOP, fill=tk.X)  # прижимаем тулбар к верху и растягиваем по горизонтали
17
18        self.add_img = tk.PhotoImage(
19            file='add.gif')  # Добавляем картинку к кнопке, если картинка GIF лежит не в корневой папке, топ рописываем путь до нее полностью
20        btn_open_dialog = tk.Button(toolbar, text='Новый', command=self.open_dialog, bg='#d7d8e0', bd=0,
21                                    compound=tk.TOP, image=self.add_img)  # создание кнопки
22        btn_open_dialog.pack(side=tk.LEFT)  # выравнивание по левому краю
23
24        self.plus_img = tk.PhotoImage(file='plus.gif')
25        btn_plus_dialog = tk.Button(toolbar, text='Редактор', command=self.open_dialog, bg='#d7d8e0', bd=0,
26                                    compound=tk.TOP, image=self.plus_img)
27        btn_plus_dialog.pack(side=tk.LEFT)
28
29        self.vision_img = tk.PhotoImage(file='vision.gif')
30        btn_vision_dialog = tk.Button(toolbar, text='Просмотр', command=self.open_dialog, bg='#d7d8e0', bd=0,
31                                      compound=tk.TOP, image=self.vision_img)
32        btn_vision_dialog.pack(side=tk.LEFT)
33
34        self.import_img = tk.PhotoImage(file='import.gif')
35        btn_import_dialog = tk.Button(toolbar, text='Импорт', command=self.open_dialog, bg='#d7d8e0', bd=0,
36                                      compound=tk.TOP, image=self.import_img)
37        btn_import_dialog.pack(side=tk.LEFT)
38
39        self.delite_img = tk.PhotoImage(file='delite.gif')
40        btn_delite_dialog = tk.Button(toolbar, text='Удалить', command=self.delete_data, bg='#d7d8e0', bd=0,
41                                      compound=tk.TOP, image=self.delite_img)
42        btn_delite_dialog.pack(side=tk.LEFT)
43
44        self.total_img = tk.PhotoImage(file='total.gif')
45        btn_total_dialog = tk.Button(toolbar, text='Очистить БД', command=self.delete_all, bg='#d7d8e0', bd=0,
46                                     compound=tk.TOP, image=self.total_img)
47        btn_total_dialog.pack(side=tk.RIGHT)
48
49        self.tree = ttk.Treeview(self, columns=('Name', 'dots', 'Capital', 'Config', 'Gyro'), height=15,
50                                 show='headings')
51
52        self.tree.column('Name', width=129, anchor=tk.CENTER)
53        self.tree.column('dots', width=129, anchor=tk.CENTER)
54        self.tree.column('Capital', width=129, anchor=tk.CENTER)
55        self.tree.column('Config', width=129, anchor=tk.CENTER)
56        self.tree.column('Gyro', width=129, anchor=tk.CENTER)
57
58        self.tree.heading('Name', text='Имя проекта')
59        self.tree.heading('dots', text='Всего точек')
60        self.tree.heading('Capital', text='Всего опорных')
61        self.tree.heading('Config', text='Конфигурация хода')
62        self.tree.heading('Gyro', text='Всего гиросторон')
63
64        self.tree.pack()
65
66    def delete_data(self):
67        if len(self.tree.selection()) != 0:
68            row = self.tree.selection()[0]
69            name = self.tree.item(row)['values'][0]
70            self.db.c.execute("DELETE FROM matrix WHERE name = ?", (name,))
71            self.db.conn.commit()
72            #self.tree.delete(row)
73            self.view_records()
74
75    def records(self, name, dots, capital, config, gyro):
76        self.db.insert_data(name, dots, capital, config, gyro)
77        self.view_records()
78
79    def view_records(self):
80        self.db.c.execute('''SELECT * FROM matrix''')
81        [self.tree.delete(i) for i in self.tree.get_children()]
82        [self.tree.insert('', 'end', values=row) for row in self.db.c.fetchall()]
83
84    def delete_all(self):
85        self.db.c.execute('''DELETE FROM matrix''')
86        self.db.conn.commit()
87        self.view_records()
88
89    def open_dialog(self):
90        Child()
91
92
93class Child(tk.Toplevel):
94    def __init__(self):
95        super().__init__(root)
96        self.init_child()
97        self.view = app
98
99    def init_child(self):
100        self.title('Новый проект')
101        self.geometry('400x220+400+300')
102        self.resizable(False, False)
103
104        label_name = tk.Label(self, text='Название:')
105        label_name.place(x=50, y=20)
106        label_dots = tk.Label(self, text='Всего точек:')
107        label_dots.place(x=50, y=50)
108        label_config = tk.Label(self, text='Конфигурация хода:')
109        label_config.place(x=50, y=110)
110        label_gyro = tk.Label(self, text='Всего гиросторон:')
111        label_gyro.place(x=50, y=140)
112        label_capital = tk.Label(self, text='Всего опорных:')
113        label_capital.place(x=50, y=80)
114
115        self.entry_name = ttk.Entry(self)
116        self.entry_name.place(x=200, y=20)
117
118        self.entry_dots = ttk.Entry(self)
119        self.entry_dots.place(x=200, y=50)
120
121        self.entry_gyro = ttk.Entry(self)
122        self.entry_gyro.place(x=200, y=140)
123
124        self.entry_capital = ttk.Entry(self)
125        self.entry_capital.place(x=200, y=80)
126
127        self.combobox = ttk.Combobox(self, values=[u'Висячий', u'Разомкнутый', u'Замкнутый'])
128        self.combobox.current(0)
129        self.combobox.place(x=200, y=110)
130
131        btn_cancel = ttk.Button(self, text='Закрыть', command=self.destroy)
132        btn_cancel.place(x=300, y=170)
133
134        btn_ok = ttk.Button(self, text='Создать')
135        btn_ok.place(x=220, y=170)
136        btn_ok.bind('<Button-1>', lambda event: self.view.records(self.entry_name.get(),
137                                                                  self.entry_dots.get(),
138                                                                  self.entry_capital.get(),
139                                                                  self.combobox.get(),
140                                                                  self.entry_gyro.get()))  # срабатывает при нажатии левой кнопки мыши
141
142        self.grab_set()  # нельзя выполнять действия, пока не закроется окно второго уровня
143        self.focus_set()  # удержание окна на месте
144
145
146class DB:
147    def __init__(self):
148        self.conn = sqlite3.connect('matrix.db')
149        self.c = self.conn.cursor()
150        self.c.execute(
151            '''CREATE TABLE IF NOT EXISTS matrix (name text, dots integer, capital integer, config text, gyro integer)''')
152        self.conn.commit()
153
154    def insert_data(self, name, dots, capital, config, gyro):
155        self.c.execute('''INSERT INTO matrix (name, dots, capital, config, gyro) VALUES (?, ?, ?, ?, ?)''',
156                       (name, dots, capital, config, gyro))
157        self.conn.commit()
158
159
160if __name__ == "__main__":
161    root = tk.Tk()
162    db = DB()
163    app = Main(root)
164    app.pack()
165    root.title("Меню проектов")  # название окна
166    root.geometry("650x450+300+200")  # размеры окна и положение первой точки
167    root.resizable(False, False)  # не изменяется в размерах по вертикали  и по горизонтали
168    root.mainloop()