spacepaste

Paste details

reply | raw

  1.  
  2. import sys
  3. from PyQt5.QtWidgets import QItemDelegate, QTreeWidget, QVBoxLayout, QLineEdit,\
  4. QMainWindow, QWidget, QTreeWidgetItem, QApplication
  5. from PyQt5.QtCore import QRegExp, Qt
  6. from PyQt5.QtGui import QRegExpValidator, QValidator
  8. class MyValidator(QValidator):
  9. def __init__(self, siblings, parent):
  10. super(MyValidator,self).__init__(parent)
  12. self.siblings = siblings
  14. def validate(self, s, pos):
  16. # look for invalid chars
  17. reg=QRegExp('[A-z0-9\[\]_-]+')
  18. regvd=QRegExpValidator(reg)
  20. regvd_rec=regvd.validate(s,pos)[0]
  21. print('regexp validate result:',regvd_rec)
  23. # check name conflicts
  24. if s in self.siblings:
  25. siblingvd_rec=0
  26. else:
  27. current_len=len(s)
  28. sub_strings=[sii[:current_len] for sii in self.siblings]
  29. if s in sub_strings:
  30. siblingvd_rec=1
  31. else:
  32. siblingvd_rec=2
  34. print('siblings validate result:',siblingvd_rec)
  36. if regvd_rec*siblingvd_rec==0:
  37. print('Invalid')
  38. return (QValidator.Invalid, s, pos)
  39. elif regvd_rec+siblingvd_rec==4:
  40. print('Acceptable')
  41. return (QValidator.Acceptable, s, pos)
  42. else:
  43. print('Intermediate')
  44. return (QValidator.Intermediate, s, pos)
  47. def fixup(self, s):
  48. pass
  52. class TreeWidgetDelegate(QItemDelegate):
  53. def __init__(self, siblings, parent=None):
  54. QItemDelegate.__init__(self, parent=parent)
  55. self.siblings=siblings
  56. self.parent=parent
  58. def createEditor(self, parent, option, index):
  59. editor = QLineEdit(parent)
  60. vd=MyValidator(self.siblings,self.parent)
  61. editor.setValidator(vd)
  62. return editor
  65. class MainWindow(QMainWindow):
  66. def __init__(self):
  67. super(self.__class__, self).__init__()
  68. frame=QWidget()
  69. self.setCentralWidget(frame)
  70. hl=QVBoxLayout()
  71. frame.setLayout(hl)
  73. self.tree=QTreeWidget(self)
  74. hl.addWidget(self.tree)
  76. # add treewidgetitems
  77. for ii in range(5):
  78. item=QTreeWidgetItem([str(ii)*3,])
  79. self.tree.addTopLevelItem(item)
  81. self.tree.itemDoubleClicked.connect(self.rename)
  83. self.show()
  85. def getSiblings(self,item):
  86. siblings=[self.tree.topLevelItem(ii).data(0,0) for ii in \
  87. range(self.tree.topLevelItemCount())]
  88. item_text=item.data(0,0)
  89. if item_text in siblings:
  90. siblings.remove(item_text)
  91. return siblings
  93. def rename(self):
  94. item=self.tree.selectedItems()
  95. if item:
  96. item=item[0]
  97. item.setFlags(item.flags() | Qt.ItemIsEditable)
  98. self.tree.scrollToItem(item)
  100. dele=TreeWidgetDelegate(self.getSiblings(item),self)
  101. self.tree.setItemDelegate(dele)
  103. self.tree.editItem(item)
  107. if __name__ == "__main__":
  108. app = QApplication(sys.argv)
  109. form = MainWindow()
  110. form.show()
  111. sys.exit(app.exec_())
  112.  
Powered by spacepaste. Paste removal.