def headerData(self,
section: int,
orientation: QtCore.Qt.Orientation,
role: int = QtCore.Qt.DisplayRole):
if role == QtCore.Qt.DisplayRole:
if orientation == QtCore.Qt.Horizontal:
if self.header != None:
return self.header[section]
else:
# return str(self._dataframe.index[section])
return QtCore.QVariant()
return QtCore.QVariant()
def data(self, index, role):
if not index.isValid():
return QtCore.QVariant()
if role != QtCore.Qt.DisplayRole:
return QtCore.QVariant()
item = index.internalPointer()
value = item.data(index.column())
if not value:
return QtCore.QVariant()
return value
def data(self, index, role=QtCore.Qt.DisplayRole):
if not index.isValid() or not (0 <= index.row() < self.rowCount() \
and 0 <= index.column() < self.columnCount()):
return QtCore.QVariant()
row = self._dataframe.index[index.row()]
col = self._dataframe.columns[index.column()]
# print(type(self._dataframe[col]),col)
dt = self._dataframe[col].dtype
val = self._dataframe.iloc[row][col]
if role == QtCore.Qt.DisplayRole:
return str(val)
elif role == DataFrameModel.ValueRole:
return val
if role == DataFrameModel.DtypeRole:
return dt
return QtCore.QVariant()
def decode_data(self, arr):
data = []
item = {}
ds = QtCore.QDataStream(arr)
while not ds.atEnd():
ds.readInt32()
ds.readInt32()
map_items = ds.readInt32()
for i in range(map_items):
key = ds.readInt32()
value = QtCore.QVariant()
ds >> value
item[QtCore.Qt.ItemDataRole(key)] = value
data.append(item)
return data
def data(self, index, role):
if index.row() < len(self._data):
return self._data[index.row()]
return QtCore.QVariant()
def headerData(self, section, orientation, role):
if orientation == QtCore.Qt.Horizontal and role == QtCore.Qt.DisplayRole:
return self.rootItem.data(section)
return QtCore.QVariant()
