17913cc586
Initial subscription fix. Still does not always display the list from the correct folder, and it currently does not update unread count on subscriptions.
236 lines
8.6 KiB
Python
236 lines
8.6 KiB
Python
from PyQt4 import QtCore, QtGui
|
|
|
|
from utils import *
|
|
import shared
|
|
|
|
class Ui_FolderWidget(QtGui.QTreeWidgetItem):
|
|
folderWeight = {"inbox": 1, "sent": 2, "trash": 3}
|
|
def __init__(self, parent, pos = 0, address = "", folderName = "", unreadCount = 0):
|
|
super(QtGui.QTreeWidgetItem, self).__init__()
|
|
self.address = address
|
|
self.folderName = folderName
|
|
self.unreadCount = unreadCount
|
|
parent.insertChild(pos, self)
|
|
self.updateText()
|
|
|
|
def setAddress(self, address):
|
|
self.address = str(address)
|
|
self.updateText()
|
|
|
|
def setUnreadCount(self, cnt):
|
|
self.unreadCount = int(cnt)
|
|
self.updateText()
|
|
|
|
def setFolderName(self, fname):
|
|
self.folderName = str(fname)
|
|
self.updateText()
|
|
|
|
def updateText(self):
|
|
text = QtGui.QApplication.translate("MainWindow", self.folderName)
|
|
font = QtGui.QFont()
|
|
if self.unreadCount > 0:
|
|
text += " (" + str(self.unreadCount) + ")"
|
|
font.setBold(True)
|
|
else:
|
|
font.setBold(False)
|
|
self.setFont(0, font)
|
|
self.setText(0, text)
|
|
self.setToolTip(0, text)
|
|
# self.setData(0, QtCore.Qt.UserRole, [self.address, self.folderName])
|
|
|
|
# inbox, sent, thrash first, rest alphabetically
|
|
def __lt__(self, other):
|
|
if (isinstance(other, Ui_FolderWidget)):
|
|
if self.folderName in self.folderWeight:
|
|
x = self.folderWeight[self.folderName]
|
|
else:
|
|
x = 4
|
|
if other.folderName in self.folderWeight:
|
|
y = self.folderWeight[other.folderName]
|
|
else:
|
|
y = 4
|
|
reverse = False
|
|
if self.treeWidget().header().sortIndicatorOrder() == QtCore.Qt.DescendingOrder:
|
|
reverse = True
|
|
if x == y:
|
|
return self.folderName < other.folderName
|
|
else:
|
|
return (x >= y if reverse else x < y)
|
|
|
|
return super(QtGui.QTreeWidgetItem, self).__lt__(other)
|
|
|
|
|
|
class Ui_AddressWidget(QtGui.QTreeWidgetItem):
|
|
def __init__(self, parent, pos = 0, address = "", unreadCount = 0):
|
|
super(QtGui.QTreeWidgetItem, self).__init__()
|
|
self.unreadCount = unreadCount
|
|
parent.insertTopLevelItem(pos, self)
|
|
# only set default when creating
|
|
#super(QtGui.QTreeWidgetItem, self).setExpanded(shared.config.getboolean(self.address, 'enabled'))
|
|
self.setAddress(address)
|
|
|
|
def setAddress(self, address):
|
|
self.address = str(address)
|
|
self.setType()
|
|
self.setExpanded(shared.safeConfigGetBoolean(self.address, 'enabled'))
|
|
self.updateText()
|
|
|
|
def setType(self):
|
|
if shared.safeConfigGetBoolean(self.address, 'chan'):
|
|
self.type = "chan"
|
|
elif shared.safeConfigGetBoolean(self.address, 'mailinglist'):
|
|
self.type = "mailinglist"
|
|
else:
|
|
self.type = "normal"
|
|
|
|
def setUnreadCount(self, cnt):
|
|
self.unreadCount = int(cnt)
|
|
self.updateText()
|
|
|
|
def updateText(self):
|
|
text = unicode(shared.config.get(self.address, 'label'), 'utf-8)') + ' (' + self.address + ')'
|
|
|
|
font = QtGui.QFont()
|
|
if self.unreadCount > 0:
|
|
# only show message count if the child doesn't show
|
|
if not self.isExpanded():
|
|
text += " (" + str(self.unreadCount) + ")"
|
|
font.setBold(True)
|
|
else:
|
|
font.setBold(False)
|
|
self.setFont(0, font)
|
|
|
|
#set text color
|
|
if shared.safeConfigGetBoolean(self.address, 'enabled'):
|
|
if shared.safeConfigGetBoolean(self.address, 'mailinglist'):
|
|
brush = QtGui.QBrush(QtGui.QColor(137, 04, 177))
|
|
else:
|
|
brush = QtGui.QBrush(QtGui.QApplication.palette().text().color())
|
|
#self.setExpanded(True)
|
|
else:
|
|
brush = QtGui.QBrush(QtGui.QColor(128, 128, 128))
|
|
#self.setExpanded(False)
|
|
brush.setStyle(QtCore.Qt.NoBrush)
|
|
self.setForeground(0, brush)
|
|
|
|
self.setIcon(0, avatarize(self.address))
|
|
self.setText(0, text)
|
|
self.setToolTip(0, text)
|
|
# self.setData(0, QtCore.Qt.UserRole, [self.address, "inbox"])
|
|
|
|
def setExpanded(self, expand):
|
|
super(Ui_AddressWidget, self).setExpanded(expand)
|
|
self.updateText()
|
|
|
|
def edit(self):
|
|
self.setText(0, shared.config.get(self.address, 'label'))
|
|
super(QtGui.QAbstractItemView, self).edit()
|
|
|
|
# label (or address) alphabetically, disabled at the end
|
|
def __lt__(self, other):
|
|
if (isinstance(other, Ui_AddressWidget)):
|
|
reverse = False
|
|
if self.treeWidget().header().sortIndicatorOrder() == QtCore.Qt.DescendingOrder:
|
|
reverse = True
|
|
if shared.config.getboolean(self.address, 'enabled') == shared.config.getboolean(other.address, 'enabled'):
|
|
if shared.config.get(self.address, 'label'):
|
|
x = shared.config.get(self.address, 'label').decode('utf-8').lower()
|
|
else:
|
|
x = self.address.decode('utf-8').lower()
|
|
if shared.config.get(other.address, 'label'):
|
|
y = shared.config.get(other.address, 'label').decode('utf-8').lower()
|
|
else:
|
|
y = other.address.decode('utf-8').lower()
|
|
return x < y
|
|
# else:
|
|
return (not reverse if shared.config.getboolean(self.address, 'enabled') else reverse)
|
|
|
|
return super(QtGui.QTreeWidgetItem, self).__lt__(other)
|
|
|
|
|
|
class Ui_SubscriptionWidget(Ui_AddressWidget):
|
|
def __init__(self, parent, pos = 0, address = "", unreadCount = 0, label = "", enabled = ""):
|
|
super(QtGui.QTreeWidgetItem, self).__init__()
|
|
self.unreadCount = unreadCount
|
|
parent.insertTopLevelItem(pos, self)
|
|
# only set default when creating
|
|
#super(QtGui.QTreeWidgetItem, self).setExpanded(shared.config.getboolean(self.address, 'enabled'))
|
|
self.setEnabled(enabled)
|
|
self.setLabel(label)
|
|
self.setAddress(address)
|
|
|
|
def setLabel(self, label):
|
|
self.label = label
|
|
|
|
def setAddress(self, address):
|
|
self.address = str(address)
|
|
self.setType()
|
|
self.setExpanded(self.isEnabled)
|
|
self.updateText()
|
|
|
|
def setEnabled(self, enabled):
|
|
self.isEnabled = enabled
|
|
|
|
def setType(self):
|
|
self.type = "subscription"
|
|
|
|
def setUnreadCount(self, cnt):
|
|
self.unreadCount = int(cnt)
|
|
self.updateText()
|
|
|
|
def updateText(self):
|
|
text = unicode(self.label, 'utf-8)') + ' (' + self.address + ')'
|
|
|
|
font = QtGui.QFont()
|
|
if self.unreadCount > 0:
|
|
# only show message count if the child doesn't show
|
|
if not self.isExpanded():
|
|
text += " (" + str(self.unreadCount) + ")"
|
|
font.setBold(True)
|
|
else:
|
|
font.setBold(False)
|
|
self.setFont(0, font)
|
|
|
|
#set text color
|
|
if self.isEnabled:
|
|
brush = QtGui.QBrush(QtGui.QColor(137, 04, 177))
|
|
#self.setExpanded(True)
|
|
else:
|
|
brush = QtGui.QBrush(QtGui.QColor(128, 128, 128))
|
|
#self.setExpanded(False)
|
|
brush.setStyle(QtCore.Qt.NoBrush)
|
|
self.setForeground(0, brush)
|
|
|
|
self.setIcon(0, avatarize(self.address))
|
|
self.setText(0, text)
|
|
self.setToolTip(0, text)
|
|
# self.setData(0, QtCore.Qt.UserRole, [self.address, "inbox"])
|
|
|
|
def setExpanded(self, expand):
|
|
super(Ui_SubscriptionWidget, self).setExpanded(expand)
|
|
self.updateText()
|
|
|
|
def edit(self):
|
|
self.setText(0, self.label)
|
|
super(QtGui.QAbstractItemView, self).edit()
|
|
|
|
# label (or address) alphabetically, disabled at the end
|
|
def __lt__(self, other):
|
|
if (isinstance(other, Ui_SubscriptionWidget)):
|
|
reverse = False
|
|
if self.treeWidget().header().sortIndicatorOrder() == QtCore.Qt.DescendingOrder:
|
|
reverse = True
|
|
if self.isEnabled == other.isEnabled:
|
|
if self.label:
|
|
x = self.label.decode('utf-8').lower()
|
|
else:
|
|
x = self.address.decode('utf-8').lower()
|
|
if other.label:
|
|
y = other.label.decode('utf-8').lower()
|
|
else:
|
|
y = other.address.decode('utf-8').lower()
|
|
return x < y
|
|
# else:
|
|
return (not reverse if self.isEnabled else reverse)
|
|
|
|
return super(QtGui.QTreeWidgetItem, self).__lt__(other) |