python3+PyQt5泛型委托详解

#!/usr/bin/env python3import platformimport sysimport htmlfrom PyQt5.QtCore import QSize, Qt,pyqtSignalfrom PyQt5.QtGui import QColor, QFont,QFontMetrics, QIcon, QKeySequence, QPixmap,QTextCharFormatfrom PyQt5.QtWidgets import QAction,QApplication,QMenu,QTextEditclass RichTextLineEdit(QTextEdit): returnPressed=pyqtSignal() (Bold, Italic, Underline, StrikeOut, Monospaced, Sans, Serif,  NoSuperOrSubscript, Subscript, Superscript) = range(10) def __init__(self, parent=None):  super(RichTextLineEdit, self).__init__(parent)  self.monofamily = "courier"  self.sansfamily = "helvetica"  self.seriffamily = "times"  self.setLineWrapMode(QTextEdit.NoWrap)  self.setTabChangesFocus(True)  self.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOff)  self.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)  fm = QFontMetrics(self.font())  h = int(fm.height() * (1.4 if platform.system() == "Windows"         else 1.2))  self.setMinimumHeight(h)  self.setMaximumHeight(int(h * 1.2))  self.setToolTip("Press <b>Ctrl+M</b> for the text effects "    "menu and <b>Ctrl+K</b> for the color menu") def toggleItalic(self):  self.setFontItalic(not self.fontItalic()) def toggleUnderline(self):  self.setFontUnderline(not self.fontUnderline()) def toggleBold(self):  self.setFontWeight(QFont.Normal    if self.fontWeight() &gt; QFont.Normal else QFont.Bold) def sizeHint(self):  return QSize(self.document().idealWidth() + 5,      self.maximumHeight()) def minimumSizeHint(self):  fm = QFontMetrics(self.font())  return QSize(fm.width("WWWW"), self.minimumHeight()) def contextMenuEvent(self, event):  self.textEffectMenu() def keyPressEvent(self, event):  if event.modifiers() &amp; Qt.ControlModifier:   handled = False   if event.key() == Qt.Key_B:    self.toggleBold()    handled = True   elif event.key() == Qt.Key_I:    self.toggleItalic()    handled = True   elif event.key() == Qt.Key_K:    self.colorMenu()    handled = True   elif event.key() == Qt.Key_M:    self.textEffectMenu()    handled = True   elif event.key() == Qt.Key_U:    self.toggleUnderline()    handled = True   if handled:    event.accept()    return  if event.key() in (Qt.Key_Enter, Qt.Key_Return):   self.returnPressed.emit()   event.accept()  else:   QTextEdit.keyPressEvent(self, event) def colorMenu(self):  pixmap = QPixmap(22, 22)  menu = QMenu("Colour")  for text, color in (    ("&amp;Black", Qt.black),    ("B&amp;lue", Qt.blue),    ("Dark Bl&amp;ue", Qt.darkBlue),    ("&amp;Cyan", Qt.cyan),    ("Dar&amp;k Cyan", Qt.darkCyan),    ("&amp;Green", Qt.green),    ("Dark Gr&amp;een", Qt.darkGreen),    ("M&amp;agenta", Qt.magenta),    ("Dark Mage&amp;nta", Qt.darkMagenta),    ("&amp;Red", Qt.red),    ("&amp;Dark Red", Qt.darkRed)):   color = QColor(color)   pixmap.fill(color)   action = menu.addAction(QIcon(pixmap), text, self.setColor)   action.setData(color)  self.ensureCursorVisible()  menu.exec_(self.viewport().mapToGlobal(     self.cursorRect().center())) def setColor(self):  action = self.sender()  if action is not None and isinstance(action, QAction):   color = QColor(action.data())   if color.isValid():    self.setTextColor(color) def textEffectMenu(self):  format = self.currentCharFormat()  menu = QMenu("Text Effect")  for text, shortcut, data, checked in (    ("&amp;Bold", "Ctrl+B", RichTextLineEdit.Bold,     self.fontWeight() &gt; QFont.Normal),    ("&amp;Italic", "Ctrl+I", RichTextLineEdit.Italic,     self.fontItalic()),    ("Strike &amp;out", None, RichTextLineEdit.StrikeOut,     format.fontStrikeOut()),    ("&amp;Underline", "Ctrl+U", RichTextLineEdit.Underline,     self.fontUnderline()),    ("&amp;Monospaced", None, RichTextLineEdit.Monospaced,     format.fontFamily() == self.monofamily),    ("&amp;Serifed", None, RichTextLineEdit.Serif,     format.fontFamily() == self.seriffamily),    ("S&amp;ans Serif", None, RichTextLineEdit.Sans,     format.fontFamily() == self.sansfamily),    ("&amp;No super or subscript", None,     RichTextLineEdit.NoSuperOrSubscript,     format.verticalAlignment() ==     QTextCharFormat.AlignNormal),    ("Su&amp;perscript", None, RichTextLineEdit.Superscript,     format.verticalAlignment() ==     QTextCharFormat.AlignSuperScript),    ("Subs&amp;cript", None, RichTextLineEdit.Subscript,     format.verticalAlignment() ==     QTextCharFormat.AlignSubScript)):   action = menu.addAction(text, self.setTextEffect)   if shortcut is not None:    action.setShortcut(QKeySequence(shortcut))   action.setData(data)   action.setCheckable(True)   action.setChecked(checked)  self.ensureCursorVisible()  menu.exec_(self.viewport().mapToGlobal(     self.cursorRect().center())) def setTextEffect(self):  action = self.sender()  if action is not None and isinstance(action, QAction):   what = action.data()   if what == RichTextLineEdit.Bold:    self.toggleBold()    return   if what == RichTextLineEdit.Italic:    self.toggleItalic()    return   if what == RichTextLineEdit.Underline:    self.toggleUnderline()    return   format = self.currentCharFormat()   if what == RichTextLineEdit.Monospaced:    format.setFontFamily(self.monofamily)   elif what == RichTextLineEdit.Serif:    format.setFontFamily(self.seriffamily)   elif what == RichTextLineEdit.Sans:    format.setFontFamily(self.sansfamily)   if what == RichTextLineEdit.StrikeOut:    format.setFontStrikeOut(not format.fontStrikeOut())   if what == RichTextLineEdit.NoSuperOrSubscript:    format.setVerticalAlignment(      QTextCharFormat.AlignNormal)   elif what == RichTextLineEdit.Superscript:    format.setVerticalAlignment(      QTextCharFormat.AlignSuperScript)   elif what == RichTextLineEdit.Subscript:    format.setVerticalAlignment(      QTextCharFormat.AlignSubScript)   self.mergeCurrentCharFormat(format) def toSimpleHtml(self):  htmltext = ""  black = QColor(Qt.black)  block = self.document().begin()  while block.isValid():   iterator = block.begin()   while iterator != block.end():    fragment = iterator.fragment()    if fragment.isValid():     format = fragment.charFormat()     family = format.fontFamily()     color = format.foreground().color()          text=html.escape(fragment.text())     if (format.verticalAlignment() ==      QTextCharFormat.AlignSubScript):      text = "<sub>{0}</sub>".format(text)     elif (format.verticalAlignment() ==       QTextCharFormat.AlignSuperScript):      text = "<sup>{0}</sup>".format(text)     if format.fontUnderline():      text = "<u>{0}</u>".format(text)     if format.fontItalic():      text = "<i>{0}</i>".format(text)     if format.fontWeight() &gt; QFont.Normal:      text = "<b>{0}</b>".format(text)     if format.fontStrikeOut():      text = "<s>{0}</s>".format(text)     if color != black or family:      attribs = ""      if color != black:       attribs += ' color="{0}"'.format(color.name())      if family:       attribs += ' face="{0}"'.format(family)      text = "<font>{1}</font>".format(attribs,text)     htmltext += text    iterator += 1   block = block.next()  return htmltextif __name__ == "__main__": def printout(lineedit):  print(str(lineedit.toHtml()))  print(str(lineedit.toPlainText()))  print(str(lineedit.toSimpleHtml()))     app = QApplication(sys.argv) lineedit = RichTextLineEdit() lineedit.returnPressed.connect(lambda:printout(lineedit)) lineedit.show() lineedit.setWindowTitle("RichTextEdit") app.exec_()

#!/usr/bin/env python3from PyQt5.QtCore import (QDate, QSize, Qt)from PyQt5.QtWidgets import (QApplication, QDateEdit, QLineEdit,  QSpinBox, QStyledItemDelegate,QStyle)from PyQt5.QtGui import QColor,QTextDocumentimport richtextlineeditclass GenericDelegate(QStyledItemDelegate): def __init__(self, parent=None):  super(GenericDelegate, self).__init__(parent)  self.delegates = {} def insertColumnDelegate(self, column, delegate):  delegate.setParent(self)  self.delegates[column] = delegate def removeColumnDelegate(self, column):  if column in self.delegates:   del self.delegates[column] def paint(self, painter, option, index):  delegate = self.delegates.get(index.column())  if delegate is not None:   delegate.paint(painter, option, index)  else:   QStyledItemDelegate.paint(self, painter, option, index) def createEditor(self, parent, option, index):  delegate = self.delegates.get(index.column())  if delegate is not None:   return delegate.createEditor(parent, option, index)  else:   return QStyledItemDelegate.createEditor(self, parent, option,             index) def setEditorData(self, editor, index):  delegate = self.delegates.get(index.column())  if delegate is not None:   delegate.setEditorData(editor, index)  else:   QStyledItemDelegate.setEditorData(self, editor, index) def setModelData(self, editor, model, index):  delegate = self.delegates.get(index.column())  if delegate is not None:   delegate.setModelData(editor, model, index)  else:   QStyledItemDelegate.setModelData(self, editor, model, index)class IntegerColumnDelegate(QStyledItemDelegate): def __init__(self, minimum=0, maximum=100, parent=None):  super(IntegerColumnDelegate, self).__init__(parent)  self.minimum = minimum  self.maximum = maximum def createEditor(self, parent, option, index):  spinbox = QSpinBox(parent)  spinbox.setRange(self.minimum, self.maximum)  spinbox.setAlignment(Qt.AlignRight|Qt.AlignVCenter)  return spinbox def setEditorData(self, editor, index):  value = index.model().data(index, Qt.DisplayRole)  editor.setValue(value) def setModelData(self, editor, model, index):  editor.interpretText()  model.setData(index, editor.value())class DateColumnDelegate(QStyledItemDelegate): def __init__(self, minimum=QDate(),     maximum=QDate.currentDate(),     format="yyyy-MM-dd", parent=None):  super(DateColumnDelegate, self).__init__(parent)  self.minimum = minimum  self.maximum = maximum  self.format = format def createEditor(self, parent, option, index):  dateedit = QDateEdit(parent)  #dateedit=QDateTimeEdit(parent)  dateedit.setDateRange(self.minimum, self.maximum)  dateedit.setAlignment(Qt.AlignRight|Qt.AlignVCenter)  dateedit.setDisplayFormat(self.format)  dateedit.setCalendarPopup(True)  return dateedit def setEditorData(self, editor, index):  value = index.model().data(index, Qt.DisplayRole)  #if value.isNull:  editor.setDate(value)  #editor.setDisplayFormat(self.format) def setModelData(self, editor, model, index):  model.setData(index, editor.date()) def paint(self, painter, option, index):  text = index.model().data(index, Qt.DisplayRole).toString(self.format)  palette = QApplication.palette()  document = QTextDocument()  document.setDefaultFont(option.font)  if option.state &amp; QStyle.State_Selected:   document.setHtml("<font>{1}</font>".format(palette.highlightedText().color().name(),text))  else:   document.setHtml(text)  painter.save()  color = (palette.highlight().color()     if option.state &amp; QStyle.State_Selected     else QColor(index.model().data(index,        Qt.BackgroundColorRole)))  painter.fillRect(option.rect, color)  painter.translate(option.rect.x(), option.rect.y())  document.drawContents(painter)  painter.restore()  class PlainTextColumnDelegate(QStyledItemDelegate): def __init__(self, parent=None):  super(PlainTextColumnDelegate, self).__init__(parent) def createEditor(self, parent, option, index):  lineedit = QLineEdit(parent)  return lineedit def setEditorData(self, editor, index):  value = index.model().data(index, Qt.DisplayRole)  editor.setText(value) def setModelData(self, editor, model, index):  model.setData(index, editor.text())class RichTextColumnDelegate(QStyledItemDelegate): def __init__(self, parent=None):  super(RichTextColumnDelegate, self).__init__(parent) def paint(self, painter, option, index):  text = index.model().data(index, Qt.DisplayRole)  palette = QApplication.palette()  document = QTextDocument()  document.setDefaultFont(option.font)  if option.state &amp; QStyle.State_Selected:   document.setHtml("<font>{1}</font>".format(palette.highlightedText().color().name(),text))  else:   document.setHtml(text)  painter.save()  color = (palette.highlight().color()     if option.state &amp; QStyle.State_Selected     else QColor(index.model().data(index,        Qt.BackgroundColorRole)))  painter.fillRect(option.rect, color)  painter.translate(option.rect.x(), option.rect.y())  document.drawContents(painter)  painter.restore() def sizeHint(self, option, index):  text = index.model().data(index).toString()  document = QTextDocument()  document.setDefaultFont(option.font)  document.setHtml(text)  return QSize(document.idealWidth() + 5,      option.fontMetrics.height()) def createEditor(self, parent, option, index):  lineedit = richtextlineedit.RichTextLineEdit(parent)  return lineedit def setEditorData(self, editor, index):  value = index.model().data(index, Qt.DisplayRole)  editor.setHtml(value) def setModelData(self, editor, model, index):  model.setData(index, editor.toSimpleHtml())

#!/usr/bin/env python3import bisectimport osimport platformimport sysfrom PyQt5.QtCore import (QAbstractTableModel, QDate, QModelIndex,  QVariant, Qt,pyqtSignal)from PyQt5.QtWidgets import (QApplication, QMainWindow,  QShortcut, QTableView)from PyQt5.QtGui import QKeySequenceimport genericdelegates(LICENSE, CUSTOMER, HIRED, MILEAGEOUT, RETURNED, MILEAGEBACK, NOTES, MILEAGE, DAYS) = range(9)class CarHireLog(object): def __init__(self, license, customer, hired, mileageout,     returned=QDate(), mileageback=0, notes=""):  self.license = license  # plain text  self.customer = customer # plain text  self.hired = hired     # QDate  self.mileageout = mileageout   # int  self.returned = returned    # QDate  self.mileageback = mileageback  # int  self.notes = notes   # HTML def field(self, column):  if column == LICENSE:   return self.license  elif column == CUSTOMER:   return self.customer  elif column == HIRED:   return self.hired  elif column == MILEAGEOUT:   return self.mileageout  elif column == RETURNED:   return self.returned  elif column == MILEAGEBACK:   return self.mileageback  elif column == NOTES:   return self.notes  elif column == MILEAGE:   return self.mileage()  elif column == DAYS:   return self.days()  assert False def mileage(self):  return (0 if self.mileageback == 0     else self.mileageback - self.mileageout) def days(self):  return (0 if not self.returned.isValid()     else self.hired.daysTo(self.returned)) def __hash__(self):  return super(CarHireLog, self).__hash__() def __eq__(self, other):  if self.hired != other.hired:   return False  if self.customer != other.customer:   return False  if self.license != other.license:   return False  return id(self) == id(other) def __lt__(self, other):  if self.hired <b>damaged</b>",    "Returned with <i>empty fuel tank</i>",    "Customer <b>complained</b> about the <u>engine</u>",    "Customer <b>complained</b> about the <u>gears</u>",    "Customer <b>complained</b> about the <u>clutch</u>",    "Returned <font><b>dirty</b></font>",)  today = QDate.currentDate()  for i in range(250):   license = []   for c in range(5):    license.append(random.choice(string.ascii_uppercase))   license = ("".join(license[:2]) + random.choice(years) +      "".join(license[2:]))   customer = random.choice(titles) + random.choice(surnames)   hired = today.addDays(-random.randint(0, 365))   mileageout = random.randint(10000, 30000)   notes = ""   if random.random() &gt;= 0.2:    days = random.randint(1, 21)    returned = hired.addDays(days)    mileageback = (mileageout +        (days * random.randint(30, 300)))    if random.random() &gt; 0.75:     notes = random.choice(notetexts)   else:    returned = QDate()    mileageback = 0   log = CarHireLog(license, customer, hired, mileageout,        returned, mileageback, notes)   bisect.insort(self.logs, log) def rowCount(self, index=QModelIndex()):  return len(self.logs) def columnCount(self, index=QModelIndex()):  return 9 def data(self, index, role):  if not index.isValid():   return QVariant()  if role == Qt.DisplayRole:   log = self.logs[index.row()]   value = log.field(index.column())   if (index.column() in (MILEAGEBACK, MILEAGE, DAYS) and    value == 0):    return 0   return value  if (role == Qt.TextAlignmentRole and   index.column() not in (LICENSE, CUSTOMER, NOTES)):   return QVariant(int(Qt.AlignRight|Qt.AlignVCenter))  if role == Qt.BackgroundColorRole:   palette = QApplication.palette()   if index.column() in (LICENSE, MILEAGE, DAYS):    return QVariant(palette.alternateBase())   else:    return QVariant(palette.base())  return QVariant() def setData(self, index, value, role=Qt.EditRole):  if (index.isValid() and role == Qt.EditRole and   index.column() not in (LICENSE, MILEAGE, DAYS)):   log = self.logs[index.row()]   column = index.column()   if column == CUSTOMER:    log.customer = value   elif column == HIRED:    #log.hired = value.toDate()    log.hired = value   elif column == MILEAGEOUT:    log.mileageout = value   elif column == RETURNED:    #log.returned = value.toDate()    log.returned = value   elif column == MILEAGEBACK:    log.mileageback = value   elif column == NOTES:    log.notes = value   self.dataChanged[QModelIndex,QModelIndex].emit(index,index)   return True  return False def headerData(self, section, orientation, role):  if role == Qt.TextAlignmentRole:   if orientation == Qt.Horizontal:    return QVariant(int(Qt.AlignCenter))   return QVariant(int(Qt.AlignRight|Qt.AlignVCenter))  if role != Qt.DisplayRole:   return QVariant()  if orientation == Qt.Horizontal:   if section == LICENSE:    return "License"   elif section == CUSTOMER:    return "Customer"   elif section == HIRED:    return "Hired"   elif section == MILEAGEOUT:    return "Mileage #1"   elif section == RETURNED:    return "Returned"   elif section == MILEAGEBACK:    return "Mileage #2"   elif section == DAYS:    return "Days"   elif section == MILEAGE:    return "Miles"   elif section == NOTES:    return "Notes"  return section + 1 def flags(self, index):  flag = QAbstractTableModel.flags(self, index)  if index.column() not in (LICENSE, MILEAGE, DAYS):   flag |= Qt.ItemIsEditable  return flagclass HireDateColumnDelegate(genericdelegates.DateColumnDelegate): def createEditor(self, parent, option, index):  i = index.sibling(index.row(), RETURNED)  self.maximum = i.model().data(i, Qt.DisplayRole).addDays(-1)  return genericdelegates.DateColumnDelegate.createEditor(    self, parent, option, index)class ReturnDateColumnDelegate(genericdelegates.DateColumnDelegate): def createEditor(self, parent, option, index):  i = index.sibling(index.row(), HIRED)  self.minimum = i.model().data(i, Qt.DisplayRole).addDays(1)  return genericdelegates.DateColumnDelegate.createEditor(    self, parent, option, index)class MileageOutColumnDelegate(genericdelegates.IntegerColumnDelegate): def createEditor(self, parent, option, index):  i = index.sibling(index.row(), MILEAGEBACK)  maximum = i.model().data(i, Qt.DisplayRole)  self.maximum = 1000000 if maximum == 0 else maximum - 1  return genericdelegates.IntegerColumnDelegate.createEditor(    self, parent, option, index)class MileageBackColumnDelegate(genericdelegates.IntegerColumnDelegate): def createEditor(self, parent, option, index):  i = index.sibling(index.row(), MILEAGEOUT)  self.minimum = i.model().data(i, Qt.DisplayRole) + 1  return genericdelegates.IntegerColumnDelegate.createEditor(    self, parent, option, index)class MainForm(QMainWindow): def __init__(self, parent=None):  super(MainForm, self).__init__(parent)  model = CarHireModel(self)  self.view = QTableView()  self.view.setModel(model)  self.view.resizeColumnsToContents()  delegate = genericdelegates.GenericDelegate(self)  delegate.insertColumnDelegate(CUSTOMER,    genericdelegates.PlainTextColumnDelegate())  earliest = QDate.currentDate().addYears(-3)  delegate.insertColumnDelegate(HIRED,    HireDateColumnDelegate(earliest))  delegate.insertColumnDelegate(MILEAGEOUT,    MileageOutColumnDelegate(0, 1000000))  delegate.insertColumnDelegate(RETURNED,    ReturnDateColumnDelegate(earliest))  delegate.insertColumnDelegate(MILEAGEBACK,    MileageBackColumnDelegate(0, 1000000))  delegate.insertColumnDelegate(NOTES,    genericdelegates.RichTextColumnDelegate())  self.view.setItemDelegate(delegate)  self.setCentralWidget(self.view)  QShortcut(QKeySequence("Escape"), self, self.close)  QShortcut(QKeySequence("Ctrl+Q"), self, self.close)  self.setWindowTitle("Car Hire Logs")app = QApplication(sys.argv)form = MainForm()rect = QApplication.desktop().availableGeometry()form.resize(int(rect.width() * 0.7), int(rect.height() * 0.8))form.move(0, 0)form.show()app.exec_()