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 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 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
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 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)
class CustomWidget(QtGui.QMainWindow):
def __init__(self):
"""
Constructor
"""
QtGui.QMainWindow.__init__(self)
self.name = "Custom widget"
self.central_widget = QtGui.QWidget()
self.setCentralWidget(self.central_widget)
# TODO: This is ugly, improve it
self.iconp = JConfig().icons_path
self._createLayout()
def _createGui(self):
"""
Subclasses must override this
depending on the elements they want to add
self._createOutputTree(),
self._createOutputTable(),
self._createOutputWindow() and add them to
the corresponding layouts.
"""
raise NotImplementedError
def _createToolBar(self, name):
"""
Subclasses need to define the
specific Actions
"""
self.toolbar = self.addToolBar(name)
self.toolbar.setMovable(False)
def _createLayout(self):
"""
This creates the basic layout:
Buttons & Outputs
"""
# Layouts (This is a common disposition)
main_layout = QtGui.QVBoxLayout()
self.button_layout = QtGui.QHBoxLayout()
output_layout = QtGui.QVBoxLayout()
# You will need to create your buttons
# and add them to your layout like this:
# self.button_layout.addWidget(button_1)
# Output Layout Inner (QSplitter)
# Add as many widgets as you please
# They will be ordered vertically and
# be resizable by the user
# self.splitter.addWidget(self.table_label)
# self.splitter.addWidget(...)
self.splitter = QSplitter(QtCore.Qt.Vertical)
# Nested layouts
main_layout.addLayout(self.button_layout)
output_layout.addWidget(self.splitter)
main_layout.addLayout(output_layout)
self.central_widget.setLayout(main_layout)
def _createOutputWindow(self):
"""
Some binary analysis commands will output to this.
"""
self.output_label = QtGui.QLabel('Output')
self.output_window = QTextEdit()
self.output_window.setFontPointSize(10)
self.output_window.setReadOnly(True)
# Save it for later use
self.output_window.original_textcolor = self.output_window.textColor()
def _createOutputTable(self):
"""
A vanilla QTableWidget. Callbacks modify
its properties, like number of columns, etc.
"""
self.table_label = QtGui.QLabel('Table Output')
self.table = QTableWidget()
self.table.setColumnCount(3)
self.table.setColumnWidth(0, 100)
self.table.setColumnWidth(1, 300)
self.table.setColumnWidth(2, 300)
self.table.setContextMenuPolicy(QtCore.Qt.CustomContextMenu)
# Connect signals to slots
self.table.customContextMenuRequested.connect(self._tablePopup)
self.table.horizontalHeader().sectionDoubleClicked.connect(
self._tableHeaderDoubleClicked)
self.table.cellDoubleClicked.connect(self._tableCellDoubleClicked)
def _createOutputTree(self):
"""
A QtreeWidget. Initially used to display those pesky
dword comparison to immediate values.
"""
self.tree_label = QtGui.QLabel('Tree Output')
self.tree = QTreeWidget()
self.tree.setColumnCount(3)
self.tree.setColumnWidth(0, 150)
self.tree.setColumnWidth(1, 150)
self.tree.setColumnWidth(2, 50)
# Connect signals to slots
self.tree.itemDoubleClicked.connect(self._treeElementDoubleClicked)
#################################################################
# GUI Callbacks
#################################################################
def _console_output(self, s="", err=False):
"""
Convenience wrapper
"""
if err:
# Error message
err_color = QColor('red')
self.output_window.setTextColor(err_color)
self.output_window.append(s)
# restore original color
self.output_window.setTextColor(
self.output_window.original_textcolor)
else:
self.output_window.append(s)
def _tableCellDoubleClicked(self, row, col):
"""
Most of the info displayed in QTableWidgets represent addresses.
Default action: try to jump to it.
"""
it = self.table.item(row, col).text()
try:
addr = int(it, 16)
jump_to_address(addr)
except ValueError:
self._console_output("[!] That does not look like an address...",
err=True)
return
def _tablePopup(self, pos):
"""
Popup menu activated clicking the secondary
button on the table
"""
menu = QtGui.QMenu()
# Add menu entries
delRow = menu.addAction(QIcon(self.iconp + "close.png"), "Delete Row")
selItem = menu.addAction(QIcon(self.iconp + "bookmark.png"),
"Mark entry")
menu.addSeparator()
origFunc = menu.addAction(QIcon(self.iconp + "lightning.png"),
"Select origin function")
destFunc = menu.addAction(QIcon(self.iconp + "flag.png"),
"Select destination function")
# Get entry clicked
action = menu.exec_(self.mapToGlobal(pos))
# Dispatch :)
if action == delRow:
self.table.removeRow(self.table.currentRow())
elif action == selItem:
self.table.currentItem().setBackground(QtGui.QColor('red'))
elif action == origFunc:
try:
addr = self.table.currentItem().text()
InfoUI.function_orig_ea = int(addr, 16)
except:
self._console_output("This does not look like an address...",
err=True)
elif action == destFunc:
try:
addr = self.table.currentItem().text()
InfoUI.function_dest_ea = int(addr, 16)
except:
self._console_output("This does not look like an address...",
err=True)
def _tableHeaderDoubleClicked(self, index):
"""
Used to sort the contents
"""
self.table.sortItems(index, order=QtCore.Qt.AscendingOrder)
def _treeElementDoubleClicked(self, item, column):
"""
QTreeWidgetElement callback. Basically it jumps to
the selected address in IDA disassembly window.
"""
try:
# Only interested in addresses
addr_int = int(item.text(column), 16)
jump_to_address(addr_int)
# Paint some color
item.setBackground(column, QtGui.QColor('green'))
except:
self._console_output("[!] That does not look like an address...",
err=True)
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 Ui_MainWindow(object):
def setupUi(self, MainWindow):
MainWindow.setObjectName("MainWindow")
MainWindow.setFixedSize(800, 600)
self.centralwidget = QWidget(MainWindow)
self.centralwidget.setObjectName("centralwidget")
self.FilterLbl = QLabel(self.centralwidget)
self.FilterLbl.setGeometry(QtCore.QRect(30, 150, 60, 15))
self.FilterLbl.setObjectName("FilterLbl")
self.FilterCB = QComboBox(self.centralwidget)
self.FilterCB.setGeometry(QtCore.QRect(450, 150, 100, 22))
self.FilterCB.setObjectName("FilterCB")
self.FilterCB.addItem("")
self.FilterCB.addItem("")
self.FilterCB.addItem("")
self.FilterCB.addItem("")
self.FilterTF = QLineEdit(self.centralwidget)
self.FilterTF.setGeometry(QtCore.QRect(100, 150, 320, 20))
self.tableView = QTableWidget(self.centralwidget)
self.tableView.setGeometry(QtCore.QRect(10, 180, 781, 511))
self.tableView.setObjectName("tableView")
self.tableView.setColumnCount(4)
self.tableView.setRowCount(0)
item = QTableWidgetItem("Cena za kg/l")
self.tableView.setHorizontalHeaderItem(0, item)
item = QTableWidgetItem("Cena ze kus")
self.tableView.setHorizontalHeaderItem(1, item)
item = QTableWidgetItem(u"Gramaž")
self.tableView.setHorizontalHeaderItem(2, item)
item = QTableWidgetItem("Popis")
item.setTextAlignment(QtCore.Qt.AlignHCenter|QtCore.Qt.AlignVCenter|QtCore.Qt.AlignCenter)
font = QtGui.QFont()
font.setPointSize(8)
item.setFont(font)
self.tableView.setHorizontalHeaderItem(3, item)
self.tableView.horizontalHeader().setStretchLastSection(True)
self.SaveBtn = QPushButton(self.centralwidget)
self.SaveBtn.setGeometry(QtCore.QRect(30, 10, 100, 23))
self.SaveBtn.setObjectName("SaveBtn")
self.PrintSelectedToFileBtn = QPushButton(self.centralwidget)
self.PrintSelectedToFileBtn.setGeometry(QtCore.QRect(225, 10, 100, 23))
self.PrintSelectedToFileBtn.setObjectName("PrintSelectedToFileBtn")
self.PriceForUnitTF = QLineEdit(self.centralwidget)
self.PriceForUnitTF.setGeometry(QtCore.QRect(100, 70, 113, 20))
self.PriceForUnitTF.setObjectName("PriceForUnitTF")
self.PriceForUnitLbl = QLabel(self.centralwidget)
self.PriceForUnitLbl.setGeometry(QtCore.QRect(30, 70, 60, 13))
self.PriceForUnitLbl.setObjectName("PriceForUnitLbl")
self.ArtikelTF = QLineEdit(self.centralwidget)
self.ArtikelTF.setGeometry(QtCore.QRect(100, 100, 113, 20))
self.ArtikelTF.setObjectName("ArtikelTF")
self.ArtikelLbl = QLabel(self.centralwidget)
self.ArtikelLbl.setGeometry(QtCore.QRect(30, 100, 46, 13))
self.ArtikelLbl.setObjectName("ArtikelLbl")
self.DescriptionLbl = QLabel(self.centralwidget)
self.DescriptionLbl.setGeometry(QtCore.QRect(455, 70, 75, 13))
self.DescriptionLbl.setObjectName("DescriptionLbl")
self.UnitLbl = QLabel(self.centralwidget)
self.UnitLbl.setGeometry(QtCore.QRect(250, 70, 60, 15))
self.UnitLbl.setObjectName("UnitLbl")
self.WeightLbl = QLabel(self.centralwidget)
self.WeightLbl.setGeometry(QtCore.QRect(250, 100, 60, 13))
self.WeightLbl.setObjectName("UnitLbl")
self.WeightTF = QLineEdit(self.centralwidget)
self.WeightTF.setGeometry(QtCore.QRect(320, 100, 100, 20))
self.WeightTF.setObjectName("WeightTF")
self.UnitCB = QComboBox(self.centralwidget)
self.UnitCB.setGeometry(QtCore.QRect(320, 70, 100, 22))
self.UnitCB.setObjectName("UnitCB")
self.UnitCB.addItem("")
self.UnitCB.addItem("")
self.DescriptionTE = QPlainTextEdit(self.centralwidget)
self.DescriptionTE.setGeometry(QtCore.QRect(540, 30, 241, 61))
self.DescriptionTE.setObjectName("DescriptionTE")
self.PrintToFileBtn = QPushButton(self.centralwidget)
self.PrintToFileBtn.setGeometry(QtCore.QRect(140, 10, 75, 23))
self.PrintToFileBtn.setObjectName("PrintToFileBtn")
self.AddRecordBtn = QPushButton(self.centralwidget)
self.AddRecordBtn.setGeometry(QtCore.QRect(450, 100, 75, 23))
self.AddRecordBtn.setObjectName("AddRecordBtn")
self.SaveChangeBtn = QPushButton(self.centralwidget)
self.SaveChangeBtn.setGeometry(QtCore.QRect(550, 100, 75, 23))
self.SaveChangeBtn.setObjectName("SaveChangeBtn")
self.DeleteRecordBtn = QPushButton(self.centralwidget)
self.DeleteRecordBtn.setGeometry(QtCore.QRect(650, 100, 75, 23))
self.DeleteRecordBtn.setObjectName("DeleteRecordBtn")
MainWindow.setCentralWidget(self.centralwidget)
self.menubar = QMenuBar(MainWindow)
self.menubar.setGeometry(QtCore.QRect(0, 0, 800, 21))
self.menubar.setObjectName("menubar")
MainWindow.setMenuBar(self.menubar)
self.statusbar = QStatusBar(MainWindow)
self.statusbar.setObjectName("statusbar")
self.FilterTF.textChanged.connect(self.on_lineEdit_textChanged)
self.FilterCB.currentIndexChanged.connect(self.on_comboBox_currentIndexChanged)
self.retranslateUi(MainWindow)
QtCore.QMetaObject.connectSlotsByName(MainWindow)
def myFilter(self,col=None):
filt = self.FilterTF.text()
for ix in range(self.tableView.rowCount()):
match = False
if col == None:
for jx in range(self.tableView.columnCount()):
item = self.tableView.item(ix,jx)
if filt in item.text():
match = True
break
self.tableView.setRowHidden(ix, not match)
else:
item = self.tableView.item(ix, col)
if filt in item.text():
match = True
self.tableView.setRowHidden(ix, not match)
#@QtCore.pyqtSlot(str)
def on_lineEdit_textChanged(self, text):
self.myFilter()
#@QtCore.pyqtSlot(int)
def on_comboBox_currentIndexChanged(self, index):
self.myFilter(col=index)
def retranslateUi(self, MainWindow):
_translate = QtCore.QCoreApplication.translate
MainWindow.setWindowTitle(_translate("MainWindow", "Stitky - {0}".format(__version__)))
self.SaveBtn.setText(_translate("MainWindow", "Uloz stav tabulky"))
self.PrintSelectedToFileBtn.setText(_translate("MainWindow", "Tisk vybranych"))
self.PriceForUnitLbl.setText(_translate("MainWindow", "Cena za kus:"))
self.ArtikelLbl.setText(_translate("MainWindow", "Artikl:"))
self.DescriptionLbl.setText(_translate("MainWindow", "Popis produktu:"))
self.UnitLbl.setText(_translate("MainWindow", "Jednotka:"))
self.FilterLbl.setText(_translate("MainWindow", "Filtr:"))
self.WeightLbl.setText(_translate("MainWindow", "Hmotnost:"))
self.PrintToFileBtn.setText(_translate("MainWindow", "Vytvor txt"))
self.SaveChangeBtn.setText(_translate("MainWindow", "Uloz zmeny"))
self.AddRecordBtn.setText(_translate("MainWindow", "Pridej zaznam"))
self.DeleteRecordBtn.setText(_translate("MainWindow", "Smaz zaznam"))
self.UnitCB.setItemText(0, _translate("MainWindow", "g"))
self.UnitCB.setItemText(1, _translate("MainWindow", "ml"))
self.FilterCB.setItemText(0, _translate("MainWindow", "Cena za kg/l"))
self.FilterCB.setItemText(1, _translate("MainWindow", "Cena ze kus"))
self.FilterCB.setItemText(2, _translate("MainWindow", "Gramaz"))
self.FilterCB.setItemText(3, _translate("MainWindow", "Popis"))
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 MenuWeekday():
"""Tab to manage menu entry for each day"""
global logger
def __init__(self, day):
####
logger.info('Inside MenuWeekday')
self.menudetail_tab_1 = QWidget()
self.menudetail_tab_1.setObjectName("menudetail_tab_1")
self.gridLayout_20 = QGridLayout(self.menudetail_tab_1)
self.gridLayout_20.setObjectName("gridLayout_20")
self.menu_table = QTableWidget(self.menudetail_tab_1)
self.menu_table.setSortingEnabled(True)
self.menu_table.setObjectName("menu_table")
self.menu_table.setColumnCount(6)
self.menu_table.setRowCount(0)
item = QTableWidgetItem()
self.menu_table.setHorizontalHeaderItem(0, item)
item = QTableWidgetItem()
self.menu_table.setHorizontalHeaderItem(1, item)
item = QTableWidgetItem()
self.menu_table.setHorizontalHeaderItem(2, item)
item = QTableWidgetItem()
self.menu_table.setHorizontalHeaderItem(3, item)
item = QTableWidgetItem()
self.menu_table.setHorizontalHeaderItem(4, item)
item = QTableWidgetItem()
self.menu_table.setHorizontalHeaderItem(5, item)
self.menu_table.horizontalHeader().setCascadingSectionResizes(False)
self.menu_table.horizontalHeader().setStretchLastSection(True)
self.menu_table.verticalHeader().setVisible(True)
self.menu_table.verticalHeader().setCascadingSectionResizes(True)
self.gridLayout_20.addWidget(self.menu_table, 0, 0, 1, 3)
spacerItem22 = QSpacerItem(612, 20, QSizePolicy.Expanding, QSizePolicy.Minimum)
self.gridLayout_20.addItem(spacerItem22, 1, 0, 1, 1)
self.menu_table_refresh_button = QPushButton(self.menudetail_tab_1)
self.menu_table_refresh_button.setObjectName("menu_table_refresh_button")
self.gridLayout_20.addWidget(self.menu_table_refresh_button, 1, 1, 1, 1)
self.menu_table_add_button = QPushButton(self.menudetail_tab_1)
self.menu_table_add_button.setObjectName("menu_table_add_button")
self.gridLayout_20.addWidget(self.menu_table_add_button, 1, 2, 1, 1)
####retranslate
self.menu_table.horizontalHeaderItem(0).setText(
QApplication.translate("MainWindow", "Code", None, QApplication.UnicodeUTF8))
self.menu_table.horizontalHeaderItem(1).setText(
QApplication.translate("MainWindow", "Item", None, QApplication.UnicodeUTF8))
self.menu_table.horizontalHeaderItem(2).setText(
QApplication.translate("MainWindow", "Category", None, QApplication.UnicodeUTF8))
self.menu_table.horizontalHeaderItem(3).setText(
QApplication.translate("MainWindow", "Rate", None, QApplication.UnicodeUTF8))
self.menu_table.horizontalHeaderItem(4).setText(
QApplication.translate("MainWindow", "Per Day", None, QApplication.UnicodeUTF8))
self.menu_table.horizontalHeaderItem(5).setText(
QApplication.translate("MainWindow", "Available", None, QApplication.UnicodeUTF8))
self.menu_table_refresh_button.setText(
QApplication.translate("MainWindow", "Refresh", None, QApplication.UnicodeUTF8))
self.menu_table_add_button.setText(
QApplication.translate("MainWindow", "Add Dish", None, QApplication.UnicodeUTF8))
###signals and slots && other stuffs
# self.mainwindow = Ui_MainWindow # just for the ease of finding the attributes in pycharm
self.day = day
self.menu_table_add_button.clicked.connect(self.add_menu)
self.menu_table.itemDoubleClicked.connect(self.popup_edit)
self.menu_table_refresh_button.clicked.connect(self.update_menu)
self.menu_table.setSelectionBehavior(QAbstractItemView.SelectRows)
self.menu_table.setEditTriggers(QAbstractItemView.NoEditTriggers)
self.menu_table.setShowGrid(False)
self.menu_table.setAlternatingRowColors(True)
self.update_menu()
self.popup = object
self.menudetail_tab_1.setFocusPolicy(Qt.StrongFocus)
self.menudetail_tab_1.focusInEvent = self.load_rows
self.assign_shortcuts()
def assign_shortcuts(self):
"""assign shortcuts"""
QShortcut(QKeySequence(settings.custom_shortcut['key_tabcon_menumgtmonsun_view']),
self.menudetail_tab_1, self.row_selected)
QShortcut(QKeySequence(settings.custom_shortcut['key_tabcon_menumgtmonsun_refresh']),
self.menudetail_tab_1, self.update_menu)
QShortcut(QKeySequence(settings.custom_shortcut['key_tabcon_menumgtmonsun_add']),
self.menudetail_tab_1, self.add_menu)
def row_selected(self):
"""selets a row entry"""
rows = sorted(set(index.row() for index in
self.menu_table.selectedIndexes()))
if rows:
code = self.menu_table.item(rows[0], 0).text()
name = self.menu_table.item(rows[0], 1).text()
quantity = self.menu_table.item(rows[0], 4).text()
self.add_menu(code, name, quantity)
def add_menu(self, *args):
"""
:return: Pops up a new dialogue to add the menus
"""
if not args:
self.popup = AddMenuFromList(parent=self, day=self.day)
self.popup.exec_()
else:
self.popup = MenuQuantityEditPopup(parent=self, day=self.day, code=args[0], name=args[1], quantity=args[2])
self.popup.exec_()
def update_menu(self):
"""
:return:Populates the menu table
"""
menu = WeeklyMenu(day=self.day)
menulist = menu.load_dish_per_Day()
if not menulist == []:
self.add_row(*menulist)
else:
self.menu_table.clearContents()
self.menu_table.setRowCount(0)
def popup_edit(self, item):
"""
Pops up the menu to be edited
:param item: item clicked
:return:none
"""
model_index = self.menu_table.indexFromItem(item)
row = model_index.row()
code = self.menu_table.item(row, 0).text()
name = self.menu_table.item(row, 1).text()
quantity = self.menu_table.item(row, 4).text()
self.add_menu(code, name, quantity)
def add_row(self, *args):
"""creates a new row entry"""
table = self.menu_table
if args:
table.setRowCount(len(args))
for i, j in enumerate(args):
item = QTableWidgetItem(j['item_no'])
table.setItem(i, 0, item)
item = QTableWidgetItem(j['item'])
table.setItem(i, 1, item)
item = QTableWidgetItem(j['category'])
table.setItem(i, 2, item)
item = QTableWidgetItem(j['rate'])
table.setItem(i, 3, item)
item = QTableWidgetItem(j['per_day'])
table.setItem(i, 4, item)
item = QTableWidgetItem(j['available'])
table.setItem(i, 5, item)
table.setColumnWidth(0, table.width() / 6)
table.setColumnWidth(1, table.width() / 6)
table.setColumnWidth(2, table.width() / 6)
table.setColumnWidth(3, table.width() / 6)
table.setColumnWidth(4, table.width() / 6)
def load_rows(self, event):
"""
:return:checks and adds new rows
"""
self.add_row()
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 CustomWidget(QtGui.QMainWindow):
def __init__(self):
"""
Constructor
"""
QtGui.QMainWindow.__init__(self)
self.name = "Custom widget"
self.central_widget = QtGui.QWidget()
self.setCentralWidget(self.central_widget)
# TODO: This is ugly, improve it
self.iconp = JConfig().icons_path
self._createLayout()
def _createGui(self):
"""
Subclasses must override this
depending on the elements they want to add
self._createOutputTree(),
self._createOutputTable(),
self._createOutputWindow() and add them to
the corresponding layouts.
"""
raise NotImplementedError
def _createToolBar(self, name):
"""
Subclasses need to define the
specific Actions
"""
self.toolbar = self.addToolBar(name)
self.toolbar.setMovable(False)
def _createLayout(self):
"""
This creates the basic layout:
Buttons & Outputs
"""
# Layouts (This is a common disposition)
main_layout = QtGui.QVBoxLayout()
self.button_layout = QtGui.QHBoxLayout()
output_layout = QtGui.QVBoxLayout()
# You will need to create your buttons
# and add them to your layout like this:
# self.button_layout.addWidget(button_1)
# Output Layout Inner (QSplitter)
# Add as many widgets as you please
# They will be ordered vertically and
# be resizable by the user
# self.splitter.addWidget(self.table_label)
# self.splitter.addWidget(...)
self.splitter = QSplitter(QtCore.Qt.Vertical)
# Nested layouts
main_layout.addLayout(self.button_layout)
output_layout.addWidget(self.splitter)
main_layout.addLayout(output_layout)
self.central_widget.setLayout(main_layout)
def _createOutputWindow(self):
"""
Some binary analysis commands will output to this.
"""
self.output_label = QtGui.QLabel('Output')
self.output_window = QTextEdit()
self.output_window.setFontPointSize(10)
self.output_window.setReadOnly(True)
# Save it for later use
self.output_window.original_textcolor = self.output_window.textColor()
def _createOutputTable(self):
"""
A vanilla QTableWidget. Callbacks modify
its properties, like number of columns, etc.
"""
self.table_label = QtGui.QLabel('Table Output')
self.table = QTableWidget()
self.table.setColumnCount(3)
self.table.setColumnWidth(0, 100)
self.table.setColumnWidth(1, 300)
self.table.setColumnWidth(2, 300)
self.table.setContextMenuPolicy(QtCore.Qt.CustomContextMenu)
# Connect signals to slots
self.table.customContextMenuRequested.connect(self._tablePopup)
self.table.horizontalHeader().sectionDoubleClicked.connect(self._tableHeaderDoubleClicked)
self.table.cellDoubleClicked.connect(self._tableCellDoubleClicked)
def _createOutputTree(self):
"""
A QtreeWidget. Initially used to display those pesky
dword comparison to immediate values.
"""
self.tree_label = QtGui.QLabel('Tree Output')
self.tree = QTreeWidget()
self.tree.setColumnCount(3)
self.tree.setColumnWidth(0, 150)
self.tree.setColumnWidth(1, 150)
self.tree.setColumnWidth(2, 50)
# Connect signals to slots
self.tree.itemDoubleClicked.connect(self._treeElementDoubleClicked)
#################################################################
# GUI Callbacks
#################################################################
def _console_output(self, s = "", err = False):
"""
Convenience wrapper
"""
if err:
# Error message
err_color = QColor('red')
self.output_window.setTextColor(err_color)
self.output_window.append(s)
# restore original color
self.output_window.setTextColor(self.output_window.original_textcolor)
else:
self.output_window.append(s)
def _tableCellDoubleClicked(self, row, col):
"""
Most of the info displayed in QTableWidgets represent addresses.
Default action: try to jump to it.
"""
it = self.table.item(row, col).text()
try:
addr = int(it, 16)
jump_to_address(addr)
except ValueError:
self._console_output("[!] That does not look like an address...", err = True)
return
def _tablePopup(self, pos):
"""
Popup menu activated clicking the secondary
button on the table
"""
menu = QtGui.QMenu()
# Add menu entries
delRow = menu.addAction(QIcon(self.iconp + "close.png"), "Delete Row")
selItem = menu.addAction(QIcon(self.iconp + "bookmark.png"), "Mark entry")
menu.addSeparator()
origFunc = menu.addAction(QIcon(self.iconp + "lightning.png"), "Select origin function")
destFunc = menu.addAction(QIcon(self.iconp + "flag.png"), "Select destination function")
# Get entry clicked
action = menu.exec_(self.mapToGlobal(pos))
# Dispatch :)
if action == delRow:
self.table.removeRow(self.table.currentRow())
elif action == selItem:
self.table.currentItem().setBackground(QtGui.QColor('red'))
elif action == origFunc:
try:
addr = self.table.currentItem().text()
InfoUI.function_orig_ea = int(addr, 16)
except:
self._console_output("This does not look like an address...", err = True)
elif action == destFunc:
try:
addr = self.table.currentItem().text()
InfoUI.function_dest_ea = int(addr, 16)
except:
self._console_output("This does not look like an address...", err = True)
def _tableHeaderDoubleClicked(self, index):
"""
Used to sort the contents
"""
self.table.sortItems(index, order = QtCore.Qt.AscendingOrder)
def _treeElementDoubleClicked(self, item, column):
"""
QTreeWidgetElement callback. Basically it jumps to
the selected address in IDA disassembly window.
"""
try:
# Only interested in addresses
addr_int = int(item.text(column), 16)
jump_to_address(addr_int)
# Paint some color
item.setBackground(column, QtGui.QColor('green'))
except:
self._console_output("[!] That does not look like an address...", err = True)
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)
