class LogViewer(ScreenWithBackButton):
def __init__(self, controller, mainWindow):
self.controller = controller
ScreenWithBackButton.__init__(self, "System Log", mainWindow)
def makeContent(self):
layout = QVBoxLayout()
self.table = QTableWidget()
layout.addWidget(self.table)
return layout
@handlePyroErrors
def displayLog(self):
entries = self.controller.getLog()
self.table.clearContents()
self.table.setRowCount(len(entries))
self.table.setColumnCount(3)
self.table.setHorizontalHeaderLabels(["Time", "Severity", "Message"])
i = 0
for entry in entries:
self.table.setItem(i, 0, QTableWidgetItem(entry.asctime))
self.table.setItem(i, 1, QTableWidgetItem(entry.levelname))
self.table.setItem(i, 2, QTableWidgetItem(entry.message))
i = i + 1
self.table.resizeColumnsToContents()
self.table.horizontalHeader().setStretchLastSection(True)
self.table.scrollToBottom()
class SelectActivities(QDialog):
select = Signal()
def __init__(self, parent = None):
super(SelectActivities, self).__init__(parent)
layout = QVBoxLayout(self)
self.table = QTableWidget(parent)
self.table.setColumnCount(3)
self.table.setHorizontalHeaderLabels(["", "Date", "Size"])
self.table.setColumnWidth(0, 25)
self.table.setColumnWidth(1, 100)
self.table.setColumnWidth(3, 80)
layout.addWidget(self.table)
self.buttonbox = QDialogButtonBox(
QDialogButtonBox.Ok | QDialogButtonBox.Cancel)
self.buttonbox.accepted.connect(self.accept)
self.buttonbox.rejected.connect(self.reject)
self.select.connect(self._select)
layout.addWidget(self.buttonbox)
self.setMinimumSize(320, 200)
def selectActivities(self, antfiles):
logger.debug("SelectActivities.selectActivities")
self.antfiles = antfiles
self.lock = threading.Condition()
self.lock.acquire()
self.select.emit()
logger.debug("SelectActivities.selectActivities: signal emitted")
self.lock.wait()
self.lock.release()
logger.debug("SelectActivities.selectActivities: returning %s selected activities", len(self._selected))
return self._selected
def _select(self):
logger.debug("SelectActivities._select")
self.table.clear()
self.table.setRowCount(len(self.antfiles))
for row in range(len(self.antfiles)):
f = self.antfiles[row]
self.table.setCellWidget(row, 0, QCheckBox(self))
self.table.setItem(row, 1,
QTableWidgetItem(f.get_date().strftime("%d %b %Y %H:%M")))
self.table.setItem(row, 2,
QTableWidgetItem("{:d}".format(f.get_size())))
result = self.exec_()
self._selected = []
if result == QDialog.Accepted:
for i in range(len(self.antfiles)):
f = self.antfiles[i]
cb = self.table.cellWidget(i, 0)
if cb.isChecked():
self._selected.append(f)
self.lock.acquire()
logger.debug("SelectActivities._select: lock acquired")
self.lock.notify()
self.lock.release()
class ListView(View):
base = True
def show(self):
super(ListView, self).show()
win = self.binder._parent
minimum = self.get_minimal_size()
height = win.height() if win.height() > minimum[0] else minimum[0]
width = win.width() if win.width() > minimum[1] else minimum[1]
win.resize(width, height)
def create_design(self):
header = self.get_header_labels()
self.lay = QVBoxLayout(self)
self.list = QTableWidget(0, len(header), self)
self.list.setSelectionBehavior(QAbstractItemView.SelectRows)
self.list.setHorizontalHeaderLabels(header)
self.list.doubleClicked.connect(self.on_doubleClicked)
self.lay.addWidget(self.list)
def append_row(self, cols, element_id=None):
rowCount = self.list.rowCount()
self.list.insertRow(rowCount)
for index, value in enumerate(cols):
col = QTableWidgetItem(value)
col.element_id = element_id
col.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled)
self.list.setItem(rowCount, index, col)
def on_doubleClicked(self, index):
item = self.list.item(index.row(), index.column())
self.binder._parent.open('edit', item.element_id)
def clear_list(self):
while self.list.rowCount():
self.list.removeRow(0)
def generate_signals(self):
super(ListView, self).generate_signals()
self.add_signal(self.fill)
def fill(self, elements):
self.clear_list()
for element in elements:
cols = self.get_row_from_obj(element)
self.append_row(cols, self.get_element_id(element))
def get_element_id(self, element):
return element.id
class SubscriberDialog(QDialog):
if USE_MAEMO_5:
switchRequested = Signal()
def __init__(self, parent=None):
QDialog.__init__(self, parent)
self.ui = Ui_SubscriberDialog()
self.ui.setupUi(self)
self.subscriber = None
self.tableWidget = None
self.listWidget = None
if USE_MAEMO_5:
switchButton = self.ui.buttonBox.addButton(
self.tr('Switch'), QDialogButtonBox.ActionRole)
switchButton.clicked.connect(self.switchRequested)
self.tableWidget = self.ui.tableWidget
headerLabels = ('Key', 'Value', 'Type')
self.tableWidget.setColumnCount(3)
self.tableWidget.setHorizontalHeaderLabels(headerLabels)
horizontalHeader = self.tableWidget.horizontalHeader()
horizontalHeader.setStretchLastSection(True)
verticalHeader = self.tableWidget.verticalHeader()
verticalHeader.setVisible(False)
self.tableWidget.setColumnWidth(0, 200)
self.tableWidget.setColumnWidth(1, 400)
else:
desktopWidget = QDesktopWidget()
if desktopWidget.availableGeometry().width() < 400:
# Screen is too small to fit a table widget without scrolling, use a list widget instead.
self.listWidget = QListWidget()
self.listWidget.setAlternatingRowColors(True)
self.ui.verticalLayout.insertWidget(2, self.listWidget)
else:
self.tableWidget = QTableWidget()
headerLabels = ('Key', 'Value', 'Type')
self.tableWidget.setColumnCount(3)
self.tableWidget.setHorizontalHeaderLabels(headerLabels)
horizontalHeader = self.tableWidget.horizontalHeader()
horizontalHeader.setStretchLastSection(True)
self.tableWidget.verticalHeader()
self.setVisible(False)
self.ui.verticalLayout.insertWidget(2, self.tableWidget)
self.ui.connectButton.clicked.connect(self.changeSubscriberPath)
self.changeSubscriberPath()
# if the default path does not exist reset it to /
value = self.subscriber.value()
subPaths = self.subscriber.subPaths()
if not value and not subPaths:
self.ui.basePath.setText('/')
self.changeSubscriberPath()
def changeEvent(self, e):
QDialog.changeEvent(self, e)
if e.type() == QEvent.LanguageChange:
self.ui.retranslateUi(self)
def changeSubscriberPath(self):
if self.listWidget:
self.listWidget.clear()
elif self.tableWidget:
self.tableWidget.clearContents()
if not self.subscriber:
self.subscriber = QValueSpaceSubscriber(self.ui.basePath.text(),
self)
else:
self.subscriber.setPath(self.ui.basePath.text())
self.subscriber.contentsChanged.connect(self.subscriberChanged)
self.subscriber.connectNotify("contentsChanged()")
self.subscriberChanged()
def subscriberChanged(self):
subPaths = self.subscriber.subPaths()
if self.listWidget:
self.listWidget.clear()
elif self.tableWidget:
self.tableWidget.clearContents()
self.tableWidget.setRowCount(len(subPaths))
for i in xrange(len(subPaths)):
v = self.subscriber.value(subPaths[i])
if self.listWidget:
item = QListWidgetItem('%s (%s)\n%s' %
(subPaths[i], str(type(v)), str(v)))
item.setFlags(item.flags() & ~Qt.ItemIsEditable)
self.listWidget.addItem(item)
elif self.tableWidget:
pathItem = QTableWidgetItem(subPaths[i])
pathItem.setFlags(pathItem.flags() & ~Qt.ItemIsEditable)
valueItem = QTableWidgetItem(str(v))
valueItem.setFlags(pathItem.flags() & ~Qt.ItemIsEditable)
typeItem = QTableWidgetItem(str(type(v)))
typeItem.setFlags(pathItem.flags() & ~Qt.ItemIsEditable)
self.tableWidget.setItem(i, 0, pathItem)
self.tableWidget.setItem(i, 1, valueItem)
self.tableWidget.setItem(i, 2, typeItem)
app = QApplication(sys.argv)
# Create and populate the tableWidget
tableWidget = QTableWidget(4, 4)
tableWidget.setItemDelegate(StarDelegate())
tableWidget.setEditTriggers(QAbstractItemView.DoubleClicked
| QAbstractItemView.SelectedClicked)
tableWidget.setSelectionBehavior(QAbstractItemView.SelectRows)
tableWidget.setHorizontalHeaderLabels(
["Title", "Genre", "Artist", "Rating"])
data = [["Mass in B-Minor", "Baroque", "J.S. Bach", 5],
["Three More Foxes", "Jazz", "Maynard Ferguson", 4],
["Sex Bomb", "Pop", "Tom Jones", 3],
["Barbie Girl", "Pop", "Aqua", 5]]
for r in range(len(data)):
tableWidget.setItem(r, 0, QTableWidgetItem(data[r][0]))
tableWidget.setItem(r, 1, QTableWidgetItem(data[r][1]))
tableWidget.setItem(r, 2, QTableWidgetItem(data[r][2]))
item = QTableWidgetItem()
item.setData(0, StarRating(data[r][3]).starCount)
tableWidget.setItem(r, 3, item)
tableWidget.resizeColumnsToContents()
tableWidget.resize(500, 300)
tableWidget.show()
sys.exit(app.exec_())
class MainWindow(QDialog):
def __init__(self, parent=None):
super(MainWindow, self).__init__(parent)
self.stopSign = 0
self.ws = 0
self.comment = ''
self.clipboard = QApplication.clipboard()
self.setWindowTitle(u'勤務表')
self.setWindowIcon(QIcon('./img/tray.png'))
self.systemTrayIcon = QSystemTrayIcon(self)
self.systemTrayIcon.setIcon(QIcon('./img/tray.png'))
self.systemTrayIcon.setVisible(True)
self.systemTrayIcon.show()
self.systemTrayIcon.activated.connect(self.on_systemTrayIcon_activated)
self.tableLabel = QLabel('select *.xls file')
self.pathText = QLineEdit()
self.triggerBtn = QPushButton(u'檢查 / 開始')
self.browseBtn = QPushButton(u' 瀏覽 ')
self.stopBtn = QPushButton(u'停止')
self.table = QTableWidget(26,0,self)
self.setUpTable()
self.hbox1 = QHBoxLayout()
self.hbox2 = QHBoxLayout()
self.hbox3 = QHBoxLayout()
self.hbox4 = QHBoxLayout()
self.hbox1.addWidget(self.pathText)
self.hbox1.addWidget(self.browseBtn)
self.hbox1.addWidget(self.triggerBtn)
self.browseBtn.clicked.connect(self.OpenFile)
self.triggerBtn.clicked.connect(self.setupTypeThread)
self.stopBtn.clicked.connect(self.threadStop)
self.status = QTreeWidget(self)
self.status.setHeaderHidden(True)
self.hbox2.addWidget(self.status)
self.hbox3.addWidget(self.table)
self.hbox4.addWidget(self.stopBtn)
self.setGeometry(200, 200, 700, 400)
self.status.setFixedHeight (80)
self.layout = QVBoxLayout()
self.layout.addWidget(self.tableLabel)
self.layout.addLayout(self.hbox1)
self.layout.addLayout(self.hbox2)
self.layout.addLayout(self.hbox3)
self.layout.addLayout(self.hbox4)
self.setLayout(self.layout)
self.stopBtn.setEnabled(False)
def setUpTable(self):
self.table.horizontalHeader().setVisible(False)
for i in xrange(0, 26, 1):
self.vhfont = QFont('Times', pointSize = 10, weight=QFont.Bold)
timePos = i+6
if i == 0:
item = QTableWidgetItem(u'[標題]')
elif i == 1:
item = QTableWidgetItem(u'[設定]')
elif 2 < timePos < 24:
item = QTableWidgetItem(('{0}~{1}').format(timePos, timePos+1))
else:
item = QTableWidgetItem(('{0}~{1}').format(timePos-24, timePos-23))
item.setFont(self.vhfont)
item.setTextAlignment(Qt.AlignCenter)
self.table.setVerticalHeaderItem(i, item)
def on_systemTrayIcon_activated(self, reason):
if reason == QSystemTrayIcon.DoubleClick:
if self.isHidden():
try:
self.typeThread
self.threadStop()
except:
pass
self.show()
else:
self.hide()
def setupTypeThread(self):
self.clipboard.setText('')
det = self.loadDetec()
self.ws, vaild = self.checkTableValidation(det, self.table)
ws = []
for col in self.ws:
ws.append(col[0])
self.addStatus( u' 狀態: 檢查勤務表狀態....', 0)
errStat = self.check(ws)
if vaild != True and self.table.columnCount()!=0:
self.addStatus( vaild, 1)
elif self.table.columnCount() ==0:
self.addStatus( u' 錯誤： 請載入勤務表', 1)
self.typeThread = startType(self.ws)
self.typeThread.threadDone.connect(self.showDoneMsg, Qt.QueuedConnection)
self.typeThread.toTray.connect(self.toTray, Qt.QueuedConnection)
self.typeThread.toCilpboard.connect(self.toCilpboard, Qt.QueuedConnection)
self.typeThread.enableButtons.connect(self.enableButtons, Qt.QueuedConnection)
self.typeThread.showErrMsg.connect(self.showErrMsg, Qt.QueuedConnection)
self.typeThread.addStatus.connect(self.addStatus, Qt.QueuedConnection)
if not self.typeThread.isRunning() and vaild == True and errStat == True:
self.addStatus( u' 狀態: 檢查通過，開始進行登打作業....', -1)
self.browseBtn.setEnabled(False)
self.triggerBtn.setEnabled(False)
self.stopBtn.setEnabled(True)
self.typeThread.start()
def toTray(self, state):
if state:
pass
self.hide()
self.systemTrayIcon.showMessage(u'輸入中',u'勤務表背景輸入中....\n\n雙擊圖示可暫停程序', msecs=1000000)
else:
self.show()
def toCilpboard(self, text):
self.clipboard.setText(text)
def threadStop(self):
self.typeThread.stopSign = 1
if self.typeThread.isRunning():
self.browseBtn.setEnabled(True)
self.triggerBtn.setEnabled(True)
self.stopBtn.setEnabled(False)
def addStatus(self, text, err):
subTreeItem = QTreeWidgetItem(self.status)
subTreeItem.setText(0, text)
self.activateWindow()
if err == 1:
font = QFont('Serif', 10, QFont.Bold)
subTreeItem.setFont(0, font)
subTreeItem.setForeground(0, QBrush(Qt.white))
subTreeItem.setBackground(0, QBrush(QColor(150,0,0)))
elif err == 0:
font = QFont('Serif', 10, QFont.Bold)
subTreeItem.setFont(0, font)
subTreeItem.setForeground(0, QBrush(Qt.black))
subTreeItem.setBackground(0, QBrush(Qt.white))
else:
font = QFont('Serif', 10, QFont.Bold)
subTreeItem.setFont(0, font)
subTreeItem.setForeground(0, QBrush(Qt.white))
subTreeItem.setBackground(0, QBrush(QColor(0,150,0)))
self.status.scrollToItem(subTreeItem, QAbstractItemView.PositionAtCenter)
def showErrMsg(self, title, msg):
self.addStatus( u'錯誤: ' + msg, 1)
QMessageBox.warning(self, title, msg, QMessageBox.Ok)
def showDoneMsg(self):
self.addStatus( u'完成: 完成!', 1)
QMessageBox.warning(self, u'完成!', u'完成!', QMessageBox.Ok)
def enableButtons(self):
self.clipboard.setText(self.comment)
self.browseBtn.setEnabled(True)
self.triggerBtn.setEnabled(True)
self.stopBtn.setEnabled(False)
self.show()
def OpenFile(self):
fileName = QFileDialog.getOpenFileName(self, "Open File.", "/home")
self.comment = ''
ext = fileName[0].split('.')[-1]
if ext == 'xls' or ext == 'xlsx':
self.pathText.setText(fileName[0])
sheet = open_workbook(fileName[0]).sheets()[0]
self.setFixedHeight(550)
self.addStatus( u' 狀態: 載入勤務表: [' + fileName[0] + ']', -1)
for col in xrange(self.table.columnCount()-1,-1,-1):
self.table.removeColumn(col)
ws, header, headerNum = self.loadWS(sheet)
self.appendTable(header, ws)
self.check(ws)
self.comment += self.yieldcomment(sheet)
self.ws = ws
else:
self.showErrMsg(u'錯誤',u'選取檔案不是EXCEL檔')
self.ws = 0
def checkTableValidation(self, detCol, table):
# .[ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
ws = [[], [], [], [], [], [], [], [], [], []]
errStat = True
if len(detCol) == 0 or (0 in detCol):
errStat = u' 狀態: 勤務表標頭錯誤,點選 [ --請選擇-- ] 選取有效標頭'
for i in xrange(len(detCol)-1):
if sorted(detCol)[i] == sorted(detCol)[i+1]:
errStat = u' 狀態: 勤務表標頭重複'
print detCol
for c in xrange(table.columnCount()):
col = []
colNum = detCol[c]
for r in xrange(2, 26, 1):
col.append(table.item(r,c).text())
ws[colNum-1].append(col)
for i in xrange(len(ws)):
if len(ws[i]) == 0:
ws[i].append(['','','','','','','','','','','','','','','','','','','','','','','',''])
return (ws), errStat
def loadWS(self, sheet):
header, headerNum, ws= [],[],[]
for c in xrange(3, 26 ,1):
title = (sheet.cell_value(3, c))
if len(title) != 0 and len(header) <6:
header.append(title)
col = []
for m in xrange(7, 31, 1):
try:
col.append(str(sheet.cell_value(m, c)).strip('()').replace('.0', '').replace('.', ','))
except:
col.append(u'error')
ws.append(col)
return ws, header, headerNum
def appendTable(self, header, ws):
try:
font = QFont('TypeWriter', pointSize = 10, weight=QFont.Bold)
for text in header:
self.table.insertColumn(0)
for c in xrange(len(header)):
det = self.determine(header[c])
item = QTableWidgetItem(header[c])
if det == 0:
item.setBackground(QBrush(QColor('#FF8D00')))
else:
item.setBackground(QBrush(QColor('#005588')))
item.setFont(font)
item.setTextAlignment(Qt.AlignCenter)
self.table.setItem(0, c, item)
nComboBox = self.newCombo()
nComboBox.setCurrentIndex(det)
self.table.setCellWidget(1, c, (nComboBox))
for r in xrange(2,26,1):
item = QTableWidgetItem(ws[c][r-2])
item.setFont(font)
item.setTextAlignment(Qt.AlignCenter)
self.table.setItem(r, c, item)
self.addStatus( u' 狀態: 勤務表預覽成功', -1)
return 0
except:
self.addStatus( u' 狀態: 勤務表預覽失敗', 1)
return 'error'
def loadDetec(self):
det = []
for col in xrange(self.table.columnCount()):
det.append( self.table.cellWidget(1, col).currentIndex())
return det
def newCombo(self):
taskList = [u'--請選擇--', u'值班', u'救護勤務', u'備勤', u'待命服勤', u'水源查察', u'消防查察', u'宣導勤務', u'訓(演)練', u'專案勤務', u'南山救護站']
nComboBox = QComboBox()
nComboBox.addItems(taskList)
for i in xrange(len(taskList)):
if i == 0:
nComboBox.setItemData(i, QColor('#550000'), Qt.BackgroundColorRole)
nComboBox.setItemData(i, Qt.AlignCenter, Qt.TextAlignmentRole)
nComboBox.setStyleSheet("text-align: right; font: bold 13px;")
return nComboBox
def setTableErr(self, row, state):
if state == 'illegal' : color = '#CC0033'
elif state == 'repeated' : color = '#FF8D00'
elif state == 'legal' : color = '#201F1F'
for col in xrange(self.table.columnCount()):
self.table.item(row,col).setBackground(QBrush(QColor(color)))
def check(self, ws):
errStat = True
for m in xrange(2,26):
self.setTableErr(m, 'legal')
for i in xrange(24):
ary = []
for j in xrange(len(ws)):
for each in ws[j][i].replace(' ', '').split(','):
try:
if each == "A": ary.append(-1)
elif each == '' : pass
else: each == ary.append(int(each))
except:
errStat = False
timePos = i+8
rptErrMSG = u" 錯誤: 於 {0} ~ {1} 時 數字: {2} --不合法, 請修正"
self.setTableErr(i+2, 'illegal')
print timePos
if timePos < 23:
print 'a'
self.addStatus(rptErrMSG.format(timePos, timePos+1, each), 1)
else:
self.addStatus(rptErrMSG.format(timePos-24, timePos-23, each), 1)
ary = sorted(ary)
for idx in xrange(len(ary)-1):
if ary[idx] == ary[idx+1]:
errStat = False
timePos = i+8
rptErrMSG = u" 錯誤: 於 {0} ~ {1} 時 數字: {2} --番號重複, 請修正"
self.setTableErr(i+2, 'repeated')
if timePos < 23:
self.addStatus(rptErrMSG.format(timePos, timePos+1, str(ary[idx]).replace('-1', 'A')), 1)
else:
self.addStatus(rptErrMSG.format(timePos-24, timePos-23, str(ary[idx]).replace('-1', 'A')), 1)
return errStat
def determine(self, title):
cmpfactor = 0
smp = [u'值班', u'救護分隊務', u'備', u'待命服', u'水源', u'消防', u'宣導', u'訓演練', u'專案其它', u'南山站']
for index, each in zip(xrange(len(smp)),smp):
for text in each:
for elem in title:
cmpfactor += ( elem == text )
if cmpfactor > 0:
return index+1
return 0
def yieldcomment(self, sheet):
comment0, comment1 = '', ''
# for i,j in [[24,27], [27,27], [29,27], [29,35]]:
# try:
# comment0 += (smart_str(sheet.cell_value(i, j)) + '\n')
# except:
# pass
for i,j in [[31,3], [32,3], [33,3], [34,3]]:
try:
comment1 += (sheet.cell_value(i, j) + '\n')
except:
pass
return comment1
class AdminWidget(QWidget):
"""Represents admin panel"""
def __init__(self, window):
super(AdminWidget, self).__init__(window)
self.parentWidget().setWindowTitle("Administartor panel")
self.__setupMenu()
self.layout = QGridLayout(self)
self.user_table = QTableWidget(self)
self.user_table.setColumnCount(3)
self.user_table.setHorizontalHeaderLabels(["Username", "Blocked", "Password restriction"])
self.user_table.setSelectionBehavior(QAbstractItemView.SelectRows)
self.user_table.setSelectionMode(QAbstractItemView.SingleSelection)
self.user_table.setColumnWidth(0, 210)
self.user_table.setColumnWidth(2, 170)
self.user_table.verticalHeader().hide()
changePasswordBtn = QPushButton("Change Password")
changePasswordBtn.clicked.connect(window.requestPasswordChange)
createUserBtn = QPushButton("Add User")
createUserBtn.clicked.connect(self._request_new_account)
logOutBtn = QPushButton("Logout")
logOutBtn.clicked.connect(lambda: (window.app.logOut(), window.hide(), window.requestCredentials()))
self.layout.addWidget(self.user_table, 0, 0, 1, 0)
self.layout.addWidget(changePasswordBtn, 1, 0)
self.layout.addWidget(createUserBtn, 1, 1)
self.layout.addWidget(logOutBtn, 1, 2)
self.setLayout(self.layout)
self.__loadUsers()
def __loadUsers(self):
"""Loads user's data from DB"""
users = self.parentWidget().app.getUsers()
self.user_table.clearContents()
self.user_table.setRowCount(len(users))
for i in range(len(users)):
username_item = QTableWidgetItem(users[i].username)
username_item.setFlags(username_item.flags() ^ Qt.ItemIsEditable)
blocked_checkbox = QCheckBox()
if users[i].blocked:
blocked_checkbox.setChecked(True)
def create_blocked_toggle(checkbox, user):
def blocked_toggle():
user.blocked = (1 if checkbox.isChecked() else 0)
self.parentWidget().app.updateUser(user)
self.__loadUsers()
return blocked_toggle
blocked_checkbox.toggled.connect(create_blocked_toggle(blocked_checkbox, users[i]))
password_restrict_checkbox = QCheckBox()
if users[i].restrictions:
password_restrict_checkbox.setChecked(True)
def create_password_restrict_toggle(checkbox, user):
def password_restrict_toggle():
user.restrictions = (1 if checkbox.isChecked() else 0)
self.parentWidget().app.updateUser(user)
self.__loadUsers()
return password_restrict_toggle
password_restrict_checkbox.toggled.connect(
create_password_restrict_toggle(password_restrict_checkbox, users[i]))
self.user_table.setItem(i, 0, username_item)
self.user_table.setCellWidget(i, 1, blocked_checkbox)
self.user_table.setCellWidget(i, 2, password_restrict_checkbox)
def __setupMenu(self):
add_account_action = QAction("Add account", self)
add_account_action.triggered.connect(self._request_new_account)
self.parentWidget().account_menu.addAction(add_account_action)
def _request_new_account(self):
dialog = None
def ok_handler(username):
if not 3 <= len(username) <= 20:
QMessageBox.critical(dialog, "Error", "Username's length must be between 3 and 20", modal=True)
return
user = User(username=username)
status = self.parentWidget().app.addUser(user)
{
Core.ERROR_USER_EXISTS: lambda:
QMessageBox.critical(dialog, "Error", "Username already exists", modal=True),
Core.SUCCESS: lambda: (
self.__loadUsers(),
dialog.close()
)
}[status]()
def cancel_handler():
dialog.close()
dialog = AdminWidget.AddAccountDialog(self, ok_handler, cancel_handler)
dialog.show()
class AddAccountDialog(QDialog):
def __init__(self, parent, ok_handler, cancel_handler):
super(AdminWidget.AddAccountDialog, self).__init__(parent)
self.setWindowTitle("Add account")
self.setFixedSize(300, 100)
self.setModal(True)
self.layout = QGridLayout(self)
self.username_label = QLabel(self)
self.username_label.setText("Username")
self.edit_username = QLineEdit(self)
self.buttons = QDialogButtonBox(self)
self.buttons.addButton(QDialogButtonBox.Ok)
self.buttons.addButton(QDialogButtonBox.Cancel)
self.buttons.button(QDialogButtonBox.Ok).setText("Add")
self.buttons.button(QDialogButtonBox.Cancel).setText("Cancel")
self.buttons.button(QDialogButtonBox.Cancel).clicked.connect(cancel_handler)
self.buttons.button(QDialogButtonBox.Ok).clicked.connect(lambda: ok_handler(self.edit_username.text()))
self.layout.addWidget(self.username_label, 0, 0)
self.layout.addWidget(self.edit_username, 0, 1)
self.layout.addWidget(self.buttons, 1, 0, 1, 2, Qt.AlignCenter)
self.setLayout(self.layout)
class TableWidgetController(AbstractViewController, Ui_TableWidget):
"""
The controller part for a data table widget. The UI part is declared
in tablewidget_v1.ui and then converted to ui_tablewidget.py.
This widget can be fed tabular data and has several options for showing it.
@since: 2010-11-10
"""
__author__ = "Moritz Wade"
__contact__ = "*****@*****.**"
__copyright__ = "Zuse Institute Berlin 2010"
def __init__(self, parent=None, host=None, title="Table", mode=None):
"""
Constructor, setting up lots of instance variables.
"""
super(TableWidgetController, self).__init__(parent)
self.setupUi(self)
self.setWindowTitle(title)
self._initialize()
self._mode = MODE_DEFAULT
self.maxValue = -1
self.host = host
self.data = None
self.dataTableHeaders = []
self.dataTableColumnData = []
self.dataTableRowCount = -1
self.dataTableRowHeaders = None
self.dataTableWidget = None
self.isColored = False
self.checkBoxShowUnits.setChecked(DEFAULT_SHOW_UNITS)
self.showUnits = DEFAULT_SHOW_UNITS
self.orientation = ORIENTATION_HORIZONTAL
self.sortColumn = -1 # default: do not sort at load
self.colorThreshold = None
self.colorThresholdBase = self.doubleSpinBox_Coloring_Threshold.value()
self.colorThresholdExponent = self.spinBox_Coloring_Exponent.value()
if mode:
self.setMode(mode)
self._updateThreshold()
def setMode(self, mode):
self._mode = mode
if mode == MODE_SUBCONDITIONS:
rtolScientificString = "{:e}".format(float(self.host.getRTol()))
exponentStr = rtolScientificString.split("e")[1]
self.spinBox_Coloring_Exponent.setValue(float(exponentStr))
self.doubleSpinBox_Coloring_Threshold.setValue(1.0)
self.label_Coloring_Threshold.setText("Anticipated Relative Measurement Error")
self.groupBox_Coloring.setChecked(True)
self.isColored = True
elif mode == MODE_JACOBIAN:
self.groupBox_Coloring.setChecked(True) # just activate coloring with default threshold
self.isColored = True
def _updateView(self, data=None):
"""
Overriding the "abstract" base class method. This does the main "drawing" of data, i.e.
generates the table and fills it with data.
"""
if data:
self.data = data
if self.data:
self._updateDataTable(self.data)
def _clearView(self):
if self.dataTableWidget:
self.dataTableWidget.clear()
def _setRowHeaders(self, dataDescriptors):
"""
Sets the given datadescriptors (from an EntityData object) as row headers
of the table, thereby checking for floats and rounding them, as not to show too
many digits after the point.
"""
if not dataDescriptors:
logging.debug("TableWidgetController._setRowHeaders(): Empty dataDescriptor list.")
return
self.dataTableRowHeaders = []
for descriptor in dataDescriptors:
try:
descriptor = float(descriptor)
descriptor = round(descriptor, 4)
except ValueError:
pass
self.dataTableRowHeaders.append(str(descriptor)) # the QTableWidget needs a list of Strings
def _computeColor(self, value):
if type(value) == str:
if value == "N/A":
return COLOR_LOW
try:
value = float(value)
except:
return None
if self._mode == MODE_DEFAULT or self._mode == MODE_JACOBIAN:
if value <= self.colorThreshold:
color = COLOR_LOW
else:
color = COLOR_HIGH
elif self._mode == MODE_SUBCONDITIONS:
if value >= self.colorThreshold:
color = COLOR_LOW
else:
percentage = (self.maxValue - value + 1) / float(
self.maxValue
) # +1 because it's the "lowest" subcondition, and represents 100%
highRed, highGreen, highBlue, highAlpha = (
COLOR_HIGH.red(),
COLOR_HIGH.green(),
COLOR_HIGH.blue(),
COLOR_HIGH.alpha(),
)
mediumRed, mediumGreen, mediumBlue, mediumAlpha = (
COLOR_MEDIUM.red(),
COLOR_MEDIUM.green(),
COLOR_MEDIUM.blue(),
COLOR_MEDIUM.alpha(),
)
diffRed, diffGreen, diffBlue, diffAlpha = (
highRed - mediumRed,
highGreen - mediumGreen,
highBlue - mediumBlue,
highAlpha - mediumAlpha,
)
valueRed = diffRed * percentage + mediumRed
valueGreen = diffGreen * percentage + mediumGreen
valueBlue = diffBlue * percentage + mediumBlue
valueAlpha = diffAlpha * percentage + mediumAlpha
color = QColor(valueRed, valueGreen, valueBlue, valueAlpha)
else:
color = QColor(0, 0, 0, 255) # transparent
return color
def _updateDataTable(self, data):
"""
Updates the data table with data from the last integration.
"""
# prepare data
self.dataTableHeaders = []
self.dataTableColumnData = []
self.dataTableRowCount = -1
self.dataTableRowHeaders = None
self.maxValue = -1
# BEGIN FOR: iterate over all data block
for (entity, entityDataList) in data.items():
for entityData in entityDataList:
dataDescriptors = entityData.dataDescriptors
if not self.dataTableRowHeaders:
self._setRowHeaders(dataDescriptors)
elif len(self.dataTableRowHeaders) != len(dataDescriptors):
logging.error(
"Different number of time points for two Species. Last Species (%s) will be skipped." % entity
)
continue
# set header for first column (dataDescriptor/Timepoint col)
# in first iteration
if len(self.dataTableHeaders) == 0:
dataDescriptorName = entityData.dataDescriptorName
dataDescriptorUnit = entityData.dataDescriptorUnit
if not dataDescriptorUnit and "timepoint" in str(dataDescriptorName).lower():
dataDescriptorUnit = self.host.optionTimeUnit
firstColHeader = ""
if dataDescriptorName:
if self.showUnits:
firstColHeader = "%s [%s]" % (dataDescriptorName, dataDescriptorUnit)
elif dataDescriptorName:
firstColHeader = "%s" % dataDescriptorName
self.dataTableHeaders.append(firstColHeader)
if len(dataDescriptors) > self.dataTableRowCount:
self.dataTableRowCount = len(dataDescriptors)
# datapoints = entityData.datapoints
datapoints = self.__convertEntityData(entityData)
# shorten datapoints
# for i, datapoint in enumerate(entityData.datapoints):
# try:
# floatValue = float(datapoint) # will jump to except if no float
# valueString = "N/A" if math.isnan(floatValue) else ' {0:-.4f}'.format(floatValue)
# datapoints.append(valueString)
# preparing color computation
# logging.debug(entityData.dataDescriptorName)
# logging.debug(entityData.dataDescriptors[i] == settingsandvalues.SUBCONDITION_HEADER_ABSOLUTE)
# if self._mode == MODE_SUBCONDITIONS\
# and entityData.dataDescriptors[i] == settingsandvalues.SUBCONDITION_HEADER_ABSOLUTE\
# and floatValue > self.maxValue\
# and floatValue < self.colorThreshold:
# self.maxValue = floatValue
# except:
# datapoints.append(str(datapoint))
# logging.debug("TableWidgetController - datapoints: %s" % datapoints) # too much overhead
# self.dataTableHeaders.append("Data species %s [%s]" % (str(speciesID), entityData.getUnit()))
if self.showUnits:
self.dataTableHeaders.append("%s [%s]" % (entity.getCombinedId(), entityData.getUnit()))
else:
self.dataTableHeaders.append("%s" % entity.getCombinedId())
self.dataTableColumnData.append(datapoints)
# END FOR:
# Put those labels into the actual data that would be the vertical/row labels.
# We can't use .setVerticalHeaderLabers() because those labels don't get sorted together with the data.
# Very weird but that seems to be the intended behaviour of Qt.
if self.orientation == ORIENTATION_VERTICAL:
self.dataTableHeaders = self.dataTableHeaders[1:] # remove unnecessary dataDescriptor name
for i in xrange(len(self.dataTableColumnData)):
entry = self.dataTableColumnData[i]
entry.insert(0, self.dataTableHeaders[i])
self.dataTableRowHeaders.insert(0, "")
else:
self.dataTableColumnData.insert(0, self.dataTableRowHeaders)
if not self.dataTableWidget: # create for the first time
tableLayout = QVBoxLayout(self.tableWrapper)
self.dataTableWidget = QTableWidget(self)
tableLayout.addWidget(self.dataTableWidget)
# prepare table
self.dataTableWidget.setSortingEnabled(
True
) # do here to avoid performance penalty (this actually does one sorting run)
if self.orientation == ORIENTATION_HORIZONTAL:
self.dataTableWidget.setColumnCount(len(self.dataTableHeaders))
self.dataTableWidget.setRowCount(self.dataTableRowCount)
self.dataTableWidget.setHorizontalHeaderLabels(self.dataTableHeaders)
elif self.orientation == ORIENTATION_VERTICAL:
self.dataTableWidget.setRowCount(len(self.dataTableHeaders))
self.dataTableWidget.setColumnCount(len(self.dataTableRowHeaders))
self.dataTableWidget.setHorizontalHeaderLabels(
self.dataTableRowHeaders
) # has to be called after setRowCount?
# put data into table
self.__addData()
# for col in xrange(len(self.dataTableColumnData)):
# for row in xrange(len(self.dataTableColumnData[col])):
# try:
# value = self.dataTableColumnData[col][row] # don't touch "values"; they could be pre-formatted strings
# newItem = SortedTableWidgetItem() # use custom item class
# newItem.setData(Qt.DisplayRole, value)
# newItem.setTextAlignment(Qt.AlignRight)
# newItem.setFont(QFont("Fixed"))
# if self.isColored:
# if not(self._mode == MODE_SUBCONDITIONS and row != 2): #color only row 2 of subcondition tables
# color = self._computeColor(value)
# if color:
# newItem.setBackground(QBrush(color))
# except Exception, e:
# logging.debug(
# "TableWidgetController._updateDataTable(): Could not put value into widget item: %s\nError: %s" % (
# value, e))
# if self.orientation == ORIENTATION_HORIZONTAL:
# self.dataTableWidget.setItem(row, col, newItem)
# elif self.orientation == ORIENTATION_VERTICAL:
# self.dataTableWidget.setItem(col, row, newItem)
if self.sortColumn != -1 and self.orientation == ORIENTATION_HORIZONTAL:
self.dataTableWidget.sortItems(self.sortColumn)
self.dataTableWidget.resizeColumnsToContents()
def __addData(self): # oddly in the vertical mode some data entries seem to be missing
for col in xrange(len(self.dataTableColumnData)):
dataList = self.dataTableColumnData[col]
for row in xrange(len(dataList)):
try:
value = dataList[row] # don't touch "values"; they could be pre-formatted strings
newItem = SortedTableWidgetItem() # use custom item class
newItem.setData(Qt.DisplayRole, value)
newItem.setTextAlignment(Qt.AlignRight)
newItem.setFont(QFont("Fixed"))
if self.isColored:
if not (
self._mode == MODE_SUBCONDITIONS and row != 2
): # color only row 2 of subcondition tables
color = self._computeColor(value)
if color:
newItem.setBackground(QBrush(color))
except Exception, e:
logging.debug(
"TableWidgetController._updateDataTable(): Could not put value into widget item: %s\nError: %s"
% (value, e)
)
if self.orientation == ORIENTATION_HORIZONTAL:
self.dataTableWidget.setItem(row, col, newItem)
elif self.orientation == ORIENTATION_VERTICAL:
self.dataTableWidget.setItem(col, row, newItem)
class FoamDictWidget(QWidget):
"QWidget to view and edit simple Foam Dictionary"
def __init__(self, variable_setting, parent=None):
super(FoamDictWidget, self).__init__(parent)
self.buttonLayout = QHBoxLayout()
self.pushButtonInsert = QPushButton("Insert")
#self.pushButtonLoad = QPushButton("Load default")
self.pushButtonRestore = QPushButton("Restore")
self.pushButtonClear = QPushButton("Clear")
self.buttonLayout.addWidget(self.pushButtonInsert)
#self.buttonLayout.addWidget(self.pushButtonLoad)
self.buttonLayout.addWidget(self.pushButtonRestore)
self.buttonLayout.addWidget(self.pushButtonClear)
self.buttonPreview = QPushButton('Preview FoamFile write-out')
self.textPreview = QTextEdit('')
self.textPreview.setVisible(False)
self.textPreview.setEnabled(False)
self.tableWidget = QTableWidget()
#header, should not sort, has vertical scrollbar
# set column count, fixed to 2, size of TableItem
self.tableWidget.setColumnCount(2)
#5self.tableWidget.setHorizontalHeaderItem(0, )
self.tableWidget.setHorizontalHeaderLabels(['key', 'value text'])
# set a default row count, insert as needed
self.tableWidget.setRowCount(0)
#PySide has different name other than @QtCore.pyqtSlot, but PySide.QtCore.SLOT
QtCore.QObject.connect(self.pushButtonInsert, QtCore.SIGNAL("clicked()"), self.insertRow)
QtCore.QObject.connect(self.pushButtonRestore, QtCore.SIGNAL("clicked()"), self.restoreDict)
QtCore.QObject.connect(self.pushButtonClear, QtCore.SIGNAL("clicked()"), self.clearDict)
#
QtCore.QObject.connect(self.tableWidget, QtCore.SIGNAL("doubleClicked()"), self.showPreview) # does not work for PySide
QtCore.QObject.connect(self.buttonPreview, QtCore.SIGNAL("clicked()"), self.showPreview)
self._previewing = False
self.settings = variable_setting
self.restoreDict()
self.myLayout = QVBoxLayout()
self.myLayout.addLayout(self.buttonLayout)
self.myLayout.addWidget(self.tableWidget)
self.myLayout.addWidget(self.buttonPreview)
self.myLayout.addWidget(self.textPreview)
self.setLayout(self.myLayout)
def dict(self):
_settings = OrderedDict()
for i in range(self.tableWidget.rowCount()):
k = self.tableWidget.item(i, 0).text()
v = self.tableWidget.item(i, 1).text() # data() will return QVariant type-> python type
# validated by non-empty string
if k and v:
_settings[k] = v
return _settings
def setDict(self, settings):
self.settings = settings
self.updateDictView(self.settings)
def restoreDict(self):
self.updateDictView(self.settings)
def updateDictView(self, varible_settings):
i = 0
self.clearDict() # will clear contents, but leave row text empty
N = self.tableWidget.rowCount()
for k,v in varible_settings.items():
# translate seq into unicode
if i>=N:
self.tableWidget.insertRow(i)
kitem = QTableWidgetItem(k) # also set flags and state, type
vitem = QTableWidgetItem(v) # automaticall convert str to unicode to feed QWidget?
self.tableWidget.setItem(i, 0, kitem)
self.tableWidget.setItem(i, 1, vitem) # currently only works for string value !
i += 1
#@pyqtSlot() # PySide use another name "QtCore.Slot()"
def insertRow(self):
nRows = self.tableWidget.rowCount()
self.tableWidget.insertRow(nRows) # inset one row at the end
kitem = QTableWidgetItem("") # also set flags and state, type
vitem = QTableWidgetItem("")
self.tableWidget.setItem(nRows, 0, kitem)
self.tableWidget.setItem(nRows, 1, vitem)
def clearDict(self):
self.tableWidget.clearContents() # keep the header, clear all items
def showPreview(self):
if self._previewing:
self._previewing = False
self.textPreview.setVisible(False)
self.buttonPreview.setText('click to preview write out')
else:
self._previewing = True
self.buttonPreview.setText('click on text to hide preview')
# enable scrollbar ?
self.textPreview.setText(self.printDict())
self.textPreview.setVisible(True)
def loadDefault(self):
pass
def printDict(self):
dictText = "{\n"
for k,v in self.dict().items():
dictText += " {} {};\n".format(str(k), str(v))
dictText += "}"
return dictText
class FoamDictWidget(QWidget):
"QWidget to view and edit simple Foam Dictionary"
def __init__(self, variable_setting, parent=None):
super(FoamDictWidget, self).__init__(parent)
self.buttonLayout = QHBoxLayout()
self.pushButtonInsert = QPushButton("Insert")
#self.pushButtonLoad = QPushButton("Load default")
self.pushButtonRestore = QPushButton("Restore")
self.pushButtonClear = QPushButton("Clear")
self.buttonLayout.addWidget(self.pushButtonInsert)
#self.buttonLayout.addWidget(self.pushButtonLoad)
self.buttonLayout.addWidget(self.pushButtonRestore)
self.buttonLayout.addWidget(self.pushButtonClear)
#PySide has different name other than @QtCore.pyqtSlot, but PySide.QtCore.SLOT
QtCore.QObject.connect(self.pushButtonInsert,
QtCore.SIGNAL("clicked()"), self.insertRow)
QtCore.QObject.connect(self.pushButtonRestore,
QtCore.SIGNAL("clicked()"), self.restoreDict)
QtCore.QObject.connect(self.pushButtonClear,
QtCore.SIGNAL("clicked()"), self.clearDict)
self.tableWidget = QTableWidget()
#header, should not sort, has vertical scrollbar
# set column count, fixed to 2, size of TableItem
self.tableWidget.setColumnCount(2)
#5self.tableWidget.setHorizontalHeaderItem(0, )
self.tableWidget.setHorizontalHeaderLabels(['key', 'value text'])
# set a default row count, insert as needed
self.tableWidget.setRowCount(0)
self.initialDict = variable_setting
self.restoreDict()
# debug print to console# does not work for PySide
QtCore.QObject.connect(self.tableWidget,
QtCore.SIGNAL("doubleClicked()"),
self.printDict)
self.myLayout = QVBoxLayout()
self.myLayout.addWidget(self.tableWidget)
self.myLayout.addLayout(self.buttonLayout)
self.setLayout(self.myLayout)
def dict(self):
_settings = OrderedDict()
for i in range(self.tableWidget.rowCount()):
k = self.tableWidget.item(i, 0).text()
v = self.tableWidget.item(
i, 1).text() # data() will return QVariant type-> python type
# validated by non-empty string
if k and v:
_settings[k] = v
return _settings
def updateDictView(self, varible_settings):
i = 0
self.clearDict() # will clear contents, but leave row text empty
N = self.tableWidget.rowCount()
for k, v in varible_settings.items():
# translate seq into unicode
if i >= N:
self.tableWidget.insertRow(i)
kitem = QTableWidgetItem(
unicode(k)) # also set flags and state, type
vitem = QTableWidgetItem(unicode(v))
#print(i, self.tableWidget.item(i, 0)) # debug: None
self.tableWidget.setItem(i, 0, kitem)
self.tableWidget.setItem(i, 1, vitem)
i += 1
def restoreDict(self):
self.updateDictView(self.initialDict)
#@pyqtSlot() # PySide use another name "QtCore.Slot()"
def insertRow(self):
nRows = self.tableWidget.rowCount()
self.tableWidget.insertRow(nRows) # inset one row at the end
kitem = QTableWidgetItem("") # also set flags and state, type
vitem = QTableWidgetItem("")
self.tableWidget.setItem(nRows, 0, kitem)
#print(nRows, self.tableWidget.item(nRows, 0))
self.tableWidget.setItem(nRows, 1, vitem)
def clearDict(self):
self.tableWidget.clearContents() # keep the header, clear all items
def printDict(self):
print(self.dict())
def loadDefault(self):
pass
class ui(QWidget):
def __init__(self):
QWidget.__init__(self)
self.setupUI()
self.id = 1
self.lines = []
self.editable = True
self.des_sort = True
self.faker = Factory.create()
self.btn_add.clicked.connect(self.add_line)
self.btn_del.clicked.connect(self.del_line)
self.btn_modify.clicked.connect(self.modify_line)
self.btn_select_line.clicked.connect(self.select_line)
self.btn_select_single.clicked.connect(self.deny_muti_line)
self.btn_sort.clicked.connect(self.sortItem)
self.btn_set_header.clicked.connect(self.setheader)
self.btn_set_middle.clicked.connect(self.middle)
self.table.cellChanged.connect(self.cellchange)
self.btn_noframe.clicked.connect(self.noframe)
# # Sess = sessionmaker(bind = engine)
def setupUI(self):
self.setWindowTitle(windowTital)
self.resize(640, 480)
self.table = QTableWidget(self)
self.btn_add = QPushButton(u'增加')
self.btn_del = QPushButton(u'删除')
self.btn_modify = QPushButton(u'可以编辑')
self.btn_select_line = QPushButton(u'选择整行')
self.btn_select_single = QPushButton(u'禁止选多行')
self.btn_sort = QPushButton(u'以分数排序')
self.btn_set_header = QPushButton(u'标头设置')
self.btn_set_middle = QPushButton(u'文字居中加颜色')
self.btn_noframe = QPushButton(u'取消边框颜色交替')
self.spacerItem = QSpacerItem(20, 20, QSizePolicy.Minimum,
QSizePolicy.Expanding)
self.vbox = QVBoxLayout()
self.vbox.addWidget(self.btn_add)
self.vbox.addWidget(self.btn_del)
self.vbox.addWidget(self.btn_modify)
self.vbox.addWidget(self.btn_select_line)
self.vbox.addWidget(self.btn_select_single)
self.vbox.addWidget(self.btn_sort)
self.vbox.addWidget(self.btn_set_header)
self.vbox.addWidget(self.btn_set_middle)
self.vbox.addWidget(self.btn_noframe)
self.vbox.addSpacerItem(
self.spacerItem) #可以用addItem也可以用addSpacerItem方法添加，没看出哪里不一样
self.txt = QLabel()
self.txt.setMinimumHeight(50)
self.vbox2 = QVBoxLayout()
self.vbox2.addWidget(self.table)
self.vbox2.addWidget(self.txt)
self.hbox = QHBoxLayout()
self.hbox.addLayout(self.vbox2)
self.hbox.addLayout(self.vbox)
self.setLayout(self.hbox)
self.table.setColumnCount(4) ##设置列数
self.headers = [u'id', u'选择', u'姓名', u'成绩', u'住址']
self.table.setHorizontalHeaderLabels(self.headers)
self.show()
def add_line(self):
self.table.cellChanged.disconnect()
row = self.table.rowCount()
self.table.setRowCount(row + 1)
id = str(self.id)
ck = QCheckBox()
h = QHBoxLayout()
h.setAlignment(Qt.AlignCenter)
h.addWidget(ck)
w = QWidget()
w.setLayout(h)
name = self.faker.name()
score = str(random.randint(50, 99))
add = self.faker.address()
self.table.setItem(row, 0, QTableWidgetItem(id))
self.table.setCellWidget(row, 1, w)
self.table.setItem(row, 2, QTableWidgetItem(name))
self.table.setItem(row, 3, QTableWidgetItem(score))
self.table.setItem(row, 4, QTableWidgetItem(add))
self.id += 1
self.lines.append([id, ck, name, score, add])
self.settext(u'自动生成随机一行数据！,checkbox设置为居中显示')
self.table.cellChanged.connect(self.cellchange)
def del_line(self):
removeline = []
for line in self.lines:
if line[1].isChecked():
row = self.table.rowCount()
for x in range(row, 0, -1):
if line[0] == self.table.item(x - 1, 0).text():
self.table.removeRow(x - 1)
removeline.append(line)
for line in removeline:
self.lines.remove(line)
self.settext(u'删除在左边checkbox中选中的行，使用了一个笨办法取得行号\n，不知道有没有其他可以直接取得行号的方法！')
def modify_line(self):
if self.editable == True:
self.table.setEditTriggers(QAbstractItemView.NoEditTriggers)
self.btn_modify.setText(u'禁止编辑')
self.editable = False
else:
self.table.setEditTriggers(QAbstractItemView.AllEditTriggers)
self.btn_modify.setText(u'可以编辑')
self.editable = True
self.settext(u'设置，是否可以编辑整个表格')
def select_line(self):
if self.table.selectionBehavior() == 0:
self.table.setSelectionBehavior(1)
self.btn_select_line.setStyleSheet('background-color:lightblue')
else:
self.table.setSelectionBehavior(0)
self.btn_select_line.setStyleSheet('')
self.settext(u'默认时，点击单元格，只可选择一个格，此处设置为可选择整行')
def deny_muti_line(self):
if self.table.selectionMode() in [2, 3]:
self.table.setSelectionMode(QAbstractItemView.SingleSelection)
self.btn_select_single.setStyleSheet('background-color:lightblue')
else:
self.table.setSelectionMode(QAbstractItemView.ExtendedSelection)
self.btn_select_single.setStyleSheet('')
self.settext(u'点击时会轮换以多行或单行选择，默认是可以同时选择多行')
def sortItem(self):
if self.des_sort == True:
self.table.sortItems(3, Qt.DescendingOrder)
self.des_sort = False
self.btn_sort.setStyleSheet('background-color:lightblue')
self.table.setSortingEnabled(True) # 设置表头可以自动排序
else:
self.table.sortItems(3, Qt.AscendingOrder)
self.des_sort = True
self.btn_sort.setStyleSheet('background-color:lightblue')
self.table.setSortingEnabled(False)
self.settext(u'点击时会轮换以升序降序排列，但排序时，会使自动列宽失效！')
def setheader(self):
font = QFont(u'微软雅黑', 12)
font.setBold(True)
self.table.horizontalHeader().setFont(font) # 设置表头字体
self.table.setColumnWidth(0, 50)
self.table.setColumnWidth(1, 50)
self.table.setColumnWidth(3, 100)
self.table.horizontalHeader().setSectionResizeMode(
2, QHeaderView.Stretch)
self.table.horizontalHeader().setStyleSheet(
'QHeaderView::section{background:gray}')
self.table.horizontalHeader().setFixedHeight(50)
self.table.setColumnHidden(0, True)
self.btn_set_header.setStyleSheet('background-color:lightblue')
self.settext(
u'设置标头字体及字号，隐藏ID列，设置标头除姓名外全部为固定宽度\n，设置姓名列自动扩展宽度，设置标头行高，设置标头背景色')
def middle(self):
self.btn_set_middle.setStyleSheet('background-color:lightblue')
self.table.setStyleSheet('color:green;')
row = self.table.rowCount()
for x in range(row):
for y in range(4):
if y != 1:
item = self.table.item(x, y)
item.setTextAlignment(Qt.AlignCenter)
else:
pass
self.btn_set_middle.setStyleSheet('background-color:lightblue')
self.settext(u'将文字居中显示,设置文字颜色')
def cellchange(self, row, col):
item = self.table.item(row, col)
txt = item.text()
self.settext(u'第%s行，第%s列 , 数据改变为:%s' % (row, col, txt))
def noframe(self):
self.table.setAlternatingRowColors(True)
self.table.setFrameStyle(QFrame.NoFrame)
self.table.setStyleSheet('color:green;'
'gridline-color:white;'
'border:0px solid gray')
self.settext(u'取消表的框线,\n 取消表格内框')
def settext(self, txt):
font = QFont(u'微软雅黑', 10)
self.txt.setFont(font)
self.txt.setText(txt)
app = QApplication(sys.argv)
# Create and populate the tableWidget
tableWidget = QTableWidget(4, 4)
tableWidget.setItemDelegate(StarDelegate())
tableWidget.setEditTriggers(QAbstractItemView.DoubleClicked |
QAbstractItemView.SelectedClicked)
tableWidget.setSelectionBehavior(QAbstractItemView.SelectRows)
tableWidget.setHorizontalHeaderLabels(["Title", "Genre", "Artist", "Rating"])
data = [ ["Mass in B-Minor", "Baroque", "J.S. Bach", 5],
["Three More Foxes", "Jazz", "Maynard Ferguson", 4],
["Sex Bomb", "Pop", "Tom Jones", 3],
["Barbie Girl", "Pop", "Aqua", 5] ]
for r in range(len(data)):
tableWidget.setItem(r, 0, QTableWidgetItem(data[r][0]))
tableWidget.setItem(r, 1, QTableWidgetItem(data[r][1]))
tableWidget.setItem(r, 2, QTableWidgetItem(data[r][2]))
item = QTableWidgetItem()
item.setData(0, StarRating(data[r][3]).starCount)
tableWidget.setItem(r, 3, item)
#tableWidget.resizeColumnsToContents()
tableWidget.resize(500, 300)
tableWidget.show()
sys.exit(app.exec_())
class PositionPanel(QtGui.QWidget):
positionTableWidget = None
summaryTable = None
movementFilterWidget = None
row = 0
summaryRow = 0
positionColumnList = "Asset Name;Position;Unit Cost;Market Price;Change%;Invested amount;Valuated amount;Tenor;Maturity Date;Gross PNL;Net PNL;Gross%PNL;Net%PNL;Realized Pnl;%Portfolio;WeightedPNL%".split(
";")
summaryColumnList = "Custody;Asset type;Invested Amount;Valuated Amount;Net PnL;Net%PNL;Realized Pnl;RPnL + NPnL;%Portfolio;WeightedPNL%".split(
";")
def __init__(self):
super(self.__class__, self).__init__()
self.layout = QtGui.QGridLayout(self)
self.movementFilterWidget = MovementFilterWidget()
self.layout.addWidget(self.movementFilterWidget, 1, 0)
def clearTables(self):
self.row = 0
self.summaryRow = 0
self.createTable()
self.createSummaryTable()
self.createGeneralInfoPanel()
def createSummaryTable(self):
self.summaryTableWidget = QTableWidget()
self.summaryTableWidget.setRowCount(7)
self.summaryTableWidget.setColumnCount(len(self.summaryColumnList))
self.summaryTableWidget.setEditTriggers(
QtGui.QAbstractItemView.NoEditTriggers)
self.summaryTableWidget.setHorizontalHeaderLabels(
self.summaryColumnList)
#self.summaryTableWidget.setSortingEnabled(True)
#self.summaryTableWidget.sortItems(0)
self.summaryTableWidget.resizeColumnsToContents()
self.summaryTableWidget.resizeRowsToContents()
self.summaryTableWidget.setFixedSize(800, 200)
self.layout.addWidget(self.summaryTableWidget, 1, 1)
def createGeneralInfoPanel(self):
self.generalInfoPanel = QWidget()
self.generalInfoLayout = QtGui.QGridLayout(self.generalInfoPanel)
self.lblUSDMXN = QLabel("USD/MXN")
self.lblUSDMXN.font()
self.lblUSDMXN.setFont(
QtGui.QFont("MS Shell Dlg", 12, QtGui.QFont.Normal))
self.generalInfoLayout.addWidget(self.lblUSDMXN, 1, 1)
self.lblUSDMXNValue = QLabel("0")
self.lblUSDMXNValue.setFont(
QtGui.QFont("MS Shell Dlg", 12, QtGui.QFont.Bold))
self.generalInfoLayout.addWidget(self.lblUSDMXNValue, 1, 2)
self.generalInfoPanel.setFixedSize(200, 50)
self.layout.addWidget(self.generalInfoPanel, 1, 2)
def createTable(self):
self.positionTableWidget = QTableWidget()
self.positionTableWidget.setRowCount(27)
self.positionTableWidget.setColumnCount(
len(self.positionColumnList) + 1)
self.positionTableWidget.setColumnHidden(
Constant.CONST_COLUMN_POSITION_HIDDEN_ID, True)
self.positionTableWidget.setEditTriggers(
QtGui.QAbstractItemView.NoEditTriggers)
self.positionTableWidget.setHorizontalHeaderLabels(
self.positionColumnList)
#self.positionTableWidget.setSortingEnabled(True)
#self.positionTableWidget.sortItems(0)
self.positionTableWidget.doubleClicked.connect(self.openMovementView)
self.positionTableWidget.resizeColumnsToContents()
self.positionTableWidget.resizeRowsToContents()
self.layout.addWidget(self.positionTableWidget, 2, 0, 3, 3)
def renderGeneralInfoPanel(self, usdMXNvalue):
self.lblUSDMXNValue.setText(str(round(usdMXNvalue, 4)))
#self.generalInfoLayout.addWidget(self.lblUSDMXNValue, 1, 2)
def renderSummary(self, summaryDict):
for (key, summaryItem) in sorted(summaryDict.iteritems()):
if summaryItem.custodyName is None:
isBold = True
else:
isBold = False
#custodyName
custodyNameItem = QTableWidgetItemString(summaryItem.custodyName,
isBold)
self.summaryTableWidget.setItem(
self.summaryRow,
Constant.CONST_COLUMN_SUMMARY_CUST_CUSTODY_NAME,
custodyNameItem)
#assetTypeName
assetTypeNameItem = QTableWidgetItemString(summaryItem.assetType,
isBold)
self.summaryTableWidget.setItem(
self.summaryRow,
Constant.CONST_COLUMN_SUMMARY_CUST_ASSET_TYPE_NAME,
assetTypeNameItem)
#InvestedAmount
investedAmountItem = QTableWidgetItemDecimal(
summaryItem.investedAmount, isBold)
self.summaryTableWidget.setItem(
self.summaryRow,
Constant.CONST_COLUMN_SUMMARY_CUST_INVESTED_AMOUNT,
investedAmountItem)
#valuatedAmount
valuatedAmountItem = QTableWidgetItemDecimal(
summaryItem.valuatedAmount, isBold)
self.summaryTableWidget.setItem(
self.summaryRow,
Constant.CONST_COLUMN_SUMMARY_CUST_VALUATED_AMOUNT,
valuatedAmountItem)
#subTotalNetPnL
netPnLItem = QTableWidgetItemDecimal(summaryItem.netPnL, isBold)
self.summaryTableWidget.setItem(
self.summaryRow, Constant.CONST_COLUMN_SUMMARY_CUST_NET_PNL,
netPnLItem)
#netPNLPercentage
netPNLPercentageItem = QTableWidgetItemDecimal(
summaryItem.getNetPnLPercentage(), isBold)
self.summaryTableWidget.setItem(
self.summaryRow,
Constant.CONST_COLUMN_SUMMARY_CUST_NET_PNL_PERCENTAGE,
netPNLPercentageItem)
#realizedPnl
realizedPnlItem = QTableWidgetItemDecimal(summaryItem.realizedPnl,
isBold)
self.summaryTableWidget.setItem(
self.summaryRow,
Constant.CONST_COLUMN_SUMMARY_CUST_REALIZED_PNL,
realizedPnlItem)
#realizedPnlPlusNetPnL
realizedPnlPlusNetPnLItem = QTableWidgetItemDecimal(
summaryItem.realizedPnl + summaryItem.netPnL, isBold)
self.summaryTableWidget.setItem(
self.summaryRow,
Constant.CONST_COLUMN_SUMMARY_CUST_REALIZED_PNL_PLUS_NET_PNL,
realizedPnlPlusNetPnLItem)
#positionPercentage
positionPercentageItem = QTableWidgetItemDecimal(
summaryItem.positionPercentage, isBold)
self.summaryTableWidget.setItem(
self.summaryRow,
Constant.CONST_COLUMN_SUMMARY_CUST_POSITION_PERCENTAGE,
positionPercentageItem)
#weightedPnL
weightedPnLItem = QTableWidgetItemDecimal(summaryItem.weightedPnL,
isBold)
self.summaryTableWidget.setItem(
self.summaryRow,
Constant.CONST_COLUMN_SUMMARY_CUST_WEIGHTED_PNL,
weightedPnLItem)
self.summaryRow += 1
def renderSubtotal(self, positionDict, assetType, isSIC):
accValuatedAmount = Engine.getSubTotalValuatedAmount(
positionDict, assetType, isSIC)
accInvestedAmount = Engine.getSubTotalInvestedAmount(
positionDict, assetType, isSIC)
accRealizedPnl = Engine.getAccRealizedPnL(positionDict, assetType,
isSIC)
accPositionPercentage = Engine.getAccPositionPercentage(
positionDict, assetType, isSIC)
accGrossPnlPercentage = Engine.getAccGrossPnlPercentage(
positionDict, assetType, isSIC)
accNetPnlPercentage = Engine.getAccNetPnlPercentage(
positionDict, assetType, isSIC)
accNetPNL = Engine.getAccNetPNL(positionDict, assetType, isSIC)
accWeightedPNL = Engine.getAccWeightedPNL(positionDict, assetType,
isSIC)
#Invested amount
investedAmountItem = QTableWidgetItemDecimal(accInvestedAmount, True)
self.positionTableWidget.setItem(
self.row, Constant.CONST_COLUMN_POSITION_INVESTED_AMOUNT,
investedAmountItem)
#sub total valuated amount
subTotalValuatedAmountItem = QTableWidgetItemDecimal(
accValuatedAmount, True)
self.positionTableWidget.setItem(
self.row, Constant.CONST_COLUMN_POSITION_VALUATED_AMOUNT,
subTotalValuatedAmountItem)
#sub total Gross PNL
subTotalGrossPNLItem = QTableWidgetItemDecimalColor(
Engine.getSubtotalGrossPNL(positionDict, assetType, isSIC), True)
self.positionTableWidget.setItem(
self.row, Constant.CONST_COLUMN_POSITION_GROSS_PNL,
subTotalGrossPNLItem)
#sub total Net PNL
subTotalNetPNLItem = QTableWidgetItemDecimalColor(accNetPNL, True)
self.positionTableWidget.setItem(
self.row, Constant.CONST_COLUMN_POSITION_NET_PNL,
subTotalNetPNLItem)
#subTotalGrossPnLPercentage
subTotalGrossPnLPercentage = QTableWidgetItemDecimalColor(
accGrossPnlPercentage, True)
self.positionTableWidget.setItem(
self.row, Constant.CONST_COLUMN_POSITION_GROSS_PNL_PERCENTAGE,
subTotalGrossPnLPercentage)
#pnLNetPercentage
subTotalNetPnLPercentage = QTableWidgetItemDecimalColor(
accNetPnlPercentage, True)
self.positionTableWidget.setItem(
self.row, Constant.CONST_COLUMN_POSITION_GROSS_NET_PERCENTAGE,
subTotalNetPnLPercentage)
#realizedPnL
realizedPnLItem = QTableWidgetItemDecimalColor(accRealizedPnl, True)
self.positionTableWidget.setItem(
self.row, Constant.CONST_COLUMN_POSITION_REALIZED_PNL,
realizedPnLItem)
#positionPercentage
positionPercentageItem = QTableWidgetItemDecimal(
accPositionPercentage, True)
self.positionTableWidget.setItem(
self.row, Constant.CONST_COLUMN_POSITION_POSITION_PERCENTAGE,
positionPercentageItem)
#weightedPercentageItem
weightedPercentageItem = QTableWidgetItemDecimal(accWeightedPNL, True)
self.positionTableWidget.setItem(
self.row, Constant.CONST_COLUMN_POSITION_WEIGHTED_PNL,
weightedPercentageItem)
#HiddenID
hiddenIDItem = QTableWidgetItemDecimal(self.row, False)
self.positionTableWidget.setItem(
self.row, Constant.CONST_COLUMN_POSITION_HIDDEN_ID, hiddenIDItem)
def renderPositions(self, positionDict, assetType, isSIC):
positionList = Engine.getPositionByAssetType(positionDict, assetType,
isSIC)
for position in positionList:
print('rendering ' + position.getAssetName())
if (position.getTotalQuantity() != 0):
position.row = self.row
#assetName
assetNameItem = QTableWidgetItemString(position.getAssetName(),
False)
self.positionTableWidget.setItem(
self.row, Constant.CONST_COLUMN_POSITION_ASSET_NAME,
assetNameItem)
#totalQuantity
totalQuantityItem = QTableWidgetItemInt(
position.getTotalQuantity(), False)
self.positionTableWidget.setItem(
self.row, Constant.CONST_COLUMN_POSITION_QUANTITY,
totalQuantityItem)
#UnitCostOrRate
unitCostItem = QTableWidgetItemDecimal(
position.getUnitCostOrRate(), False)
self.positionTableWidget.setItem(
self.row, Constant.CONST_COLUMN_POSITION_PPP, unitCostItem)
#Market price
marketPriceItem = QTableWidgetItemDuoDecimal(
position.getMarketPrice(), position.getMarketPriceOrig())
self.positionTableWidget.setItem(
self.row, Constant.CONST_COLUMN_POSITION_MARKET_PRICE,
marketPriceItem)
#changePercentage
changePercentageItem = QTableWidgetItemStringPlusMinus(
position.changePercentage, False)
self.positionTableWidget.setItem(
self.row, Constant.CONST_COLUMN_POSITION_CHANGE_PERCENTAGE,
changePercentageItem)
#Invested amount
investedAmountItem = QTableWidgetItemDecimal(
position.getInvestedAmount(), False)
self.positionTableWidget.setItem(
self.row, Constant.CONST_COLUMN_POSITION_INVESTED_AMOUNT,
investedAmountItem)
#Valuated amount
valuatedAmountItem = QTableWidgetItemDuoDecimal(
position.getValuatedAmount(),
position.getValuatedAmountOrig())
self.positionTableWidget.setItem(
self.row, Constant.CONST_COLUMN_POSITION_VALUATED_AMOUNT,
valuatedAmountItem)
#Tenor
tenorItem = QTableWidgetItemDuoInt(position.tenor,
position.getElapsedDays())
self.positionTableWidget.setItem(
self.row, Constant.CONST_COLUMN_POSITION_TENOR, tenorItem)
#Maturity Date
maturityDateItem = QTableWidgetItemString(
position.getMaturityDate(), False)
self.positionTableWidget.setItem(
self.row, Constant.CONST_COLUMN_POSITION_MATURITY_DATE,
maturityDateItem)
#GrossPnL
grossPnlItem = QTableWidgetItemDecimalColor(
position.getGrossPnL(), False)
self.positionTableWidget.setItem(
self.row, Constant.CONST_COLUMN_POSITION_GROSS_PNL,
grossPnlItem)
#netPnL
netPnlItem = QTableWidgetItemDecimalColor(
position.getNetPnL(), False)
self.positionTableWidget.setItem(
self.row, Constant.CONST_COLUMN_POSITION_NET_PNL,
netPnlItem)
#pnLGrossPercentage
pnLGrossPercentageItem = QTableWidgetItemDecimalColor(
position.getGrossPnLPercentage(), False)
self.positionTableWidget.setItem(
self.row,
Constant.CONST_COLUMN_POSITION_GROSS_PNL_PERCENTAGE,
pnLGrossPercentageItem)
#pnLNetPercentage
pnLNetPercentageItem = QTableWidgetItemDecimalColor(
position.getNetPnLPercentage(), False)
self.positionTableWidget.setItem(
self.row,
Constant.CONST_COLUMN_POSITION_GROSS_NET_PERCENTAGE,
pnLNetPercentageItem)
#realizedPnL
realizedPnLItem = QTableWidgetItemDecimalColor(
position.getConsolidatedRealizedPnl(), False)
self.positionTableWidget.setItem(
self.row, Constant.CONST_COLUMN_POSITION_REALIZED_PNL,
realizedPnLItem)
#positionPercentage
positionPercentageItem = QTableWidgetItemDecimal(
position.getPositionPercentage(), False)
self.positionTableWidget.setItem(
self.row,
Constant.CONST_COLUMN_POSITION_POSITION_PERCENTAGE,
positionPercentageItem)
#weightedPercentageItem
weightedPercentageItem = QTableWidgetItemDecimal(
position.getWeightedPnl(), False)
self.positionTableWidget.setItem(
self.row, Constant.CONST_COLUMN_POSITION_WEIGHTED_PNL,
weightedPercentageItem)
#HiddenID
hiddenIDItem = QTableWidgetItemDecimal(self.row, False)
self.positionTableWidget.setItem(
self.row, Constant.CONST_COLUMN_POSITION_HIDDEN_ID,
hiddenIDItem)
self.row += 1
self.renderSubtotal(positionDict, assetType, isSIC)
self.row += 1
def openMovementView(self):
assetName = self.positionTableWidget.item(
self.positionTableWidget.currentRow(),
Constant.CONST_COLUMN_POSITION_ASSET_NAME).text()
movementList = Engine.getMovementListByAsset(
assetName, (self.movementFilterWidget.dateFromDate.date()
).toString("yyyy-M-dd"),
(self.movementFilterWidget.dateToDate.date()
).toString("yyyy-M-dd"))
self.movementView = MovementView(movementList)
self.movementView.show()
class RecorderClipSelectionScreen(QWidget):
def __init__(self, hyperdeck, state, mainWindow):
super(RecorderClipSelectionScreen, self).__init__()
self.hyperdeck = hyperdeck
self.state = state
self.mainWindow = mainWindow
self.selected_clip = None
layout = QGridLayout()
lblTitle = TitleLabel("Select clip")
layout.addWidget(lblTitle, 0, 0, 1, 3)
self.clipTable = QTableWidget()
self.clipTable.setColumnCount(2)
self.clipTable.setHorizontalHeaderLabels(['ID', 'Clip name'])
self.clipTable.horizontalHeader().setStretchLastSection(True)
self.clipTable.setSelectionBehavior(
QAbstractItemView.SelectionBehavior.SelectRows)
self.clipTable.itemSelectionChanged.connect(self._onClipSelected)
layout.addWidget(self.clipTable, 1, 0, 1, 3)
b = ExpandingButton()
b.setText("Back")
b.setIcon(QIcon(":icons/go-previous"))
b.setToolButtonStyle(Qt.ToolButtonStyle.ToolButtonTextBesideIcon)
b.clicked.connect(mainWindow.stepBack)
layout.addWidget(b, 2, 0)
btnRefresh = ExpandingButton()
btnRefresh.setText('Refresh')
btnRefresh.setIcon(QIcon(':icons/refresh'))
btnRefresh.setToolButtonStyle(
Qt.ToolButtonStyle.ToolButtonTextBesideIcon)
def refresh():
self.populateClipsList({})
hyperdeck.broadcastClipsList()
btnRefresh.clicked.connect(refresh)
layout.addWidget(btnRefresh, 2, 1)
self.btnSelect = ExpandingButton()
self.btnSelect.setText("Cue clip")
self.btnSelect.clicked.connect(self._cueClip)
layout.addWidget(self.btnSelect, 2, 2)
layout.setRowStretch(0, 0)
layout.setRowStretch(1, 1)
layout.setRowStretch(2, 0)
self.setLayout(layout)
self.populateClipsList(state.clip_listing)
def populateClipsList(self, clipsList):
self.selected_clip = None
self.btnSelect.setEnabled(False)
self.clipTable.clearContents()
self.clipTable.setRowCount(len(clipsList))
for idx, (clip_id, val) in enumerate(clipsList.iteritems()):
self.clipTable.setItem(idx, 0, QTableWidgetItem(str(clip_id)))
self.clipTable.setItem(idx, 1, QTableWidgetItem(val['name']))
def _updateClipSelectionFromState(self, state):
if 'clip id' in state:
clip_id = state['clip id']
for row in range(self.clipTable.rowCount()):
this_id = self.clipTable.item(row, 0)
if this_id.text() == str(clip_id):
self.clipTable.selectRow(row)
break
def _onClipSelected(self):
selected_items = self.clipTable.selectedItems()
if selected_items:
self.selected_clip = int(selected_items[0].text())
self.btnSelect.setEnabled(True)
else:
self.btnSelect.setEnabled(False)
def _cueClip(self):
if self.selected_clip:
self.mainWindow.stepBack()
self.hyperdeck.gotoClip(self.selected_clip)
class MovementView(QWidget):
positionTableWidget = None
row = 0
columnList = "Asset Name;Buy Sell;Acquisition Date;Quantity;Price;Gross Amount;Net Amount;Comm %;Comm Amount; Comm VAT Amount".split(
";")
def __init__(self, movementList):
QWidget.__init__(self)
self.layout = QtGui.QGridLayout(self)
self.positionTableWidget = QTableWidget()
self.resize(800, 400)
self.positionTableWidget.setRowCount(10000)
self.positionTableWidget.setColumnCount(len(self.columnList))
self.positionTableWidget.setHorizontalHeaderLabels(self.columnList)
self.positionTableWidget.resizeColumnsToContents()
self.positionTableWidget.resizeRowsToContents()
self.layout.addWidget(self.positionTableWidget, 1, 0)
for (movement) in movementList:
self.renderMovements(movement)
self.positionTableWidget.setRowCount(self.row)
def renderMovements(self, movement):
#assetName
assetNameItem = QTableWidgetItemString(movement.asset.name, False)
self.positionTableWidget.setItem(
self.row, Constant.CONST_COLUMN_MOVEMENT_ASSET_NAME, assetNameItem)
#buysell
buySellItem = QTableWidgetItemString(movement.buySell, False)
self.positionTableWidget.setItem(
self.row, Constant.CONST_COLUMN_MOVEMENT_BUYSELL, buySellItem)
#acquisitionDate
acquisitionDateItem = QTableWidgetItemString(
movement.getAcquisitionDate(), False)
self.positionTableWidget.setItem(
self.row, Constant.CONST_COLUMN_MOVEMENT_ACQUISITION_DATE,
acquisitionDateItem)
#quantity
quantityItem = QTableWidgetItemInt(movement.quantity, False)
self.positionTableWidget.setItem(
self.row, Constant.CONST_COLUMN_MOVEMENT_QUANTITY, quantityItem)
#price
priceItem = QTableWidgetItem6Decimal(movement.price, False)
self.positionTableWidget.setItem(self.row,
Constant.CONST_COLUMN_MOVEMENT_PRICE,
priceItem)
#grossAmount
grossAmountItem = QTableWidgetItem6Decimal(movement.grossAmount, False)
self.positionTableWidget.setItem(
self.row, Constant.CONST_COLUMN_MOVEMENT_GROSS_AMOUNT,
grossAmountItem)
#netAmount
netAmountItem = QTableWidgetItem6Decimal(movement.netAmount, False)
self.positionTableWidget.setItem(
self.row, Constant.CONST_COLUMN_MOVEMENT_NET_AMOUNT, netAmountItem)
#commissionPercentage
commissionPercentageItem = QTableWidgetItem6Decimal(
movement.commissionPercentage, False)
self.positionTableWidget.setItem(
self.row, Constant.CONST_COLUMN_MOVEMENT_COMMISSION_PERCENTAGE,
commissionPercentageItem)
#commissionAmount
commissionAmountItem = QTableWidgetItem6Decimal(
movement.commissionAmount, False)
self.positionTableWidget.setItem(
self.row, Constant.CONST_COLUMN_MOVEMENT_COMMISSION_AMOUNT,
commissionAmountItem)
#commissionVATAmount
commissionVATAmountItem = QTableWidgetItem6Decimal(
movement.commissionVATAmount, False)
self.positionTableWidget.setItem(
self.row, Constant.CONST_COLUMN_MOVEMENT_COMMISSION_VAT_AMOUNT,
commissionVATAmountItem)
self.row += 1
class Ui_MainWindow(QMainWindow):
"""Cette classe contient tous les widgets de notre application."""
defaultPalette = QPalette()
defaultPalette.setColor(QPalette.Base, QColor("#151515"))
defaultPalette.setColor(QPalette.Text, Qt.white)
def __init__(self):
super(Ui_MainWindow, self).__init__()
# initialise la GUI avec un exemple
self.text = "Ceci est un petit texte d'exemple."
# les variables sont en place, initialise la GUI
self.initUI()
# exécute l'exemple
self.orgText.setText(self.text)
self.encode_text(False)
self.logLab.setText(
u"Saisir du texte ou importer un fichier, puis pousser \n"
u"le bouton correspondant à l'opération souhaitée.")
def initUI(self):
"""Met en place les éléments de l'interface."""
# -+++++++------------------- main window -------------------+++++++- #
self.setWindowTitle(u"Encodage / Décodage de Huffman")
self.centerAndResize()
centralwidget = QWidget(self)
mainGrid = QGridLayout(centralwidget)
mainGrid.setColumnMinimumWidth(0, 450)
# -+++++++------------------ groupe analyse -----------------+++++++- #
analysGroup = QGroupBox(u"Analyse", centralwidget)
self.analysGrid = QGridLayout(analysGroup)
# ----------- groupe de la table des codes ---------- #
codeTableGroup = QGroupBox(u"Table des codes", analysGroup)
codeTableGrid = QGridLayout(codeTableGroup)
# un tableau pour les codes
self.codesTableModel = MyTableModel()
self.codesTable = QTableView(codeTableGroup)
self.codesTable.setModel(self.codesTableModel)
self.codesTable.setFont(QFont("Mono", 8))
self.codesTable.resizeColumnsToContents()
self.codesTable.setSortingEnabled(True)
codeTableGrid.addWidget(self.codesTable, 0, 0, 1, 1)
self.analysGrid.addWidget(codeTableGroup, 1, 0, 1, 1)
# ----------- label du ratio de compression ---------- #
self.ratioLab = QLabel(u"Ratio de compression: ", analysGroup)
font = QFont()
font.setBold(True)
font.setWeight(75)
font.setKerning(True)
self.ratioLab.setFont(font)
self.analysGrid.addWidget(self.ratioLab, 2, 0, 1, 1)
# -+++++++-------- groupe de la table de comparaison --------+++++++- #
self.compGroup = QGroupBox(analysGroup)
self.compGroup.setTitle(u"Comparaisons")
compGrid = QGridLayout(self.compGroup)
# un tableau pour le ratio
self.compTable = QTableWidget(self.compGroup)
sizePolicy = QSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
sizePolicy.setHorizontalStretch(0)
sizePolicy.setVerticalStretch(0)
sizePolicy.setHeightForWidth(
self.compTable.sizePolicy().hasHeightForWidth())
self.compTable.setSizePolicy(sizePolicy)
self.compTable.setBaseSize(QSize(0, 0))
font = QFont()
font.setWeight(50)
self.compTable.setFont(font)
self.compTable.setEditTriggers(QAbstractItemView.NoEditTriggers)
self.compTable.setShowGrid(True)
self.compTable.setGridStyle(Qt.SolidLine)
# lignes / colonnes
self.compTable.setColumnCount(2)
self.compTable.setRowCount(3)
self.compTable.setVerticalHeaderItem(0, QTableWidgetItem("Taille (bits)"))
self.compTable.setVerticalHeaderItem(1, QTableWidgetItem("Entropie"))
self.compTable.setVerticalHeaderItem(2, QTableWidgetItem("Taille moy. (bits)"))
for i in range(2):
self.compTable.verticalHeaderItem(i).setTextAlignment(
Qt.AlignRight)
self.compTable.setHorizontalHeaderItem(0, QTableWidgetItem("ASCII"))
self.compTable.setHorizontalHeaderItem(1, QTableWidgetItem("Huffman"))
# nom des items
self.compTabASCIIMem = QTableWidgetItem()
self.compTable.setItem(0, 0, self.compTabASCIIMem)
self.compTabASCIIEnt = QTableWidgetItem()
self.compTable.setItem(1, 0, self.compTabASCIIEnt)
self.compTabASCIIAvg = QTableWidgetItem()
self.compTable.setItem(2, 0, self.compTabASCIIAvg)
self.compTabHuffMem = QTableWidgetItem()
self.compTable.setItem(0, 1, self.compTabHuffMem)
self.compTabHuffEnt = QTableWidgetItem()
self.compTable.setItem(1, 1, self.compTabHuffEnt)
self.compTabHuffAvg = QTableWidgetItem()
self.compTable.setItem(2, 1, self.compTabHuffAvg)
# parem du tableau
self.compTable.horizontalHeader().setCascadingSectionResizes(False)
self.compTable.verticalHeader().setVisible(True)
font = QFont("Mono", 8)
self.compTable.setFont(font)
compGrid.addWidget(self.compTable, 1, 0, 1, 1)
self.analysGrid.addWidget(self.compGroup, 0, 0, 1, 1)
mainGrid.addWidget(analysGroup, 0, 1, 1, 1)
# -+++++++----------------- groupe du texte -----------------+++++++- #
groupBox = QGroupBox(u"Texte", centralwidget)
textGrid = QGridLayout(groupBox)
# -+++++++------------- groupe du texte original ------------+++++++- #
orgTextGroup = QGroupBox(u"Texte original (Ctrl+T)", groupBox)
orgTextGrid = QGridLayout(orgTextGroup)
self.orgText = QTextEdit(orgTextGroup)
self.orgText.setPalette(self.defaultPalette)
orgTextGrid.addWidget(self.orgText, 0, 0, 1, 1)
textGrid.addWidget(orgTextGroup, 0, 0, 1, 2)
# -+++++++------------ groupe du texte compressé ------------+++++++- #
compressedTextGroup = QGroupBox(u"Texte compressé (Ctrl+H)", groupBox)
compressedTextGrid = QGridLayout(compressedTextGroup)
self.compressedText = QTextEdit(compressedTextGroup)
self.compressedText.setPalette(self.defaultPalette)
compressedTextGrid.addWidget(self.compressedText, 0, 0, 1, 1)
textGrid.addWidget(compressedTextGroup, 1, 0, 1, 2)
# -+++++++------------ groupe pour le texte ascii -----------+++++++- #
asciiTextGroup = QGroupBox(u"Texte ASCII", groupBox)
asciiTextGrid = QGridLayout(asciiTextGroup)
self.asciiText = QTextBrowser(asciiTextGroup)
self.asciiText.setPalette(self.defaultPalette)
asciiTextGrid.addWidget(self.asciiText, 0, 0, 1, 1)
textGrid.addWidget(asciiTextGroup, 2, 0, 1, 2)
# -+++++++-------------------- label de log -----------------+++++++- #
self.logLab = QLabel(analysGroup)
textGrid.addWidget(self.logLab, 3, 0, 1, 2)
# -+++++++----------- bouton pour encoder le texte ----------+++++++- #
self.encodeBut = QPushButton(groupBox)
self.encodeBut.setStatusTip(
u"Cliquez sur ce bouton pour encoder le texte original.")
self.encodeBut.setText(u"ENCODER")
self.encodeBut.clicked.connect(self.encode_text)
textGrid.addWidget(self.encodeBut, 4, 0, 1, 1)
# -+++++++----------- bouton pour décoder le texte ----------+++++++- #
self.decodeBut = QPushButton(groupBox)
self.decodeBut.setStatusTip(
u"Cliquez sur ce bouton pour décoder le texte compressé.")
self.decodeBut.setText(u"DÉCODER")
self.decodeBut.clicked.connect(self.decode_text)
textGrid.addWidget(self.decodeBut, 4, 1, 1, 1)
mainGrid.addWidget(groupBox, 0, 0, 1, 1)
self.setCentralWidget(centralwidget)
# -+++++++--------------- une barre de statut ---------------+++++++- #
self.setStatusBar(QStatusBar(self))
# -+++++++--------------------- le menu ---------------------+++++++- #
self.fileMenu = QMenu(u"Fichier")
self.fileMenu.addAction(u"Importer un texte...", self.open_text)
self.fileMenu.addAction(
u"Importer un texte encodé...", lambda: self.open_text(True))
self.fileMenu.addAction(u"Importer un dictionnaire...", self.open_dict)
self.fileMenu.addAction(u"Enregistrer le dictionnaire...", self.save_dict)
self.fileMenu.addAction(u"Quitter", self.close)
self.menuBar().addMenu(self.fileMenu)
QMetaObject.connectSlotsByName(self)
def open_text(self, compressed=False):
"""Ouvrir un fichier contenant un texte compressé ou non."""
fname, _ = QFileDialog.getOpenFileName(self, u'Ouvrir')
f = open(fname, 'r')
with f:
data = f.read()
if compressed:
self.compressedText.setText(data)
else:
self.orgText.setText(data)
def open_dict(self):
"""Ouvrir un dictionnaire de codes Huffman."""
fname, _ = QFileDialog.getOpenFileName(self, u'Ouvrir')
self.occ = {}
self.hCodes = {}
self.aCodes = {}
self.hCost = {}
self.aCost = {}
self.hCodes = create_dict_from_file(fname)
self.update_codes_table()
self.logLab.setText(u"Dictionnaire de Huffman importé.")
return self.hCodes
def save_dict(self):
"""Enregistrer le dictionnaire de codes Huffman."""
fname, _ = QFileDialog.getSaveFileName(self, "Enregistrer sous")
save_dict_to_file(fname, self.hCodes)
def make_tab_rows(self):
"""Génère le remplissage des lignes du tableau des codes."""
dictList = [self.occ, self.aCodes, self.hCodes, self.aCost, self.hCost]
tabList = []
charList = self.hCodes.keys() if self.hCodes else self.occ.keys()
for char in charList:
l = ["'" + char + "'"]
for dic in dictList:
try:
l.append(dic[char])
except KeyError:
l.append('')
tabList.append(l)
return tabList
def encode_text(self, wizard=True):
"""Encode le texte original."""
self.text = self.orgText.toPlainText().encode('utf-8')
if not self.text:
self.compressedText.setText(u"")
self.asciiText.setText(u"")
self.logLab.setText(
u"<font color=#A52A2A>Rien à compresser.</font>")
return
self.occ = {}
self.tree = ()
self.hCodes = {}
self.aCodes = {}
self.hCost = {}
self.aCost = {}
self.hSqueezed = []
self.aSqueezed = []
self.stringHSqueezed = ''
self.stringASqueezed = ''
if wizard:
self.launch_wizard(
EncodeWizard(self),
u"<font color=#008000>Compression achevée.</font>")
else:
self.make_occ()
self.make_tree()
self.make_codes()
self.make_cost()
self.make_encoded_text()
self.make_comp()
def decode_text(self, wizard=True):
"""Décode le texte compressé."""
self.codeString = str(
self.compressedText.toPlainText().replace(' ', ''))
if not self.codeString or not self.hCodes:
self.orgText.setText(u"")
self.asciiText.setText(u"")
if not self.codeString:
self.logLab.setText(
u"<font color=#A52A2A>Rien à décompresser.</font>")
if not self.hCodes:
self.logLab.setText(
u"<font color=#A52A2A>Dictionnaire indisponible pour la décompression.</font>")
return
self.text = ''
self.stringASqueezed = ''
if wizard:
self.launch_wizard(
DecodeWizard(self),
u"<font color=#008000>Texte décompressé.</font>")
else:
self.make_code_map()
self.make_decoded_text()
def launch_wizard(self, wizard, finishString):
"""Lance l'assistant d'en/décodage pour guider l'utilisateur.
Cache les options inaccessibles pendant l'assistant.
"""
self.logLab.setText(
u"<font color=#9E6A00>Opération en cours. "
u"Suivre les indications.</font>")
disItems = [self.orgText, self.compressedText, self.encodeBut,
self.decodeBut, self.fileMenu.actions()[1],
self.fileMenu.actions()[2], self.fileMenu.actions()[3]]
for item in disItems:
item.setEnabled(False)
self.compGroup.setVisible(False)
self.analysGrid.addWidget(wizard, 0, 0, 1, 1)
res = wizard.exec_()
if res:
self.logLab.setText(finishString)
else:
self.logLab.setText(
u"<font color=#A52A2A>Opération interrompue.</font>")
for item in disItems:
item.setEnabled(True)
self.compGroup.setVisible(True)
def update_ratio_lab(self):
"""Replace la valeur du label du ratio de compression."""
if not self.stringASqueezed:
val = '/'
else:
val = (len(self.stringHSqueezed.replace(' ', '')) /
float(len(self.stringASqueezed.replace(' ', ''))))
self.ratioLab.setText(unicode('Taux de compression: ' + str(val)))
def update_comp_table(self):
"""Met à jour le tableau de comparaison ASCII VS Huffman."""
self.compTabASCIIMem.setText(unicode(len(''.join(self.aSqueezed))))
self.compTabHuffMem.setText(unicode(len(''.join(self.hSqueezed))))
# entropie ?
self.compTabASCIIEnt.setText('0')
self.compTabHuffEnt.setText('0')
self.compTabASCIIAvg.setText(unicode(8))
self.compTabHuffAvg.setText(unicode(
average_code_length(self.hSqueezed)))
self.compTable.resizeColumnsToContents()
def update_codes_table(self, hlRow=[]):
"""Met à jour le tableau des codes et surligne les lignes de hlRow."""
self.codesTableModel.hlRow = hlRow
self.codesTableModel.emptyTable()
self.codesTableModel.fillTable(self.make_tab_rows())
self.codesTable.resizeColumnsToContents()
def centerAndResize(self):
"""Centre et redimmensionne le widget."""
screen = QDesktopWidget().screenGeometry()
self.resize(screen.width() / 1.6, screen.height() / 1.4)
size = self.geometry()
self.move(
(screen.width() - size.width()) / 2,
(screen.height() - size.height()) / 2)
#===================== METHODS FOR EN/DECODE WIZARDS =====================#
def make_encode_init(self):
self.compressedText.setText(unicode(self.stringHSqueezed))
self.asciiText.setText(unicode(self.stringASqueezed))
self.orgText.setTextColor(QColor(Qt.darkGreen))
self.orgText.setText(unicode(self.text.decode('utf-8')))
self.update_codes_table()
def make_occ(self):
self.orgText.setTextColor(QColor(Qt.white))
self.orgText.setText(unicode(self.text.decode('utf-8')))
self.occ = occurences(self.text)
self.update_codes_table([0, 1])
def make_tree(self):
self.tree = make_trie(self.occ)
self.update_codes_table()
def make_codes(self):
self.hCodes = make_codes(self.tree, {})
self.aCodes = make_ascii_codes(self.occ.keys(), {})
self.update_codes_table([2, 3])
def make_cost(self):
self.hCost = tree_cost(self.hCodes, self.occ)
self.aCost = tree_cost(self.aCodes, self.occ)
self.update_codes_table([4, 5])
def make_encoded_text(self):
self.hSqueezed = squeeze(self.text, self.hCodes)
self.aSqueezed = squeeze(self.text, self.aCodes)
self.stringHSqueezed = ' '.join(self.hSqueezed)
self.stringASqueezed = ' '.join(self.aSqueezed)
self.compressedText.setTextColor(QColor(Qt.darkGreen))
self.asciiText.setTextColor(QColor(Qt.darkYellow))
self.compressedText.setText(unicode(self.stringHSqueezed))
self.asciiText.setText(unicode(self.stringASqueezed))
self.update_codes_table()
def make_comp(self):
self.compressedText.setTextColor(QColor(Qt.white))
self.asciiText.setTextColor(QColor(Qt.white))
self.compressedText.setText(unicode(self.stringHSqueezed))
self.asciiText.setText(unicode(self.stringASqueezed))
self.update_codes_table()
self.update_comp_table()
self.update_ratio_lab()
def make_decode_init(self):
self.orgText.setText(unicode(self.text))
self.asciiText.setText(unicode(self.stringASqueezed))
self.compressedText.setTextColor(QColor(Qt.darkGreen))
self.compressedText.setText(self.compressedText.toPlainText())
def make_code_map(self):
self.compressedText.setTextColor(QColor(Qt.white))
self.compressedText.setText(self.compressedText.toPlainText())
self.orgText.setText(unicode(self.text))
self.asciiText.setText(unicode(self.stringASqueezed))
self.codeMap = dict(zip(self.hCodes.values(), self.hCodes.keys()))
self.update_codes_table([0, 3])
def make_decoded_text(self):
self.unSqueezed = unsqueeze(self.codeString, self.codeMap)
self.text = ''.join(self.unSqueezed)
self.orgText.setTextColor(QColor(Qt.darkGreen))
self.orgText.setText(unicode(self.text.decode('utf-8')))
self.asciiText.setText(unicode(self.stringASqueezed))
self.update_codes_table()
def make_ascii_decode(self):
self.orgText.setTextColor(QColor(Qt.white))
self.orgText.setText(unicode(self.text.decode('utf-8')))
self.aCodes = make_ascii_codes(self.codeMap.values(), {})
self.aSqueezed = squeeze(self.text, self.aCodes)
self.stringASqueezed = ' '.join(self.aSqueezed)
self.occ = occurences(self.text)
self.hCost = tree_cost(self.hCodes, self.occ)
self.aCost = tree_cost(self.aCodes, self.occ)
self.asciiText.setTextColor(QColor(Qt.darkGreen))
self.asciiText.setText(unicode(self.stringASqueezed))
self.update_codes_table([1, 2, 4, 5])
class SubscriberDialog(QDialog):
if USE_MAEMO_5:
switchRequested = Signal()
def __init__(self, parent=None):
QDialog.__init__(self, parent)
self.ui = Ui_SubscriberDialog()
self.ui.setupUi(self)
self.subscriber = None
self.tableWidget = None
self.listWidget = None
if USE_MAEMO_5:
switchButton = self.ui.buttonBox.addButton(self.tr('Switch'), QDialogButtonBox.ActionRole)
switchButton.clicked.connect(self.switchRequested)
self.tableWidget = self.ui.tableWidget
headerLabels = ('Key', 'Value', 'Type')
self.tableWidget.setColumnCount(3)
self.tableWidget.setHorizontalHeaderLabels(headerLabels)
horizontalHeader = self.tableWidget.horizontalHeader()
horizontalHeader.setStretchLastSection(True)
verticalHeader = self.tableWidget.verticalHeader()
verticalHeader.setVisible(False)
self.tableWidget.setColumnWidth(0, 200)
self.tableWidget.setColumnWidth(1, 400)
else:
desktopWidget = QDesktopWidget()
if desktopWidget.availableGeometry().width() < 400:
# Screen is too small to fit a table widget without scrolling, use a list widget instead.
self.listWidget = QListWidget()
self.listWidget.setAlternatingRowColors(True)
self.ui.verticalLayout.insertWidget(2, self.listWidget)
else:
self.tableWidget = QTableWidget()
headerLabels = ('Key', 'Value', 'Type')
self.tableWidget.setColumnCount(3)
self.tableWidget.setHorizontalHeaderLabels(headerLabels)
horizontalHeader = self.tableWidget.horizontalHeader()
horizontalHeader.setStretchLastSection(True)
self.tableWidget.verticalHeader()
self.setVisible(False)
self.ui.verticalLayout.insertWidget(2, self.tableWidget)
self.ui.connectButton.clicked.connect(self.changeSubscriberPath)
self.changeSubscriberPath()
# if the default path does not exist reset it to /
value = self.subscriber.value()
subPaths = self.subscriber.subPaths()
if not value and not subPaths:
self.ui.basePath.setText('/')
self.changeSubscriberPath()
def changeEvent(self, e):
QDialog.changeEvent(self, e)
if e.type() == QEvent.LanguageChange:
self.ui.retranslateUi(self)
def changeSubscriberPath(self):
if self.listWidget:
self.listWidget.clear()
elif self.tableWidget:
self.tableWidget.clearContents()
if not self.subscriber:
self.subscriber = QValueSpaceSubscriber(self.ui.basePath.text(), self)
else:
self.subscriber.setPath(self.ui.basePath.text())
self.subscriber.contentsChanged.connect(self.subscriberChanged)
self.subscriber.connectNotify("contentsChanged()")
self.subscriberChanged()
def subscriberChanged(self):
subPaths = self.subscriber.subPaths()
if self.listWidget:
self.listWidget.clear()
elif self.tableWidget:
self.tableWidget.clearContents()
self.tableWidget.setRowCount(len(subPaths))
for i in xrange(len(subPaths)):
v = self.subscriber.value(subPaths[i])
if self.listWidget:
item = QListWidgetItem('%s (%s)\n%s' % (subPaths[i], str(type(v)), str(v)))
item.setFlags(item.flags() & ~Qt.ItemIsEditable)
self.listWidget.addItem(item)
elif self.tableWidget:
pathItem = QTableWidgetItem(subPaths[i])
pathItem.setFlags(pathItem.flags() & ~Qt.ItemIsEditable)
valueItem = QTableWidgetItem(str(v))
valueItem.setFlags(pathItem.flags() & ~Qt.ItemIsEditable)
typeItem = QTableWidgetItem(str(type(v)))
typeItem.setFlags(pathItem.flags() & ~Qt.ItemIsEditable)
self.tableWidget.setItem(i, 0, pathItem)
self.tableWidget.setItem(i, 1, valueItem)
self.tableWidget.setItem(i, 2, typeItem)
class MainWindow(QMainWindow):
def __init__(self, parent = None):
super(MainWindow, self).__init__(parent)
self.setWindowTitle('Basic Drawing')
self.app = QCoreApplication.instance()
self.setGeometry(self.app.myRectangle)
st = self.app.style()
self.params = ( ('ButtonMargin', st.pixelMetric(QStyle.PM_ButtonMargin)),
('DockWidgetTitleBarButtonMargin', st.pixelMetric(QStyle.PM_DockWidgetTitleBarButtonMargin)),
('ButtonDefaultIndicator', st.pixelMetric(QStyle.PM_ButtonDefaultIndicator)),
('MenuButtonIndicator', st.pixelMetric(QStyle.PM_MenuButtonIndicator)),
('ButtonShiftHorizontal', st.pixelMetric(QStyle.PM_ButtonShiftHorizontal)),
('ButtonShiftVertical', st.pixelMetric(QStyle.PM_ButtonShiftVertical)),
('DefaultFrameWidth', st.pixelMetric(QStyle.PM_DefaultFrameWidth)),
('SpinBoxFrameWidth', st.pixelMetric(QStyle.PM_SpinBoxFrameWidth)),
('ComboBoxFrameWidth', st.pixelMetric(QStyle.PM_ComboBoxFrameWidth)),
('MdiSubWindowFrameWidth', st.pixelMetric(QStyle.PM_MdiSubWindowFrameWidth)),
('MdiSubWindowMinimizedWidth', st.pixelMetric(QStyle.PM_MdiSubWindowMinimizedWidth)),
('LayoutLeftMargin', st.pixelMetric(QStyle.PM_LayoutLeftMargin)),
('LayoutTopMargin', st.pixelMetric(QStyle.PM_LayoutTopMargin)),
('LayoutRightMargin', st.pixelMetric(QStyle.PM_LayoutRightMargin)),
('LayoutBottomMargin', st.pixelMetric(QStyle.PM_LayoutBottomMargin)),
('LayoutHorizontalSpacing', st.pixelMetric(QStyle.PM_LayoutHorizontalSpacing)),
('LayoutVerticalSpacing', st.pixelMetric(QStyle.PM_LayoutVerticalSpacing)),
('MaximumDragDistance', st.pixelMetric(QStyle.PM_MaximumDragDistance)),
('ScrollBarExtent', st.pixelMetric(QStyle.PM_ScrollBarExtent)),
('ScrollBarSliderMin', st.pixelMetric(QStyle.PM_ScrollBarSliderMin)),
('SliderThickness', st.pixelMetric(QStyle.PM_SliderThickness)),
('SliderControlThickness', st.pixelMetric(QStyle.PM_SliderControlThickness)),
('SliderLength', st.pixelMetric(QStyle.PM_SliderLength)),
('SliderTickmarkOffset', st.pixelMetric(QStyle.PM_SliderTickmarkOffset)),
('SliderSpaceAvailable', st.pixelMetric(QStyle.PM_SliderSpaceAvailable)),
('DockWidgetSeparatorExtent', st.pixelMetric(QStyle.PM_DockWidgetSeparatorExtent)),
('DockWidgetHandleExtent', st.pixelMetric(QStyle.PM_DockWidgetHandleExtent)),
('DockWidgetFrameWidth', st.pixelMetric(QStyle.PM_DockWidgetFrameWidth)),
('DockWidgetTitleMargin', st.pixelMetric(QStyle.PM_DockWidgetTitleMargin)),
('MenuBarPanelWidth', st.pixelMetric(QStyle.PM_MenuBarPanelWidth)),
('MenuBarItemSpacing', st.pixelMetric(QStyle.PM_MenuBarItemSpacing)),
('MenuBarHMargin', st.pixelMetric(QStyle.PM_MenuBarHMargin)),
('MenuBarVMargin', st.pixelMetric(QStyle.PM_MenuBarVMargin)),
('ToolBarFrameWidth', st.pixelMetric(QStyle.PM_ToolBarFrameWidth)),
('ToolBarHandleExtent', st.pixelMetric(QStyle.PM_ToolBarHandleExtent)),
('ToolBarItemMargin', st.pixelMetric(QStyle.PM_ToolBarItemMargin)),
('ToolBarItemSpacing', st.pixelMetric(QStyle.PM_ToolBarItemSpacing)),
('ToolBarSeparatorExtent', st.pixelMetric(QStyle.PM_ToolBarSeparatorExtent)),
('ToolBarExtensionExtent', st.pixelMetric(QStyle.PM_ToolBarExtensionExtent)),
('TabBarTabOverlap', st.pixelMetric(QStyle.PM_TabBarTabOverlap)),
('TabBarTabHSpace', st.pixelMetric(QStyle.PM_TabBarTabHSpace)),
('TabBarTabVSpace', st.pixelMetric(QStyle.PM_TabBarTabVSpace)),
('TabBarBaseHeight', st.pixelMetric(QStyle.PM_TabBarBaseHeight)),
('TabBarBaseOverlap', st.pixelMetric(QStyle.PM_TabBarBaseOverlap)),
('TabBarScrollButtonWidth', st.pixelMetric(QStyle.PM_TabBarScrollButtonWidth)),
('TabBarTabShiftHorizontal', st.pixelMetric(QStyle.PM_TabBarTabShiftHorizontal)),
('TabBarTabShiftVertical', st.pixelMetric(QStyle.PM_TabBarTabShiftVertical)),
('ProgressBarChunkWidth', st.pixelMetric(QStyle.PM_ProgressBarChunkWidth)),
('SplitterWidth', st.pixelMetric(QStyle.PM_SplitterWidth)),
('TitleBarHeight', st.pixelMetric(QStyle.PM_TitleBarHeight)),
('IndicatorWidth', st.pixelMetric(QStyle.PM_IndicatorWidth)),
('IndicatorHeight', st.pixelMetric(QStyle.PM_IndicatorHeight)),
('ExclusiveIndicatorWidth', st.pixelMetric(QStyle.PM_ExclusiveIndicatorWidth)),
('ExclusiveIndicatorHeight', st.pixelMetric(QStyle.PM_ExclusiveIndicatorHeight)),
('MenuPanelWidth', st.pixelMetric(QStyle.PM_MenuPanelWidth)),
('MenuHMargin', st.pixelMetric(QStyle.PM_MenuHMargin)),
('MenuVMargin', st.pixelMetric(QStyle.PM_MenuVMargin)),
('MenuScrollerHeight', st.pixelMetric(QStyle.PM_MenuScrollerHeight)),
('MenuTearoffHeight', st.pixelMetric(QStyle.PM_MenuTearoffHeight)),
('MenuDesktopFrameWidth', st.pixelMetric(QStyle.PM_MenuDesktopFrameWidth)),
('CheckListButtonSize', st.pixelMetric(QStyle.PM_CheckListButtonSize)),
('CheckListControllerSize', st.pixelMetric(QStyle.PM_CheckListControllerSize)),
('HeaderMarkSize', st.pixelMetric(QStyle.PM_HeaderMarkSize)),
('HeaderGripMargin', st.pixelMetric(QStyle.PM_HeaderGripMargin)),
('HeaderMargin', st.pixelMetric(QStyle.PM_HeaderMargin)),
('SpinBoxSliderHeight', st.pixelMetric(QStyle.PM_SpinBoxSliderHeight)),
('ToolBarIconSize', st.pixelMetric(QStyle.PM_ToolBarIconSize)),
('SmallIconSize', st.pixelMetric(QStyle.PM_SmallIconSize)),
('LargeIconSize', st.pixelMetric(QStyle.PM_LargeIconSize)),
('FocusFrameHMargin', st.pixelMetric(QStyle.PM_FocusFrameHMargin)),
('FocusFrameVMargin', st.pixelMetric(QStyle.PM_FocusFrameVMargin)),
('IconViewIconSize', st.pixelMetric(QStyle.PM_IconViewIconSize)),
('ListViewIconSize', st.pixelMetric(QStyle.PM_ListViewIconSize)),
('ToolTipLabelFrameWidth', st.pixelMetric(QStyle.PM_ToolTipLabelFrameWidth)),
('CheckBoxLabelSpacing', st.pixelMetric(QStyle.PM_CheckBoxLabelSpacing)),
('RadioButtonLabelSpacing', st.pixelMetric(QStyle.PM_RadioButtonLabelSpacing)),
('TabBarIconSize', st.pixelMetric(QStyle.PM_TabBarIconSize)),
('SizeGripSize', st.pixelMetric(QStyle.PM_SizeGripSize)),
('MessageBoxIconSize', st.pixelMetric(QStyle.PM_MessageBoxIconSize)),
('ButtonIconSize', st.pixelMetric(QStyle.PM_ButtonIconSize)),
('TextCursorWidth', st.pixelMetric(QStyle.PM_TextCursorWidth)),
('TabBar_ScrollButtonOverlap', st.pixelMetric(QStyle.PM_TabBar_ScrollButtonOverlap)),
('TabCloseIndicatorWidth', st.pixelMetric(QStyle.PM_TabCloseIndicatorWidth)),
('TabCloseIndicatorHeight', st.pixelMetric(QStyle.PM_TabCloseIndicatorHeight)),
('CustomBase', st.pixelMetric(QStyle.PM_CustomBase))
)
self.tw = QTableWidget(len(self.params), 2)
length = 0
strin = ''
for i in range(len(self.params)):
item1 = QTableWidgetItem (self.params[i][0])
item2 = QTableWidgetItem (str(self.params[i][1]))
self.tw.setItem(i, 0, item1)
self.tw.setItem(i, 1, item2)
self.setCentralWidget(self.tw)
#fm = QFontMetrics(self.tw.font())
fm = QFontMetrics(self.tw.item(1,1).font())
#fm = self.tw.fontMetrics()
for param in self.params:
l = fm.size(1, self.tr(param[0])).width()
if l > length:
length = l
strin = self.tr(param[0])
print 'MaxWidth: {0}, for string: {1}'.format(length, strin)
self.tw.setColumnWidth(0, length)
appFont = self.app.font()
cellFont = self.tw.item(1,1).font()
tableFont = self.tw.font()
print appFont
print cellFont
print tableFont
print fm
def setItem(self, row, column, item):
QTableWidget.setItem(self,row,column,item)
item.setBackground(self.appState.BACKGROUND_COLOR)
self.numColumns = max(column,self.numColumns)
class RecorderClipSelectionScreen(QWidget):
def __init__(self, hyperdeck, state, mainWindow):
super(RecorderClipSelectionScreen, self).__init__()
self.hyperdeck = hyperdeck
self.state = state
self.mainWindow = mainWindow
self.selected_clip = None
layout = QGridLayout()
lblTitle = TitleLabel("Select clip")
layout.addWidget(lblTitle, 0, 0, 1, 3)
self.clipTable = QTableWidget()
self.clipTable.setColumnCount(2)
self.clipTable.setHorizontalHeaderLabels(['ID', 'Clip name'])
self.clipTable.horizontalHeader().setStretchLastSection(True)
self.clipTable.setSelectionBehavior(QAbstractItemView.SelectionBehavior.SelectRows)
self.clipTable.itemSelectionChanged.connect(self._onClipSelected)
layout.addWidget(self.clipTable, 1, 0, 1, 3)
b = ExpandingButton()
b.setText("Back")
b.setIcon(QIcon(":icons/go-previous"))
b.setToolButtonStyle(Qt.ToolButtonStyle.ToolButtonTextBesideIcon)
b.clicked.connect(mainWindow.stepBack)
layout.addWidget(b, 2, 0)
btnRefresh = ExpandingButton()
btnRefresh.setText('Refresh')
btnRefresh.setIcon(QIcon(':icons/refresh'))
btnRefresh.setToolButtonStyle(Qt.ToolButtonStyle.ToolButtonTextBesideIcon)
def refresh():
self.populateClipsList({})
hyperdeck.broadcastClipsList()
btnRefresh.clicked.connect(refresh)
layout.addWidget(btnRefresh, 2, 1)
self.btnSelect = ExpandingButton()
self.btnSelect.setText("Cue clip")
self.btnSelect.clicked.connect(self._cueClip)
layout.addWidget(self.btnSelect, 2, 2)
layout.setRowStretch(0, 0)
layout.setRowStretch(1, 1)
layout.setRowStretch(2, 0)
self.setLayout(layout)
self.populateClipsList(state.clip_listing)
def populateClipsList(self, clipsList):
self.selected_clip = None
self.btnSelect.setEnabled(False)
self.clipTable.clearContents()
self.clipTable.setRowCount(len(clipsList))
for idx, (clip_id, val) in enumerate(clipsList.iteritems()):
self.clipTable.setItem(idx, 0, QTableWidgetItem(str(clip_id)))
self.clipTable.setItem(idx, 1, QTableWidgetItem(val['name']))
def _updateClipSelectionFromState(self, state):
if 'clip id' in state:
clip_id = state['clip id']
for row in range(self.clipTable.rowCount()):
this_id = self.clipTable.item(row, 0)
if this_id.text() == str(clip_id):
self.clipTable.selectRow(row)
break
def _onClipSelected(self):
selected_items = self.clipTable.selectedItems()
if selected_items:
self.selected_clip = int(selected_items[0].text())
self.btnSelect.setEnabled(True)
else:
self.btnSelect.setEnabled(False)
def _cueClip(self):
if self.selected_clip:
self.mainWindow.stepBack()
self.hyperdeck.gotoClip(self.selected_clip)
class MainWindow(QDialog):
"""monkey_linux UI"""
def __init__(self,parent=None):
super(MainWindow, self).__init__()
self.init_conf()
self.setParent(parent)
common.Log.info("set up ui")
self.setup_ui()
self.findDeviceAction.triggered.connect(self.get_device_status)
self.deleteDeviceAction.triggered.connect(self.del_device)
# self.storeButton.clicked.connect(self.set_conf)
self.startButton.clicked.connect(self.start)
self.stopButton.clicked.connect(self.stop)
self.checkLogButton.clicked.connect(self.check)
self.monkeyButton.clicked.connect(self.checkMonkeyLog)
self.exportButton.clicked.connect(self.exportConf)
self.importButton.clicked.connect(self.importConf)
self.setAbout.triggered.connect(self.about)
self.startTime = datetime.datetime.now()
self.secsTime = float(1) * 60 * 60
def setup_ui(self):
# main window width hand height
# self.setMinimumWidth(600)
self.setMaximumWidth(800)
self.setMinimumHeight(600)
# main window title
self.setWindowTitle(static.title)
# file menu bar
self.menuBar = QMenuBar()
self.menuBar.setMaximumHeight(23)
# self.menuFile = self.menuBar.addMenu(static.menuFile)
# self.importAction = QAction(QIcon(static.importPNG), static.importFile, self)
# self.exportAction = QAction(QIcon(static.exportPNG), static.exportFile, self)
# self.menuFile.addAction(self.importAction)
# self.menuFile.addAction(self.exportAction)
self.setEnvActioin = QAction(QIcon(static.setPNG), static.pcSet, self)
self.menuSet = self.menuBar.addMenu(static.menuSet)
self.menuSet.addAction(self.setEnvActioin)
self.setAbout = QAction(QIcon(static.setPNG), static.menuAbout, self)
self.setAbout.setStatusTip('About') # 状态栏提示
self.menuHelp = self.menuBar.addMenu(static.menuHelp)
self.menuHelp.addAction(self.setAbout)
# set all layout
self.hbox = QHBoxLayout(self)
# device ========
self.topLeft = QFrame(self)
self.topLeft.setMaximumSize(218, 300)
self.topLeft.setMinimumSize(218, 200)
self.topLeft.setFrameShape(QFrame.StyledPanel)
self.topLeftLayout = QVBoxLayout(self.topLeft)
self.toolBar = QToolBar()
# self.androidDeviceAction = QRadioButton('Android', self)
# self.androidDeviceAction.setFocusPolicy(Qt.NoFocus)
# self.ipDeviceAction = QRadioButton('IP', self)
# self.ipDeviceAction.setFocusPolicy(Qt.NoFocus)
# self.ipDeviceAction.move(10, 10)
# self.ipDeviceAction.toggle()
self.findDeviceAction = QAction(QIcon(static.findDevice), static.findDeviceButton, self)
self.deleteDeviceAction = QAction(QIcon(static.deleteDevice), static.deleteDeviceButton, self)
# self.toolBar.addWidget(self.androidDeviceAction)
# self.toolBar.addWidget(self.ipDeviceAction)
self.toolBar.addAction(self.findDeviceAction)
self.toolBar.addAction(self.deleteDeviceAction)
self.deviceLab = QLabel(static.deviceName, self)
self.device = QTableWidget(1, 2)
self.device.setHorizontalHeaderLabels(['name', 'status'])
self.device.setColumnWidth(0, 100)
self.device.setColumnWidth(1, 80)
self.topLeftLayout.addWidget(self.deviceLab)
self.topLeftLayout.addWidget(self.toolBar)
self.topLeftLayout.addWidget(self.device)
# set button or other for running monkey or not and status of device and log ========
self.topRight = QFrame(self)
self.topRight.setFrameShape(QFrame.StyledPanel)
self.topRight.setMaximumHeight(40)
self.startButton = QPushButton(QIcon(static.startPNG), "")
self.stopButton = QPushButton(QIcon(static.stopPNG), "")
self.status = QLabel(static.status)
self.statusEdit = QLineEdit(self)
self.statusEdit.setReadOnly(True)
self.statusEdit.setMaximumWidth(80)
self.statusEdit.setMinimumWidth(80)
self.statusEdit.setText("")
# check log
self.checkLogButton = QPushButton(static.checkLog)
self.checkLogButton.setMaximumHeight(20)
self.checkLogButton.setMinimumHeight(20)
self.checkLogButton.setMaximumWidth(60)
self.selectLog = QLabel(static.selectlog)
self.logfile = QComboBox()
self.dirlist = os.listdir(os.path.join(DIR, "Result"))
for d in self.dirlist:
if d != "AutoMonkey.log":
self.logfile.insertItem(0, d)
self.logfile.setMaximumWidth(150)
self.logfile.setMaximumHeight(20)
self.logfile.setMinimumHeight(20)
self.topLayout = QHBoxLayout(self.topRight)
self.topLayout.addWidget(self.startButton)
self.topLayout.addWidget(self.stopButton)
self.topLayout.addWidget(self.status)
self.topLayout.addWidget(self.statusEdit)
self.topLayout.addWidget(self.selectLog)
self.topLayout.addWidget(self.logfile)
self.topLayout.addWidget(self.checkLogButton)
# set parameter for monkey =======
self.midRight = QFrame(self)
self.midRight.setMaximumSize(555, 200)
self.midRight.setMinimumSize(555, 200)
self.midRight.setFrameShape(QFrame.StyledPanel)
self.midRightLayout = QVBoxLayout(self.midRight)
self.subLayout0 = QVBoxLayout()
self.subLayout1 = QVBoxLayout()
self.subLayout2 = QHBoxLayout()
self.subLayout3 = QVBoxLayout()
self.subLayout4 = QVBoxLayout()
self.subLayout5 = QHBoxLayout()
self.subLayout6 = QHBoxLayout()
self.toolBar = QToolBar()
# self.storeAction = QAction(QIcon(static.storePNG), static.storeButton, self)
self.startAction = QAction(QIcon(static.startPNG), static.startButton, self)
self.stopAction = QAction(QIcon(static.stopPNG), static.stopButton, self)
# self.toolBar.addAction(self.storeAction)
self.toolBar.addAction(self.startAction)
self.toolBar.addAction(self.stopAction)
self.timeLongLbl = QLabel(static.timeString, self)
self.timeLong = QLineEdit(self)
self.timeLong.setMaximumWidth(100)
self.timeLong.setMinimumWidth(100)
self.timeLong.setPlaceholderText(static.timeLong)
self.timeLongUnit = QLabel("H")
self.etSetLbl = QLabel(static.eventTypeSet)
self.etSet = QTableWidget(2, 2)
self.etSet.setMaximumHeight(150)
self.etSet.setHorizontalHeaderLabels(['option', 'value'])
self.etSet.horizontalHeader().setStretchLastSection(True)
self.etSet.setItem(0, 0, QTableWidgetItem("--throttle"))
self.etSet.setItem(0, 1, QTableWidgetItem(str(static.eventType["--throttle"])))
# set event type percent
self.etPercentLbl = QLabel(static.eventTpyePercent, self)
self.etPercent = QTableWidget(2, 2)
self.etPercent.setMaximumHeight(150)
self.etPercent.setHorizontalHeaderLabels(['option', 'value'])
self.etPercent.horizontalHeader().setStretchLastSection(True)
self.etPercent.setItem(0, 0, QTableWidgetItem("--pct-touch"))
self.etPercent.setItem(0, 1, QTableWidgetItem(str(static.eventPercent["--pct-touch"])))
self.etPercent.setItem(1, 0, QTableWidgetItem("--pct-motion"))
self.etPercent.setItem(1, 1, QTableWidgetItem(str(static.eventPercent["--pct-motion"])))
# self.storeButton = QPushButton(QIcon(static.storePNG), static.storeButton)
# self.storeButton.setToolTip(static.storeButton)
self.exportButton = QPushButton(QIcon(static.exportPNG), static.exportFile)
self.exportButton.setToolTip(static.exportFile)
self.importButton = QPushButton(QIcon(static.importPNG), static.importFile)
self.importButton.setToolTip(static.importFile)
self.subLayout2.addWidget(self.timeLongLbl)
self.subLayout2.addWidget(self.timeLong)
self.subLayout2.addWidget(self.timeLongUnit)
self.subLayout2.addWidget(QLabel(" " * 300))
# self.subLayout2.addWidget(self.storeButton)
self.subLayout2.addWidget(self.exportButton)
self.subLayout2.addWidget(self.importButton)
self.subLayout0.addLayout(self.subLayout2)
self.subLayout3.addWidget(self.etSetLbl)
self.subLayout3.addWidget(self.etSet)
self.subLayout4.addWidget(self.etPercentLbl)
self.subLayout4.addWidget(self.etPercent)
self.subLayout5.addLayout(self.subLayout0)
self.subLayout6.addLayout(self.subLayout3)
self.subLayout6.addLayout(self.subLayout4)
self.midRightLayout.addLayout(self.subLayout5)
self.midRightLayout.addLayout(self.subLayout6)
# log ========
self.bottom = QFrame(self)
self.bottom.setFrameShape(QFrame.StyledPanel)
# log information
self.logInfo = QLabel(static.logInfo)
# information filter
self.logFilter = QLabel(static.logFilter)
self.monkeyButton = QPushButton(static.openMonkeyLog)
self.combo = QComboBox()
for i in range(len(static.logLevel)):
self.combo.addItem(static.logLevel[i])
self.combo.setMaximumWidth(55)
self.combo.setMaximumHeight(20)
self.combo.setMinimumHeight(20)
# information details
self.bottomLayout = QVBoxLayout(self.bottom)
self.subLayout = QHBoxLayout()
self.subLayout.addWidget(self.logInfo)
for i in range(10):
self.subLayout.addWidget(QLabel(""))
self.subLayout.addWidget(self.monkeyButton)
self.subLayout.addWidget(self.logFilter)
self.subLayout.addWidget(self.combo)
self.bottomLayout.addLayout(self.subLayout)
self.tabwidget = TabWidget()
self.tabwidget.setMinimumHeight(100)
self.bottomLayout.addWidget(self.tabwidget)
# splitter mainWindow ++++++++++++++++++++++++++++++++++++
self.splitter2 = QSplitter(Qt.Vertical)
self.splitter2.addWidget(self.topRight)
self.splitter2.addWidget(self.midRight)
self.splitter0 = QSplitter(Qt.Horizontal)
self.splitter0.addWidget(self.topLeft)
self.splitter0.addWidget(self.splitter2)
self.splitter1 = QSplitter(Qt.Vertical)
self.splitter1.addWidget(self.menuBar)
self.splitter1.addWidget(self.splitter0)
self.splitter1.addWidget(self.bottom)
self.hbox.addWidget(self.splitter1)
self.setLayout(self.hbox)
self.show()
def about(self):
common.showDialog(static.menuAbout, static.dialogAbout)
def init_conf(self):
common.Log.info("init monkey conf")
with open(monkeyConfFile, "w") as f:
f.write("deviceList = []" + "\n")
f.write("times = " + static.times + "\n")
f.write("--throttle = " + static.eventType["--throttle"] + "\n")
f.write("--pct-touch = " + static.eventPercent["--pct-touch"] + "\n")
f.write("--pct-motion = " + static.eventPercent["--pct-motion"] + "\n")
def setup_env(self):
pass
def _set_eventType(self, confFile):
try:
option = self.etSet.item(0, 0).text()
value = self.etSet.item(0, 1).text()
if value > "0":
with open(confFile, 'a+') as f:
f.write(option + " = " + value + "\n")
except AttributeError, e:
pass
class TableWidgetController(AbstractViewController, Ui_TableWidget):
'''
The controller part for a data table widget. The UI part is declared
in tablewidget_v1.ui and then converted to ui_tablewidget.py.
This widget can be fed tabular data and has several options for showing it.
@since: 2010-11-10
'''
__author__ = "Moritz Wade"
__contact__ = "*****@*****.**"
__copyright__ = "Zuse Institute Berlin 2010"
def __init__(self, parent=None, host=None, title="Table", mode=None):
'''
Constructor
'''
super(TableWidgetController, self).__init__(parent)
self.setupUi(self)
self.setWindowTitle(title)
self._initialize()
# self.options = {
# OPTION_SORTABLE_COLUMNS: self.checkBoxSortableColumns.isChecked(),
# }
self._mode = MODE_DEFAULT
self.maxValue = -1
self.host = host
self.data = None
self.dataTableHeaders = []
self.dataTableColumnData = []
self.dataTableRowCount = -1
self.dataTableRowHeaders = None
self.dataTableWidget = None
self.isColored = False
self.checkBoxShowUnits.setChecked(DEFAULT_SHOW_UNITS)
self.showUnits = DEFAULT_SHOW_UNITS
self.orientation = ORIENTATION_HORIZONTAL
self.sortColumn = -1 # default: do not sort at load
self.colorThreshold = None
self.colorThresholdBase = self.doubleSpinBox_Coloring_Threshold.value()
self.colorThresholdExponent = self.spinBox_Coloring_Exponent.value()
if mode:
self.setMode(mode)
self._updateThreshold()
def setMode(self, mode):
self._mode = mode
if mode == MODE_SUBCONDITIONS:
rtolScientificString = '{:e}'.format(float(self.host.getRTol()))
exponentStr = rtolScientificString.split("e")[1]
self.spinBox_Coloring_Exponent.setValue(float(exponentStr))
self.doubleSpinBox_Coloring_Threshold.setValue(1.0)
self.label_Coloring_Threshold.setText(
"Anticipated Relative Measurement Error")
self.groupBox_Coloring.setChecked(True)
self.isColored = True
elif mode == MODE_JACOBIAN:
self.groupBox_Coloring.setChecked(
True) # just activate coloring with default threshold
self.isColored = True
def _updateView(self, data=None):
'''
Overriding the "abstract" base class method. This does the main "drawing" of data, i.e.
generates the table and fills it with data.
'''
if data:
self.data = data
if self.data:
self._updateDataTable(self.data)
def _clearView(self):
if self.dataTableWidget:
self.dataTableWidget.clear()
def _setRowHeaders(self, dataDescriptors):
'''
Sets the given datadescriptors (from an EntityData object) as row headers
of the table, thereby checking for floats and rounding them, as not to show too
many digits after the point.
'''
#self.dataTableRowHeaders = dataDescriptors
if not dataDescriptors:
logging.debug(
"TableWidgetController._setRowHeaders(): Empty dataDescriptor list."
)
return
self.dataTableRowHeaders = []
for descriptor in dataDescriptors:
try:
descriptor = float(descriptor)
descriptor = round(descriptor, 4)
except ValueError:
pass
self.dataTableRowHeaders.append(
str(descriptor)) # the QTableWidget needs a list of Strings
def _computeColor(self, value):
if type(value) == str:
if value == "N/A":
return COLOR_LOW
try:
value = float(value)
except:
return None
if self._mode == MODE_DEFAULT or self._mode == MODE_JACOBIAN:
if value <= self.colorThreshold:
color = COLOR_LOW
else:
color = COLOR_HIGH
elif self._mode == MODE_SUBCONDITIONS:
if value >= self.colorThreshold:
color = COLOR_LOW
else:
#percentage = value / float(self.maxValue)
percentage = (self.maxValue - value + 1) / float(
self.maxValue
) # +1 because it's the "lowest" subconditon, and represents 100%
highRed, highGreen, highBlue, highAlpha = COLOR_HIGH.red(
), COLOR_HIGH.green(), COLOR_HIGH.blue(), COLOR_HIGH.alpha()
mediumRed, mediumGreen, mediumBlue, mediumAlpha = COLOR_MEDIUM.red(
), COLOR_MEDIUM.green(), COLOR_MEDIUM.blue(
), COLOR_MEDIUM.alpha()
diffRed, diffGreen, diffBlue, diffAlpha = highRed - mediumRed, highGreen - mediumGreen, highBlue - mediumBlue, highAlpha - mediumAlpha
valueRed = diffRed * percentage + mediumRed
valueGreen = diffGreen * percentage + mediumGreen
valueBlue = diffBlue * percentage + mediumBlue
valueAlpha = diffAlpha * percentage + mediumAlpha
color = QColor(valueRed, valueGreen, valueBlue, valueAlpha)
else:
color = QColor(0, 0, 0, 255) #transparent
return color
def _updateDataTable(self, data):
'''
Updates the data table with data from the last integration.
'''
#prepare data
self.dataTableHeaders = []
self.dataTableColumnData = []
self.dataTableRowCount = -1
self.dataTableRowHeaders = None
self.maxValue = -1
for (entity, entityDataList) in data.items():
for entityData in entityDataList:
dataDescriptors = entityData.dataDescriptors
if not self.dataTableRowHeaders:
self._setRowHeaders(dataDescriptors)
elif len(self.dataTableRowHeaders) != len(dataDescriptors):
logging.debug(
"Different number of time points for two Species. Last Species (%s) will be skipped."
% entity)
continue
# set header for first column (dataDescriptor/Timepoint col)
# in first iteration
if len(self.dataTableHeaders) == 0:
dataDescriptorName = entityData.dataDescriptorName
dataDescriptorUnit = entityData.dataDescriptorUnit
if not dataDescriptorUnit and "timepoint" in str(
dataDescriptorName).lower():
dataDescriptorUnit = self.host.optionTimeUnit
firstColHeader = ""
if dataDescriptorName:
if self.showUnits:
firstColHeader = "%s [%s]" % (dataDescriptorName,
dataDescriptorUnit)
elif dataDescriptorName:
firstColHeader = "%s" % dataDescriptorName
self.dataTableHeaders.append(firstColHeader)
#self.dataTableHeaders.append("Time species %s [%s]" % (str(speciesID), self.lineEditTimeUnit.text()))
#self.dataTableColumnData.append(timepoints)
if len(dataDescriptors) > self.dataTableRowCount:
self.dataTableRowCount = len(dataDescriptors)
#datapoints = entityData.datapoints
datapoints = []
# shorten datapoints
for i, datapoint in enumerate(entityData.datapoints):
try:
#datapoints.append(round(float(datapoint), 4))
# valueString = "%g" % (float(datapoint))
floatValue = float(
datapoint) # will jump to except if no float
valueString = "N/A" if math.isnan(
floatValue) else ' {0:-.4f}'.format(floatValue)
datapoints.append(valueString)
# preparing color computation
# logging.debug(entityData.dataDescriptorName)
# logging.debug(entityData.dataDescriptors[i] == settingsandvalues.SUBCONDITION_HEADER_ABSOLUTE)
if self._mode == MODE_SUBCONDITIONS\
and entityData.dataDescriptors[i] == settingsandvalues.SUBCONDITION_HEADER_ABSOLUTE\
and floatValue > self.maxValue\
and floatValue < self.colorThreshold:
self.maxValue = floatValue
except:
# datapoints.append(round(float("nan"), 4))
datapoints.append(str(datapoint))
# logging.debug("TableWidgetController - datapoints: %s" % datapoints) # too much overhead
#self.dataTableHeaders.append("Data species %s [%s]" % (str(speciesID), entityData.getUnit()))
if self.showUnits:
# if type(entity) == str:
# self.dataTableHeaders.append("%s" % entity)
# else:
self.dataTableHeaders.append(
"%s [%s]" %
(entity.getCombinedId(), entityData.getUnit()))
else:
self.dataTableHeaders.append("%s" % entity.getCombinedId())
self.dataTableColumnData.append(datapoints)
# if len(datapoints) > self.dataTableRowCount:
# self.dataTableRowCount = len(datapoints)
# Put those labels into the actual data that would be the vertical/row labels.
# We can't use .setVerticalHeaderLabers() because those labels don't get sorted together with the data.
# Very weird but that seems to be the intended behaviour of Qt.
if self.orientation == ORIENTATION_VERTICAL:
#self.dataTableColumnData.insert(0, self.dataTableHeaders) # handle as if it were data so that sorting works
self.dataTableHeaders = self.dataTableHeaders[
1:] # remove unnecessary dataDescriptor name
for i in xrange(len(self.dataTableColumnData)):
entry = self.dataTableColumnData[i]
entry.insert(0, self.dataTableHeaders[i])
self.dataTableRowHeaders.insert(0, "")
else:
self.dataTableColumnData.insert(0, self.dataTableRowHeaders)
#self.dataTableHeaders.insert(0,"")
if not self.dataTableWidget: # create for the first time
tableLayout = QVBoxLayout(self.tableWrapper)
self.dataTableWidget = QTableWidget(self)
tableLayout.addWidget(self.dataTableWidget)
#prepare table
self.dataTableWidget.setSortingEnabled(
True
) # do here to avoid performance penalty (this actually does one sorting run)
if self.orientation == ORIENTATION_HORIZONTAL:
self.dataTableWidget.setColumnCount(len(self.dataTableHeaders))
self.dataTableWidget.setRowCount(self.dataTableRowCount)
self.dataTableWidget.setHorizontalHeaderLabels(
self.dataTableHeaders)
# self.dataTableWidget.setVerticalHeaderLabels(
# self.dataTableRowHeaders) # has to be called after setRowCount?
elif self.orientation == ORIENTATION_VERTICAL:
self.dataTableWidget.setRowCount(len(self.dataTableHeaders))
self.dataTableWidget.setColumnCount(len(self.dataTableRowHeaders))
# self.dataTableWidget.setVerticalHeaderLabels(self.dataTableHeaders)
self.dataTableWidget.setHorizontalHeaderLabels(
self.dataTableRowHeaders
) # has to be called after setRowCount?
#put data into table
for col in xrange(len(self.dataTableColumnData)):
for row in xrange(len(self.dataTableColumnData[col])):
try:
value = self.dataTableColumnData[col][
row] # don't touch "values"; they could be pre-formatted strings
newItem = SortedTableWidgetItem() # use custom item class
newItem.setData(Qt.DisplayRole, value)
newItem.setTextAlignment(Qt.AlignRight)
newItem.setFont(QFont("Fixed"))
if self.isColored:
if not (self._mode == MODE_SUBCONDITIONS and row != 2
): #color only row 2 of subcondition tables
color = self._computeColor(value)
if color:
newItem.setBackground(QBrush(color))
except Exception, e:
logging.debug(
"TableWidgetController._updateDataTable(): Could not put value into widget item: %s\nError: %s"
% (value, e))
# newItem = SortedTableWidgetItem(str(self.dataTableColumnData[col][row]))
# newItem.setTextAlignment(Qt.AlignRight)
# newItem.setFont(QFont("Fixed"))
if self.orientation == ORIENTATION_HORIZONTAL:
self.dataTableWidget.setItem(row, col, newItem)
elif self.orientation == ORIENTATION_VERTICAL:
self.dataTableWidget.setItem(col, row, newItem)
# self.dataTableWidget.setSortingEnabled(True)
if self.sortColumn != -1:
self.dataTableWidget.sortItems(self.sortColumn)
self.dataTableWidget.resizeColumnsToContents()
class PnLPanel(QtGui.QWidget):
pnLColumnList = "Custody Name;Initial Position;Final Position;Cash In;Weighted Cash In; Cash Out;Weighted Cash Out;PnL;Weighted PnL;TIR;Weighted TIR".split(
";")
def __init__(self):
super(self.__class__, self).__init__()
self.layout = QtGui.QGridLayout(self)
self.pnlFilter = PnLFilter(self)
self.layout.addWidget(self.pnlFilter, 1, 0, QtCore.Qt.AlignTop)
#self.layout.setAlignment(self.pnlFilter, QtCore.Qt.AlignTop)
#self.layout.setAlignment(self.pnlFilter, QtCore.Qt.AlignLeft)
self.layout.addWidget(self.createPnLTable(), 1, 1, QtCore.Qt.AlignTop)
#self.layout.setAlignment(self.pnLTableWidget, QtCore.Qt.AlignTop)
#self.layout.setAlignment(self.pnLTableWidget, QtCore.Qt.AlignLeft)
def createPnLTable(self):
self.pnLTableWidget = QTableWidget()
self.pnLTableWidget.setRowCount(6)
self.pnLTableWidget.setColumnCount(len(self.pnLColumnList))
self.pnLTableWidget.setEditTriggers(
QtGui.QAbstractItemView.NoEditTriggers)
self.pnLTableWidget.setHorizontalHeaderLabels(self.pnLColumnList)
#self.pnLTableWidget.resizeColumnsToContents()
self.pnLTableWidget.resizeRowsToContents()
self.pnLTableWidget.setFixedSize(1100, 150)
return self.pnLTableWidget
def doSubmit(self, fromDate, toDate):
pnlLO = Engine.buildPnlLogicObject(fromDate, toDate)
self.renderPnlTable(pnlLO.pnlVOList)
def renderPnlTable(self, pnlCalculationList):
row = 0
for pnlVO in pnlCalculationList:
#ItemNameItem
ItemNameItem = QTableWidgetItemString(pnlVO.itemName, False)
self.pnLTableWidget.setItem(row,
Constant.CONST_COLUMN_PNL_ITEM_NAME,
ItemNameItem)
#initialPositionItem
initialPositionItem = QTableWidgetItemDecimal(
pnlVO.initialPosition, False)
self.pnLTableWidget.setItem(
row, Constant.CONST_COLUMN_PNL_INITIAL_POSITION,
initialPositionItem)
#finalPositionItem
finalPositionItem = QTableWidgetItemDecimal(
pnlVO.finalPosition, False)
self.pnLTableWidget.setItem(
row, Constant.CONST_COLUMN_PNL_FINAL_POSITION,
finalPositionItem)
#totalCashIn
totalCashInItem = QTableWidgetItemDecimal(pnlVO.totalCashIn, False)
self.pnLTableWidget.setItem(row, Constant.CONST_COLUMN_PNL_CASH_IN,
totalCashInItem)
#totalWeightedCashIn
totalWeightedCashInItem = QTableWidgetItemDecimal(
pnlVO.totalWeightedCashIn, False)
self.pnLTableWidget.setItem(
row, Constant.CONST_COLUMN_PNL_WEIGHTED_CASH_IN,
totalWeightedCashInItem)
#totalCashOut
totalCashOutItem = QTableWidgetItemDecimal(pnlVO.totalCashOut,
False)
self.pnLTableWidget.setItem(row,
Constant.CONST_COLUMN_PNL_CASH_OUT,
totalCashOutItem)
#totalWeightedCashOut
totalWeightedCashOutItem = QTableWidgetItemDecimal(
pnlVO.totalWeightedCashOut, False)
self.pnLTableWidget.setItem(
row, Constant.CONST_COLUMN_PNL_WEIGHTED_CASH_OUT,
totalWeightedCashOutItem)
#pnlAmount
pnlAmountItem = QTableWidgetItemDecimal(pnlVO.pnlAmount, False)
self.pnLTableWidget.setItem(row,
Constant.CONST_COLUMN_PNL_PNL_AMOUNT,
pnlAmountItem)
#pnlWeightedAmount
pnlWeightedAmountItem = QTableWidgetItemDecimal(
pnlVO.pnlWeightedAmount, False)
self.pnLTableWidget.setItem(
row, Constant.CONST_COLUMN_PNL_WEIGHTED_PNL_AMOUNT,
pnlWeightedAmountItem)
#tir
tirItem = QTableWidgetItemDecimal(pnlVO.tir, False)
self.pnLTableWidget.setItem(row, Constant.CONST_COLUMN_PNL_TIR,
tirItem)
#weightedtir
weightedTirItem = QTableWidgetItemDecimal(pnlVO.weightedTir, False)
self.pnLTableWidget.setItem(row,
Constant.CONST_COLUMN_PNL_WEIGHTED_TIR,
weightedTirItem)
row += 1
