def on_editItem2(self):
"""
Command launched when 'Style' QPushButton is clicked
Auto assign color to all items
"""
super(Groups, self).on_editItem2()
N = len(pQt.getTopItems(self.tw_tree))
for n, item in enumerate(pQt.getTopItems(self.tw_tree)):
if n == 0:
rgb = (255, 0, 0)
elif n == (len(pQt.getTopItems(self.tw_tree)) - 1):
rgb = (0, 0, 255)
elif n < (N / 2):
rgb = (pMath.linear(0, (N / 2), 255, 0, n), pMath.linear(0, (N / 2), 0, 255, n), 0)
else:
rgb = (0, pMath.linear((N / 2), N, 255, 0, n), pMath.linear((N / 2), N, 0, 255, n))
self.ud_treeItem(item, grpColor=rgb)
if (not item in self.__editedItems__['added'] and not item in self.__editedItems__['edited'] and
not item in self.__editedItems__['deleted']):
self.__editedItems__['edited'].append(item)
self.rf_itemStyle()
def getData(self, asString=False):
"""
Get tree datas
:param asString: Return string instead of dict
:type asString: bool
:return: Tree datas
:rtype: dict
"""
data = dict()
for n, item in enumerate(pQt.getTopItems(self.tw_tree)):
data[n] = item.itemObj.getData()
if asString:
return pprint.pformat(data)
return data
def getData(self, asString=False):
"""
Get tree datas
:param asString: Return string instead of dict
:type asString: bool
:return: Tree datas
:rtype: dict
"""
data = dict()
for n, item in enumerate(pQt.getTopItems(self.tw_tree)):
data[n] = item.itemObj.getData()
if asString:
return pprint.pformat(data)
return data
