-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathTodoApp.py
94 lines (80 loc) · 3.51 KB
/
TodoApp.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
import sys
from PyQt5.QtWidgets import QApplication, QMainWindow, QPushButton,QTableWidgetItem
from PyQt5 import uic,QtWidgets
import sqlite3
class WindowTodo(QMainWindow):
def __init__(self):
super().__init__()
uic.loadUi('UiMain.ui',self)
self.state="new"
self.con = sqlite3.connect('todo.db')
self.cur = self.con.cursor()
self.btnSave.setEnabled(False)
self.btnDelete.setEnabled(False)
self.btnNew.clicked.connect(self.AddNew)
self.btnSave.clicked.connect(self.Save)
self.btnDelete.clicked.connect(self.Delete)
self.tblTodo.itemSelectionChanged.connect(self.selectedRow)
self.FillTable()
def AddNew(self):
self.state="new"
self.txtTitle.setText("")
self.txtDescription.setText("")
self.btnSave.setEnabled(True)
self.txtTitle.setFocus()
def Save(self):
widgetDate = self.dateDate.date()
selectedDate =str(widgetDate.day())+"/"+str(widgetDate.month())+"/"+str(widgetDate.year())
if(self.state=="new"):
self.cur.execute("insert into todo (title,description,todotime) values (?, ?, ?)", (self.txtTitle.text(), self.txtDescription.toPlainText(),selectedDate))
if(self.state=="edit"):
self.cur.execute("update todo set title=?,description=?,todotime=? where id=?", (self.txtTitle.text(), self.txtDescription.toPlainText(),selectedDate,self.id))
self.con.commit()
self.FillTable()
self.btnSave.setEnabled(False)
self.btnDelete.setEnabled(False)
def Delete(self):
try:
row = self.tblTodo.currentRow()
id = self.tblTodo.item(row, 0).text()
self.cur.execute("delete from todo where id=?",(id,))
self.con.commit()
self.FillTable()
self.btnSave.setEnabled(False)
self.btnDelete.setEnabled(False)
except:
print("error")
def FillTable(self):
self.cur.execute("SELECT * FROM todo")
rows = self.cur.fetchall()
self.tblTodo.setColumnCount(4)
self.tblTodo.setHorizontalHeaderLabels(['id','Title', 'Description','ToDoDate'])
self.tblTodo.setRowCount(0)
for row in rows:
inx = rows.index(row)
self.tblTodo.insertRow(inx)
# add more if there is more columns in the database.
self.tblTodo.setItem(inx, 0, QTableWidgetItem(str(row[0])))
self.tblTodo.setItem(inx, 1, QTableWidgetItem(str(row[1])))
self.tblTodo.setItem(inx, 2, QTableWidgetItem(str(row[2])))
self.tblTodo.setItem(inx, 3, QTableWidgetItem(str(row[3])))
header = self.tblTodo.horizontalHeader()
header.setSectionResizeMode(0, QtWidgets.QHeaderView.Stretch)
header.setSectionResizeMode(1, QtWidgets.QHeaderView.Stretch)
header.setSectionResizeMode(2, QtWidgets.QHeaderView.Stretch)
header.setSectionResizeMode(3, QtWidgets.QHeaderView.Stretch)
def selectedRow(self):
try:
self.state="edit"
row = self.tblTodo.currentRow()
self.id = self.tblTodo.item(row, 0).text()
self.txtTitle.setText(self.tblTodo.item(row, 1).text())
self.txtDescription.setText(self.tblTodo.item(row, 2).text())
self.btnSave.setEnabled(True)
self.btnDelete.setEnabled(True)
except:
print("error")
app=QApplication(sys.argv)
window = WindowTodo()
window.show()
sys.exit(app.exec_())