mirror of
https://github.com/olivierkes/manuskript.git
synced 2024-09-30 01:07:24 +13:00
112 lines
3.1 KiB
Python
112 lines
3.1 KiB
Python
#!/usr/bin/env python
|
|
# --!-- coding: utf8 --!--
|
|
|
|
|
|
|
|
# Because I have trouble with QDataWidgetMapper and the checkbox, I don't know why.
|
|
from PyQt5.QtCore import Qt
|
|
from PyQt5.QtWidgets import QCheckBox
|
|
|
|
from manuskript.enums import Outline
|
|
|
|
|
|
class chkOutlineCompile(QCheckBox):
|
|
def __init__(self, parent=None):
|
|
QCheckBox.__init__(self, parent)
|
|
self.stateChanged.connect(self.submit)
|
|
self._column = Outline.compile
|
|
self._index = None
|
|
self._indexes = None
|
|
self._model = None
|
|
self._updating = False
|
|
|
|
def setModel(self, model):
|
|
self._model = model
|
|
self._model.dataChanged.connect(self.update)
|
|
|
|
def setCurrentModelIndex(self, index):
|
|
self._indexes = None
|
|
if index.column() != self._column:
|
|
index = index.sibling(index.row(), self._column)
|
|
self._index = index
|
|
self.setTristate(False)
|
|
self.updateCheckState()
|
|
|
|
def setCurrentModelIndexes(self, indexes):
|
|
self._index = None
|
|
self._indexes = []
|
|
for i in indexes:
|
|
if i.column() != self._column:
|
|
i = i.sibling(i.row(), self._column)
|
|
self._indexes.append(i)
|
|
self.updateCheckState()
|
|
|
|
def getCheckedValue(self, index):
|
|
item = index.internalPointer()
|
|
return Qt.Checked if item.compile() else Qt.Unchecked
|
|
|
|
def update(self, topLeft, bottomRight):
|
|
|
|
if self._updating:
|
|
# We are currently putting data in the model, so no updates
|
|
return
|
|
|
|
if self._index:
|
|
if topLeft.row() <= self._index.row() <= bottomRight.row():
|
|
self.updateCheckState()
|
|
|
|
elif self._indexes:
|
|
update = False
|
|
|
|
for i in self._indexes:
|
|
if topLeft.row() <= i.row() <= bottomRight.row():
|
|
update = True
|
|
|
|
if update:
|
|
self.updateCheckState()
|
|
|
|
def updateCheckState(self):
|
|
|
|
if self._index:
|
|
self.setEnabled(True)
|
|
c = self.getCheckedValue(self._index)
|
|
self.setCheckState(c)
|
|
|
|
elif self._indexes:
|
|
self.setEnabled(True)
|
|
values = []
|
|
for i in self._indexes:
|
|
values.append(self.getCheckedValue(i))
|
|
|
|
same = True
|
|
for v in values[1:]:
|
|
if v != values[0]:
|
|
same = False
|
|
break
|
|
|
|
if same:
|
|
self.setCheckState(values[0])
|
|
else:
|
|
self._updating = True
|
|
self.setCheckState(Qt.PartiallyChecked)
|
|
self._updating = False
|
|
|
|
else:
|
|
self.setChecked(False)
|
|
self.setEnabled(False)
|
|
|
|
def submit(self, state):
|
|
|
|
if self._updating:
|
|
return
|
|
|
|
if self._index:
|
|
if self._model.data(self._index) != state:
|
|
self._model.setData(self._index, state)
|
|
|
|
elif self._indexes:
|
|
for i in self._indexes:
|
|
if self._model.data(i) != state:
|
|
self._model.setData(i, state)
|
|
|
|
self.setTristate(False)
|