-
Notifications
You must be signed in to change notification settings - Fork 1
/
textedit.py
81 lines (74 loc) · 2.96 KB
/
textedit.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
from PyQt5.QtCore import (QFile, QFileInfo, QIODevice,
QTextStream, Qt)
from PyQt5.QtWidgets import (QFileDialog, QMessageBox, QTextEdit)
class TextEdit(QTextEdit):
NextId = 1
def __init__(self, filename="", parent=None):
super(TextEdit, self).__init__(parent)
self.setAttribute(Qt.WA_DeleteOnClose)
self.filename = filename
if not self.filename:
self.filename = "Unnamed-{0}.txt".format(
TextEdit.NextId)
TextEdit.NextId += 1
self.document().setModified(False)
self.setWindowTitle(QFileInfo(self.filename).fileName())
def closeEvent(self, event):
if (self.document().isModified() and
QMessageBox.question(self,
"Text Editor - Unsaved Changes",
"Save unsaved changes in {0}?".format(self.filename),
QMessageBox.Yes | QMessageBox.No) ==
QMessageBox.Yes):
try:
self.save()
except EnvironmentError as e:
QMessageBox.warning(self,
"Text Editor -- Save Error",
"Failed to save {0}: {1}".format(self.filename, e))
def isModified(self):
return self.document().isModified()
def save(self):
if self.filename.startswith("Unnamed"):
filename, filetype = QFileDialog.getSaveFileName(self,
"Text Editor -- Save File As", self.filename,
"Text files (*.txt *.*)")
if not filename:
return
self.filename = filename
self.setWindowTitle(QFileInfo(self.filename).fileName())
exception = None
fh = None
try:
fh = QFile(self.filename)
if not fh.open(QIODevice.WriteOnly):
raise IOError(str(fh.errorString()))
stream = QTextStream(fh)
stream.setCodec("UTF-8")
stream << self.toPlainText()
self.document().setModified(False)
except EnvironmentError as e:
exception = e
finally:
if fh is not None:
fh.close()
if exception is not None:
raise exception
def load(self):
exception = None
fh = None
try:
fh = QFile(self.filename)
if not fh.open(QIODevice.ReadOnly):
raise IOError(str(fh.errorString()))
stream = QTextStream(fh)
stream.setCodec("UTF-8")
self.setPlainText(stream.readAll())
self.document().setModified(False)
except EnvironmentError as e:
exception = e
finally:
if fh is not None:
fh.close()
if exception is not None:
raise exception