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()
def __init__( self ):
url = QLineEdit()
quality = QLineEdit()
urlLabel = QLabel( 'Url' )
qualityLabel = QLabel( 'Quality' )
messages = QTextEdit()
messagesLabel = QLabel( 'Messages' )
links = QTableWidget( 0, 2 )
linksLabel = QLabel( 'Links' )
clearMessages = QPushButton( 'Clear Messages' )
checkIfOnline = QPushButton( 'Check If Online' )
addSelectedLink = QPushButton( 'Add Link' )
removeSelectedLink = QPushButton( 'Remove Selected Link' )
messages.setReadOnly( True )
links.setHorizontalHeaderLabels( [ 'Url', 'Status' ] )
links.horizontalHeader().setResizeMode( QHeaderView.Stretch )
links.horizontalHeader().setResizeMode( 1, QHeaderView.Fixed )
# set the events
url.returnPressed.connect( self.select_stream_from_entry )
quality.returnPressed.connect( self.select_stream_from_entry )
links.itemDoubleClicked.connect( self.select_stream_from_link )
clearMessages.clicked.connect( self.clear_messages )
checkIfOnline.clicked.connect( self.check_if_online )
addSelectedLink.clicked.connect( self.add_selected_link )
removeSelectedLink.clicked.connect( self.remove_selected_link )
#set shortcut
checkIfOnline.setShortcut(QKeySequence(Qt.Key_F5))
# set the layouts
mainLayout = QGridLayout()
# first row
mainLayout.addWidget( urlLabel, 0, 0, 1, 1 ) # spans 1 column
mainLayout.addWidget( qualityLabel, 0, 1, 1, 1 )# spans 1 column
mainLayout.addWidget( linksLabel, 0, 2, 1, 3 ) # spans 3 columns
# second row (links widget occupies 2 rows and 2 columns)
mainLayout.addWidget( url, 1, 0, 1, 1 ) # spans 1 column
mainLayout.addWidget( quality, 1, 1, 1, 1 ) # spans 1 column
mainLayout.addWidget( links, 1, 2, 2, 3 ) # spans 3 columns
# third row (messages widget occupies 2 columns)
mainLayout.addWidget( messages, 2, 0, 1, 2 )
# fourth row
mainLayout.addWidget( messagesLabel, 3, 0 )
mainLayout.addWidget( clearMessages, 3, 1 )
mainLayout.addWidget( checkIfOnline, 3, 2 )
mainLayout.addWidget( addSelectedLink, 3, 3 )
mainLayout.addWidget( removeSelectedLink, 3, 4 )
window = QWidget()
window.setLayout( mainLayout )
window.setWindowTitle( 'Live Streamer' )
window.resize( 700, 350 )
window.show()
self.url_ui = url
self.quality_ui = quality
self.messages_ui = messages
self.links_ui = links
self.window_ui = window
self.links = set()
self.data_file = os.path.join(os.path.expanduser("~"), ".config", "livestreamer-ui", "data.txt")
folder=os.path.dirname(self.data_file)
if not os.path.exists(folder):
os.makedirs(folder)
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 CCParsedTab(QWidget):
def __init__(self, parser_result):
QWidget.__init__(self)
self._parser_result = parser_result
layout = QHBoxLayout()
layout.setContentsMargins(0, 0, 0, 0)
layout.setSizeConstraint(QHBoxLayout.SetMinimumSize)
self.table_widget = QTableWidget()
self.table_widget.setSelectionMode(QAbstractItemView.NoSelection)
self.table_widget.setRowCount(2)
self.table_widget.setColumnCount(2)
self.table_widget.horizontalHeader().setResizeMode(QHeaderView.Stretch)
self.table_widget.setHorizontalHeaderLabels(["Dimension", "Choice"])
self.table_widget.verticalHeader().setVisible(False)
self.table_widget.verticalHeader().setResizeMode(QHeaderView.Fixed)
self.table_widget.verticalHeader().setDefaultSectionSize(20)
sp_table = QSizePolicy(QSizePolicy.Preferred, QSizePolicy.Preferred)
sp_table.setHorizontalStretch(1)
self.table_widget.setSizePolicy(sp_table)
layout.addWidget(self.table_widget)
self.text_widget = QTextEdit()
sp_text = QSizePolicy(QSizePolicy.Preferred, QSizePolicy.Preferred)
sp_text.setHorizontalStretch(4)
self.text_widget.setReadOnly(True)
self.text_widget.setSizePolicy(sp_text)
self.text_widget.setStyleSheet('font: 9pt "Courier";')
self.text_widget.setText(self._parser_result.get_text())
layout.addWidget(self.text_widget)
self.setLayout(layout)
self._fill_table()
def _fill_table(self):
colors = ["yellow", "orange", "green", "red"]
dimensions = self._parser_result.get_dimensions()
row = 0
for dimension in dimensions:
dimension_name = QLabel(dimension[0])
dimension_name.setStyleSheet("background: %s" % colors[row % len(colors)])
self.table_widget.setCellWidget(row, 0, dimension_name)
choices_widget = QComboBox()
choices = dimension[1]
choices.insert(0, "No choice")
choices_widget.addItems(choices)
choices_widget.setStyleSheet("background: white")
choices_widget.setStyleSheet("border: none")
choices_widget.currentIndexChanged.connect(self.on_choice_change)
self.table_widget.setCellWidget(row, 1, choices_widget)
row += 1
@Slot()
def on_choice_change(self):
text = self._parser_result.get_text()
for row in range(self.table_widget.rowCount()):
label = self.table_widget.cellWidget(row, 0)
choices = self.table_widget.cellWidget(row, 1)
if choices.currentIndex() == 0:
continue
regex = self._build_regex(label.text(), choices.count() - 1)
pattern = re.compile(regex)
matches = pattern.search(text)
for match in matches.groups():
results = re.search("<(.+)>", match)
variants = [s.strip() for s in results.groups()[0].split(",")]
text = text.replace(match, variants[choices.currentIndex() - 1])
self.text_widget.setText(text)
@staticmethod
def _build_regex(dim, count):
regex = "(" + dim + "<"
for i in range(count):
regex += ".+,[ ]?"
regex = regex[:-5]
regex += ">)"
return regex
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)
def __init__(self):
url = QLineEdit()
urlLabel = QLabel('Url')
messages = QTextEdit()
messagesLabel = QLabel('Messages')
links = QTableWidget(0, 2)
linksLabel = QLabel('Links')
clearMessages = QPushButton('Clear Messages')
checkIfOnline = QPushButton('Check If Online')
addSelectedLink = QPushButton('Add Link')
removeSelectedLink = QPushButton('Remove Selected Link')
messages.setReadOnly(True)
links.setHorizontalHeaderLabels(['Url', 'Status'])
links.horizontalHeader().setResizeMode(QHeaderView.Stretch)
links.horizontalHeader().setResizeMode(1, QHeaderView.Fixed)
# set the events
url.returnPressed.connect(self.select_stream_from_entry)
links.itemDoubleClicked.connect(self.select_stream_from_link)
clearMessages.clicked.connect(self.clear_messages)
checkIfOnline.clicked.connect(self.check_if_online)
addSelectedLink.clicked.connect(self.add_selected_link)
removeSelectedLink.clicked.connect(self.remove_selected_link)
# set the layouts
mainLayout = QGridLayout()
# first row
mainLayout.addWidget(urlLabel, 0, 0, 1, 2) # spans 2 columns
mainLayout.addWidget(linksLabel, 0, 2, 1, 3) # spans 3 columns
# second row (links widget occupies 2 rows and 2 columns)
mainLayout.addWidget(url, 1, 0, 1, 2) # spans 2 columns
mainLayout.addWidget(links, 1, 2, 2, 3) # spans 3 columns
# third row (messages widget occupies 2 columns)
mainLayout.addWidget(messages, 2, 0, 1, 2)
# fourth row
mainLayout.addWidget(messagesLabel, 3, 0)
mainLayout.addWidget(clearMessages, 3, 1)
mainLayout.addWidget(checkIfOnline, 3, 2)
mainLayout.addWidget(addSelectedLink, 3, 3)
mainLayout.addWidget(removeSelectedLink, 3, 4)
window = QWidget()
window.setLayout(mainLayout)
window.setWindowTitle('Live Streamer')
window.resize(700, 350)
window.show()
self.url_ui = url
self.messages_ui = messages
self.links_ui = links
self.window_ui = window
self.links = set()
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 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 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 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 ReportEmployeeTestDialogue(QDialog):
"""
Employee Report Popup Manager
"""
global logger
def __init__(self, code=None, parent=None):
logger.info('Inside ReportEmployeeTestDialogue')
super(ReportEmployeeTestDialogue, self).__init__(parent)
self.resize(500, 500)
self.vertical_23 = QVBoxLayout(self)
self.vertical_23.setObjectName("vertical_23")
self.label_1 = QLabel(self)
self.vertical_23.addWidget(self.label_1)
self.report_health_table = QTableWidget(self)
self.report_health_table.setObjectName("report_health_table")
self.report_health_table.setColumnCount(5)
self.report_health_table.setRowCount(0)
self.report_health_table.setSelectionBehavior(
QAbstractItemView.SelectRows)
item = QTableWidgetItem()
self.report_health_table.setHorizontalHeaderItem(0, item)
item = QTableWidgetItem()
self.report_health_table.setHorizontalHeaderItem(1, item)
item = QTableWidgetItem()
self.report_health_table.setHorizontalHeaderItem(2, item)
item = QTableWidgetItem()
self.report_health_table.setHorizontalHeaderItem(3, item)
item = QTableWidgetItem()
self.report_health_table.setHorizontalHeaderItem(4, item)
self.report_health_table.horizontalHeader().setCascadingSectionResizes(
True)
self.report_health_table.horizontalHeader().setStretchLastSection(True)
self.report_health_table.verticalHeader().setCascadingSectionResizes(
True)
self.vertical_23.addWidget(self.report_health_table)
self.horizontal_21 = QHBoxLayout()
self.report_health_newrow_buttuon = QPushButton(self)
self.report_health_newrow_buttuon.setObjectName(
"report_health_newrow_buttuon")
self.horizontal_21.addWidget(self.report_health_newrow_buttuon)
spacerItem23 = QSpacerItem(40, 20, QSizePolicy.Expanding,
QSizePolicy.Minimum)
self.horizontal_21.addItem(spacerItem23)
self.vertical_23.addLayout(self.horizontal_21)
### retanslate
self.setWindowTitle(
QApplication.translate("MainWindow", "Health Report", None,
QApplication.UnicodeUTF8))
self.label_1.setText(
QApplication.translate("MainWindow", "Health Report", None,
QApplication.UnicodeUTF8))
self.report_health_table.horizontalHeaderItem(0).setText(
QApplication.translate("MainWindow", "Code", None,
QApplication.UnicodeUTF8))
self.report_health_table.horizontalHeaderItem(1).setText(
QApplication.translate("MainWindow", "Date", None,
QApplication.UnicodeUTF8))
self.report_health_table.horizontalHeaderItem(2).setText(
QApplication.translate("MainWindow", "Organization Name", None,
QApplication.UnicodeUTF8))
self.report_health_table.horizontalHeaderItem(3).setText(
QApplication.translate("MainWindow", "Test", None,
QApplication.UnicodeUTF8))
self.report_health_table.horizontalHeaderItem(4).setText(
QApplication.translate("MainWindow", "Description", None,
QApplication.UnicodeUTF8))
self.report_health_newrow_buttuon.setText(
QApplication.translate("MainWindow", "New Row", None,
QApplication.UnicodeUTF8))
###signals and slots && other stuffs
self.health = Health(emp_id=code)
self.report_health_table.setEditTriggers(
QAbstractItemView.NoEditTriggers)
self.load_table_rows()
self.report_health_table.itemDoubleClicked.connect(
self.popup_health_edit)
self.report_health_newrow_buttuon.clicked.connect(self.new_healthTest)
self.focusInEvent = self.load_rows
def new_healthTest(self, code=None):
"""
pops up a dialog to add a new report or edit the existing report
:param code:the code of the report
"""
try:
pop = HygieneReportPop(parent=self,
table='health_table',
code=code)
pop.setWindowFlags(Qt.WindowTitleHint)
pop.exec_()
self.load_table_rows()
except Exception:
if settings.level == 10:
logger.exception('raised exception')
return False, 'Some Internal Error'
def load_table_rows(self):
"""
loads the table of reports
"""
try:
data = self.health.load_report()
if data[0]:
self.add_table_rows(*data[1])
if not data[1]:
self.report_health_table.clearContents()
self.report_health_table.setRowCount(0)
except Exception:
if settings.level == 10:
logger.exception('raised exception')
return False, 'Some Internal Error'
def popup_health_edit(self, item):
"""
Pops up the menu to be edited
:param item: item clicked
"""
try:
table = self.report_health_table
model_index = table.indexFromItem(item)
row = model_index.row()
self.new_healthTest(table.item(row, 0).text())
except Exception:
if settings.level == 10:
logger.exception('raised exception')
return False, 'Some Internal Error'
def add_table_rows(self, *args):
"""
adds a new row to the pes table
"""
try:
table = self.report_health_table
if args:
table.setRowCount(len(args))
for row, data in enumerate(args):
table.setItem(row, 0, QTableWidgetItem(data['code']))
table.setItem(row, 1, QTableWidgetItem(data['date']))
table.setItem(row, 2,
QTableWidgetItem(data['organization']))
table.setItem(row, 3, QTableWidgetItem(data['test']))
table.setItem(row, 4,
QTableWidgetItem(data['description']))
table.setColumnWidth(0, (table.width() / 5) * 0.5)
table.setColumnWidth(1, (table.width() / 5) * 0.5)
table.setColumnWidth(2, table.width() / 5)
table.setColumnWidth(3, table.width() / 5)
except Exception:
if settings.level == 10:
logger.exception('raised exception')
return False, 'Some Internal Error'
def load_rows(self, event):
"""
loads the rows of the tables
"""
self.add_table_rows()
class Hygiene():
""" The Hygiene Tab
"""
global logger
def __init__(self):
####
logger.info('Inside Hygiene')
self.reporthygiene_tab_2 = QWidget()
self.reporthygiene_tab_2.setObjectName("reporthygiene_tab_2")
self.vertical_23 = QVBoxLayout(self.reporthygiene_tab_2)
self.vertical_23.setObjectName("vertical_23")
self.label_1 = QLabel(self.reporthygiene_tab_2)
self.vertical_23.addWidget(self.label_1)
self.report_hyginepest_table = QTableWidget(self.reporthygiene_tab_2)
self.report_hyginepest_table.setObjectName("report_hyginepest_table")
self.report_hyginepest_table.setColumnCount(5)
self.report_hyginepest_table.setRowCount(0)
self.report_hyginepest_table.setSelectionBehavior(
QAbstractItemView.SelectRows)
item = QTableWidgetItem()
self.report_hyginepest_table.setHorizontalHeaderItem(0, item)
item = QTableWidgetItem()
self.report_hyginepest_table.setHorizontalHeaderItem(1, item)
item = QTableWidgetItem()
self.report_hyginepest_table.setHorizontalHeaderItem(2, item)
item = QTableWidgetItem()
self.report_hyginepest_table.setHorizontalHeaderItem(3, item)
item = QTableWidgetItem()
self.report_hyginepest_table.setHorizontalHeaderItem(4, item)
self.report_hyginepest_table.horizontalHeader(
).setCascadingSectionResizes(True)
self.report_hyginepest_table.horizontalHeader().setStretchLastSection(
True)
self.report_hyginepest_table.verticalHeader(
).setCascadingSectionResizes(True)
self.vertical_23.addWidget(self.report_hyginepest_table)
self.horizontal_21 = QHBoxLayout()
self.report_hyginepest_newrow_buttuon = QPushButton(
self.reporthygiene_tab_2)
self.report_hyginepest_newrow_buttuon.setObjectName(
"report_hyginepest_newrow_buttuon")
self.horizontal_21.addWidget(self.report_hyginepest_newrow_buttuon)
spacerItem23 = QSpacerItem(40, 20, QSizePolicy.Expanding,
QSizePolicy.Minimum)
self.horizontal_21.addItem(spacerItem23)
# self.report_hyginepest_save_button = QPushButton(self.reporthygiene_tab_2)
# self.report_hyginepest_save_button.setObjectName("report_hyginepest_save_button")
# self.horizontal_21.addWidget(self.report_hyginepest_save_button)
self.vertical_23.addLayout(self.horizontal_21)
self.label_2 = QLabel(self.reporthygiene_tab_2)
self.vertical_23.addWidget(self.label_2)
self.report_hyginewater_table = QTableWidget(self.reporthygiene_tab_2)
self.report_hyginewater_table.setObjectName("report_hyginewater_table")
self.report_hyginewater_table.setColumnCount(5)
self.report_hyginewater_table.setRowCount(0)
self.report_hyginewater_table.setSelectionBehavior(
QAbstractItemView.SelectRows)
item = QTableWidgetItem()
self.report_hyginewater_table.setHorizontalHeaderItem(0, item)
item = QTableWidgetItem()
self.report_hyginewater_table.setHorizontalHeaderItem(1, item)
item = QTableWidgetItem()
self.report_hyginewater_table.setHorizontalHeaderItem(2, item)
item = QTableWidgetItem()
self.report_hyginewater_table.setHorizontalHeaderItem(3, item)
item = QTableWidgetItem()
self.report_hyginewater_table.setHorizontalHeaderItem(4, item)
self.report_hyginewater_table.horizontalHeader(
).setCascadingSectionResizes(True)
self.report_hyginewater_table.horizontalHeader().setStretchLastSection(
True)
self.report_hyginewater_table.verticalHeader(
).setCascadingSectionResizes(True)
self.vertical_23.addWidget(self.report_hyginewater_table)
self.horizontal_22 = QHBoxLayout()
self.report_hyginewater_newrow_buttuon = QPushButton(
self.reporthygiene_tab_2)
self.report_hyginewater_newrow_buttuon.setObjectName(
"report_hyginewater_newrow_buttuon")
self.horizontal_22.addWidget(self.report_hyginewater_newrow_buttuon)
spacerItem24 = QSpacerItem(40, 20, QSizePolicy.Expanding,
QSizePolicy.Minimum)
self.horizontal_22.addItem(spacerItem24)
# self.report_hyginewater_save_button = QPushButton(self.reporthygiene_tab_2)
# self.report_hyginewater_save_button.setObjectName("report_hyginewater_save_button")
# self.horizontal_22.addWidget(self.report_hyginewater_save_button)
self.vertical_23.addLayout(self.horizontal_22)
### retanslate
self.label_1.setText(
QApplication.translate("MainWindow", "Pest Test Report", None,
QApplication.UnicodeUTF8))
self.report_hyginepest_table.horizontalHeaderItem(0).setText(
QApplication.translate("MainWindow", "Code", None,
QApplication.UnicodeUTF8))
self.report_hyginepest_table.horizontalHeaderItem(1).setText(
QApplication.translate("MainWindow", "Date", None,
QApplication.UnicodeUTF8))
self.report_hyginepest_table.horizontalHeaderItem(2).setText(
QApplication.translate("MainWindow", "Organization Name", None,
QApplication.UnicodeUTF8))
self.report_hyginepest_table.horizontalHeaderItem(3).setText(
QApplication.translate("MainWindow", "Test", None,
QApplication.UnicodeUTF8))
self.report_hyginepest_table.horizontalHeaderItem(4).setText(
QApplication.translate("MainWindow", "Description", None,
QApplication.UnicodeUTF8))
self.report_hyginepest_newrow_buttuon.setText(
QApplication.translate("MainWindow", "New Row", None,
QApplication.UnicodeUTF8))
# self.report_hyginepest_save_button.setText(
# QApplication.translate("MainWindow", "Save", None, QApplication.UnicodeUTF8))
self.label_2.setText(
QApplication.translate("MainWindow", "Water Test Report", None,
QApplication.UnicodeUTF8))
self.report_hyginewater_table.horizontalHeaderItem(0).setText(
QApplication.translate("MainWindow", "Code", None,
QApplication.UnicodeUTF8))
self.report_hyginewater_table.horizontalHeaderItem(1).setText(
QApplication.translate("MainWindow", "Date", None,
QApplication.UnicodeUTF8))
self.report_hyginewater_table.horizontalHeaderItem(2).setText(
QApplication.translate("MainWindow", "Organization Name", None,
QApplication.UnicodeUTF8))
self.report_hyginewater_table.horizontalHeaderItem(3).setText(
QApplication.translate("MainWindow", "Test", None,
QApplication.UnicodeUTF8))
self.report_hyginewater_table.horizontalHeaderItem(4).setText(
QApplication.translate("MainWindow", "Description", None,
QApplication.UnicodeUTF8))
self.report_hyginewater_newrow_buttuon.setText(
QApplication.translate("MainWindow", "New Row", None,
QApplication.UnicodeUTF8))
# self.report_hyginewater_save_button.setText(
# QApplication.translate("MainWindow", "Save", None, QApplication.UnicodeUTF8))
###signals and slots && other stuffs
self.pest = Pest()
self.water = Water()
self.report_hyginepest_table.setEditTriggers(
QAbstractItemView.NoEditTriggers)
self.report_hyginewater_table.setEditTriggers(
QAbstractItemView.NoEditTriggers)
self.load_table_rows()
self.report_hyginepest_table.itemDoubleClicked.connect(
self.popup_pest_edit)
self.report_hyginewater_table.itemDoubleClicked.connect(
self.popup_water_edit)
self.reporthygiene_tab_2.focusInEvent = self.load_rows # very important for focus
self.report_hyginepest_newrow_buttuon.clicked.connect(
self.new_pestTest) # if no focus available then we need lambda
self.report_hyginewater_newrow_buttuon.clicked.connect(
self.new_waterTest)
def new_pestTest(self, code=None):
"""
The pest report popup generator
:param code: The code of the pest report
"""
pop = HygieneReportPop(parent=self, table='pest_table', code=code)
pop.setWindowFlags(Qt.WindowTitleHint)
pop.exec_()
self.load_table_rows()
def new_waterTest(self, code=None):
"""
The water report popup generator
:param code: The code of the water report
"""
pop = HygieneReportPop(parent=self, table='water_table', code=code)
pop.setWindowFlags(Qt.WindowTitleHint)
pop.exec_()
self.load_table_rows()
def load_table_rows(self):
"""
Populates the rows for both the table
"""
logger.info('Hygiene save table initiated')
try:
data = self.pest.load_report()
if data[0]:
self.add_table_rows('pest_table', *data[1])
if not data[1]:
self.report_hyginepest_table.clearContents()
self.report_hyginepest_table.setRowCount(0)
data = self.water.load_report()
if data[0]:
self.add_table_rows('water_table', *data[1])
if not data[1]:
self.report_hyginewater_table.clearContents()
self.report_hyginewater_table.setRowCount(0)
except Exception:
if settings.level == 10:
logger.exception('raised exception')
return False, 'Some Internal Error'
def popup_pest_edit(self, item):
"""
Pops up the menu to be edited
:param item: item clicked
"""
table = self.report_hyginepest_table
model_index = table.indexFromItem(item)
row = model_index.row()
self.new_pestTest(table.item(row, 0).text())
def popup_water_edit(self, item):
"""
Pops up the menu to be edited
:param item: item clicked
"""
table = self.report_hyginewater_table
model_index = table.indexFromItem(item)
row = model_index.row()
self.new_waterTest(table.item(row, 0).text())
def add_table_rows(self, tablename, *args):
"""
adds a new row to the pes table
"""
if tablename == 'pest_table':
table = self.report_hyginepest_table
else:
table = self.report_hyginewater_table
if args:
table.setRowCount(len(args))
for row, data in enumerate(args):
table.setItem(row, 0, QTableWidgetItem(data['code']))
table.setItem(row, 1, QTableWidgetItem(data['date']))
table.setItem(row, 2, QTableWidgetItem(data['organization']))
table.setItem(row, 3, QTableWidgetItem(data['test']))
table.setItem(row, 4, QTableWidgetItem(data['description']))
table.setColumnWidth(0, (table.width() / 5) * 0.5)
table.setColumnWidth(1, (table.width() / 5) * 0.5)
table.setColumnWidth(2, table.width() / 5)
table.setColumnWidth(3, table.width() / 5)
def load_rows(self, event):
"""
loads the rows of the tables
"""
self.add_table_rows(tablename='pest_table')
self.add_table_rows(tablename='water_table')
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 NotificationTab():
"""ui class for new notification tab"""
global logger
def __init__(self):
#####
logger.info('Inside PurchaseSchedule')
self.notificationTab_tab_4 = QWidget()
self.notificationTab_tab_4.setObjectName("notificationTab_tab_4")
self.gridLayout_19 = QGridLayout(self.notificationTab_tab_4)
self.gridLayout_19.setObjectName("gridLayout_19")
##
self.horizontalLayout = QHBoxLayout()
self.horizontalLayout.setObjectName("horizontalLayout")
self.title_label = QLabel(self.notificationTab_tab_4)
self.title_label.setObjectName("title_label")
self.horizontalLayout.addWidget(self.title_label)
self.gridLayout_19.addLayout(self.horizontalLayout, 0, 0, 1, 1)
##
self.horizontalLayout_6 = QHBoxLayout()
self.horizontalLayout_6.setObjectName("horizontalLayout_6")
self.id_label = QLabel(self.notificationTab_tab_4)
self.id_label.setObjectName("id_label")
self.horizontalLayout_6.addWidget(self.id_label)
self.id_line = QLineEdit(self.notificationTab_tab_4)
self.id_line.setObjectName("id_line")
self.horizontalLayout_6.addWidget(self.id_line)
self.notification_schedule_fromdate_label = QLabel(
self.notificationTab_tab_4)
self.notification_schedule_fromdate_label.setObjectName(
"notification_schedule_fromdate_label")
self.horizontalLayout_6.addWidget(
self.notification_schedule_fromdate_label)
self.notification_schedule_fromdate_dateedit = QDateEdit(
self.notificationTab_tab_4)
self.notification_schedule_fromdate_dateedit.setMaximumDate(
QDate.currentDate())
self.notification_schedule_fromdate_dateedit.setDate(
QDate.currentDate())
self.notification_schedule_fromdate_dateedit.setCalendarPopup(True)
self.notification_schedule_fromdate_dateedit.setObjectName(
"notification_schedule_fromdate_dateedit")
self.horizontalLayout_6.addWidget(
self.notification_schedule_fromdate_dateedit)
self.notification_schedule_todate_label = QLabel(
self.notificationTab_tab_4)
self.notification_schedule_todate_label.setObjectName(
"notification_schedule_todate_label")
self.horizontalLayout_6.addWidget(
self.notification_schedule_todate_label)
self.notification_schedule_todate_dateedit = QDateEdit(
self.notificationTab_tab_4)
self.notification_schedule_todate_dateedit.setMaximumDate(
QDate.currentDate())
self.notification_schedule_todate_dateedit.setDate(QDate.currentDate())
self.notification_schedule_todate_dateedit.setCalendarPopup(True)
self.notification_schedule_todate_dateedit.setObjectName(
"notification_schedule_todate_dateedit")
self.horizontalLayout_6.addWidget(
self.notification_schedule_todate_dateedit)
self.type_label = QLabel(self.notificationTab_tab_4)
self.type_label.setObjectName("type_label")
self.horizontalLayout_6.addWidget(self.type_label)
self.notification_states = QComboBox(self.notificationTab_tab_4)
self.notification_states.setObjectName("notification_states")
self.horizontalLayout_6.addWidget(self.notification_states)
self.batch_label = QLabel(self.notificationTab_tab_4)
self.batch_label.setObjectName("batch_label")
self.horizontalLayout_6.addWidget(self.batch_label)
self.notification_results = QComboBox(self.notificationTab_tab_4)
self.notification_results.setObjectName("notification_results")
self.horizontalLayout_6.addWidget(self.notification_results)
self.gridLayout_19.addLayout(self.horizontalLayout_6, 1, 0, 1, 1)
self.gridLayout_8 = QGridLayout()
self.gridLayout_8.setObjectName("gridLayout_8")
self.notification_schedule_table = QTableWidget(
self.notificationTab_tab_4)
self.notification_schedule_table.setObjectName(
"notification_schedule_table")
self.notification_schedule_table.setColumnCount(5)
self.notification_schedule_table.setRowCount(0)
self.notification_schedule_table.setWordWrap(True)
item = QTableWidgetItem()
self.notification_schedule_table.setHorizontalHeaderItem(0, item)
item = QTableWidgetItem()
self.notification_schedule_table.setHorizontalHeaderItem(1, item)
item = QTableWidgetItem()
self.notification_schedule_table.setHorizontalHeaderItem(2, item)
item = QTableWidgetItem()
self.notification_schedule_table.setHorizontalHeaderItem(3, item)
item = QTableWidgetItem()
self.notification_schedule_table.setHorizontalHeaderItem(4, item)
self.notification_schedule_table.horizontalHeader(
).setCascadingSectionResizes(True)
self.notification_schedule_table.horizontalHeader(
).setStretchLastSection(True)
self.notification_schedule_table.verticalHeader().setVisible(False)
self.notification_schedule_table.verticalHeader(
).setCascadingSectionResizes(True)
self.notification_schedule_table.verticalHeader(
).setStretchLastSection(False)
self.gridLayout_8.addWidget(self.notification_schedule_table, 0, 0, 1,
5)
self.notification_search_button = QPushButton(
self.notificationTab_tab_4)
self.notification_search_button.setObjectName(
"notification_search_button")
self.gridLayout_8.addWidget(self.notification_search_button, 1, 0, 1,
1)
spacerItem10 = QSpacerItem(40, 20, QSizePolicy.Expanding,
QSizePolicy.Minimum)
self.gridLayout_8.addItem(spacerItem10, 1, 2, 1, 1)
self.notification_reset_button = QPushButton(
self.notificationTab_tab_4)
self.notification_reset_button.setObjectName(
"notification_reset_button")
self.gridLayout_8.addWidget(self.notification_reset_button, 1, 3, 1, 1)
self.notification_load_more_button = QPushButton(
self.notificationTab_tab_4)
self.notification_load_more_button.setObjectName(
"notification_load_more_button")
self.gridLayout_8.addWidget(self.notification_load_more_button, 1, 4,
1, 1)
self.gridLayout_19.addLayout(self.gridLayout_8, 2, 0, 1, 1)
###retranslate
self.title_label.setText(
QApplication.translate(
"MainWindow", "<html><head/><body><p align=\"center\">"
"<span style=\" font-weight:600;font-size:20px\">"
"<u>All Notifications</u></span></p></body></html>", None,
QApplication.UnicodeUTF8))
self.id_label.setText(
QApplication.translate("MainWindow", "Message Id", None,
QApplication.UnicodeUTF8))
self.notification_schedule_fromdate_label.setText(
QApplication.translate("MainWindow", "From Date", None,
QApplication.UnicodeUTF8))
self.notification_schedule_fromdate_dateedit.setDisplayFormat(
QApplication.translate("MainWindow", "dd/MM/yyyy", None,
QApplication.UnicodeUTF8))
self.notification_schedule_todate_label.setText(
QApplication.translate("MainWindow", "To Date", None,
QApplication.UnicodeUTF8))
self.notification_schedule_todate_dateedit.setDisplayFormat(
QApplication.translate("MainWindow", "dd/MM/yyyy", None,
QApplication.UnicodeUTF8))
self.type_label.setText(
QApplication.translate("MainWindow", "Type", None,
QApplication.UnicodeUTF8))
self.batch_label.setText(
QApplication.translate("MainWindow", "Number of Notifications",
None, QApplication.UnicodeUTF8))
self.notification_schedule_table.horizontalHeaderItem(0).setText(
QApplication.translate("MainWindow", "Message Id", None,
QApplication.UnicodeUTF8))
self.notification_schedule_table.horizontalHeaderItem(1).setText(
QApplication.translate("MainWindow", "Date", None,
QApplication.UnicodeUTF8))
self.notification_schedule_table.horizontalHeaderItem(2).setText(
QApplication.translate("MainWindow", "From", None,
QApplication.UnicodeUTF8))
self.notification_schedule_table.horizontalHeaderItem(3).setText(
QApplication.translate("MainWindow", "Message", None,
QApplication.UnicodeUTF8))
self.notification_schedule_table.horizontalHeaderItem(4).setText(
QApplication.translate("MainWindow", "State", None,
QApplication.UnicodeUTF8))
self.notification_search_button.setText(
QApplication.translate("MainWindow", "Search", None,
QApplication.UnicodeUTF8))
self.notification_reset_button.setText(
QApplication.translate("MainWindow", "Reset All", None,
QApplication.UnicodeUTF8))
self.notification_load_more_button.setText(
QApplication.translate("MainWindow", "Load More", None,
QApplication.UnicodeUTF8))
##signals and slotts && other stuffs
self.scheduletable_count = 0
self.addtable_count = 0
# self.mainwindow = Ui_MainWindow # just for the ease of finding the attributes in pycharm
self.notification_schedule_table.setEditTriggers(
QAbstractItemView.NoEditTriggers)
self.batch_number = None
self.notification_results.addItems([str(i) for i in range(1, 50, 5)])
self.notification_states.addItems(['All', 'New', 'To Do', 'Done'])
self.message = Messaging()
self.notification_load_more_button.clicked.connect(self.load_more)
self.notification_search_button.clicked.connect(self.search_messages)
self.notification_reset_button.clicked.connect(self.reset_all)
self.notificationTab_tab_4.setFocusPolicy(Qt.StrongFocus)
self.notificationTab_tab_4.focusInEvent = self.load_rows
# self.assign_shortcuts()
# def assign_shortcuts(self):
# QShortcut(QKeySequence(settings.custom_shortcut['key_tabcon_inventorynotification_search']),
# self.notificationTab_tab_4, self.search_messages)
# QShortcut(QKeySequence(settings.custom_shortcut['key_tabcon_inventorynotification_additem']),
# self.notificationTab_tab_4, self.add_new_blank_rows)
# QShortcut(QKeySequence(settings.custom_shortcut['key_tabcon_inventorynotification_print']),
# self.notificationTab_tab_4, self.commit_and_print)
# QShortcut(QKeySequence(settings.custom_shortcut['key_tabcon_inventorynotification_today']),
# self.notificationTab_tab_4, lambda: self.load_messages('today'))
# QShortcut(QKeySequence(settings.custom_shortcut['key_tabcon_inventorynotification_tommorow']),
# self.notificationTab_tab_4, lambda: self.load_messages('tomorrow'))
# QShortcut(QKeySequence(settings.custom_shortcut['key_tabcon_inventorynotification_thismonth']),
# self.notificationTab_tab_4, lambda: self.load_messages('month'))
# QShortcut(QKeySequence(settings.custom_shortcut['key_tabcon_inventorynotification_thisweek']),
# self.notificationTab_tab_4, lambda: self.load_messages('week'))
# QShortcut(QKeySequence(settings.custom_shortcut['key_tabcon_inventorynotification_clear']),
# self.notificationTab_tab_4, self.clear_table)
def add_messages(self, *args):
"""
Populates the Schedules when we load the tab
"""
table = self.notification_schedule_table
if args:
if args[0] != 'new':
table.clearContents()
table.setRowCount(0)
table.setRowCount(len(args))
for i, j in enumerate(args):
item = QTableWidgetItem(j['msg_id'])
table.setItem(i, 0, item)
item = QTableWidgetItem(j['date'])
table.setItem(i, 1, item)
item = QTableWidgetItem(
'Name:{}\nDesignation:{}\nAddress:{}'.format(
j['name'], j['designation'], j['address']))
table.setItem(i, 2, item)
item = QTableWidgetItem(
'Message Type:{}\nMessage:{}'.format(
j['msg_type'].title(), j['msg_body']))
table.setItem(i, 3, item)
states = QComboBox()
self.populate_states(states, j['msg_state'].title())
states.currentIndexChanged.connect(
lambda index, row=i: self.changed_state(row, index))
table.setCellWidget(i, 4, states)
elif args[0] == 'new':
initial = table.rowCount()
row = table.rowCount() + len(args[1])
table.setRowCount(row)
forward_range = range(initial, row)
for i, j in zip(forward_range, args[1]):
item = QTableWidgetItem(j['msg_id'])
table.setItem(i, 0, item)
item = QTableWidgetItem(j['date'])
table.setItem(i, 1, item)
item = QTableWidgetItem(
'Name:{}\nDesignation:{}\nAddress:{}'.format(
j['name'], j['designation'], j['address']))
table.setItem(i, 2, item)
item = QTableWidgetItem(
'Message Type:{}\nMessage:{}'.format(
j['msg_type'].title(), j['msg_body']))
table.setItem(i, 3, item)
states = QComboBox()
self.populate_states(states, j['msg_state'].title())
states.currentIndexChanged.connect(
lambda index, row=i: self.changed_state(row, index))
table.setCellWidget(i, 4, states)
table.setColumnWidth(0, (table.width() * 0.5) / 5)
table.setColumnWidth(1, (table.width() * 0.5) / 5)
table.setColumnWidth(2, table.width() / 5)
table.setColumnWidth(3, (table.width() * 2) / 5)
self.notification_schedule_table.resizeRowsToContents()
def populate_states(self, combo, state):
"""
fills the supplier list for each item line
:param combo: the combo box of suppliers
:return:none
"""
combo.setStyleSheet("QAbstractItemView{" "background: #4B77BE;" "}")
combo.addItems(['New', 'To Do', 'Done'])
index = combo.findText(state)
combo.setCurrentIndex(index)
def changed_state(self, row, index):
"""
fill the item combo box
:param combo: the combobox object
:return: none
"""
table = self.notification_schedule_table
data = {}
data['message_id'] = table.item(row, 0).text()
data['msg_state'] = table.cellWidget(row, 4).currentText()
msg = QMessageBox.information(
QMessageBox(), 'Alert!!',
'Do you want to change the status of the Message??',
QMessageBox.Yes | QMessageBox.No)
if msg == QMessageBox.Yes:
report = self.message.update_message(data)
if report:
_ = QMessageBox.information(QMessageBox(), 'Success!!',
'The Message was updated',
QMessageBox.Yes | QMessageBox.No)
else:
_ = QMessageBox.critical(QMessageBox(), 'Error!!',
'The Message was not updated',
QMessageBox.Yes | QMessageBox.No)
def search_messages(self):
"""
Searches for messages when search button is pressed
"""
logger.info('Notifications searching messages initiated')
from_date = self.notification_schedule_fromdate_dateedit.text()
to_date = self.notification_schedule_todate_dateedit.text()
limit = self.notification_results.currentText()
msg_id = self.id_line.text()
msg_state = self.notification_states.currentText()
dataobj = self.message.load_message(limit=limit,
from_date=from_date,
to_date=to_date,
msg_id=msg_id,
msg_state=msg_state)
if dataobj:
self.add_messages(*dataobj)
else:
self.notification_schedule_table.clearContents()
self.notification_schedule_table.setRowCount(0)
def load_more(self):
"""
Searches for messages when search button is pressed
"""
logger.info('Notifications searching messages initiated')
from_date = self.notification_schedule_fromdate_dateedit.text()
to_date = self.notification_schedule_todate_dateedit.text()
limit = self.notification_results.currentText()
msg_id = self.id_line.text()
msg_state = self.notification_states.currentText()
offset = self.notification_schedule_table.rowCount()
dataobj = self.message.load_message(limit=limit,
from_date=from_date,
to_date=to_date,
msg_id=msg_id,
msg_state=msg_state,
offset=offset)
if dataobj:
self.add_messages('new', dataobj)
else:
self.notification_schedule_table.clearContents()
self.notification_schedule_table.setRowCount(0)
def load_single_message(self, msg_id): # api
"""method to load a single message"""
self.reset_all()
dataobj = self.message.load_message(msg_id=msg_id)
if dataobj:
self.add_messages('new', dataobj)
self.id_line.setText(str(msg_id))
else:
self.notification_schedule_table.clearContents()
self.notification_schedule_table.setRowCount(0)
def load_rows(self, event):
"""
:return:loads the rows
"""
self.add_messages()
def reset_all(self):
"""
:return: resets the the data in the search text
"""
try:
self.id_line.clear()
self.notification_schedule_table.clearContents()
self.notification_schedule_table.setRowCount(0)
except Exception as _:
if settings.level == 10:
logger.exception('raised exception')
return False
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 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 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 WasteItems():
"""Item waste management tab"""
global logger
def __init__(self):
###
logger.info('Inside WasteItems')
self.wastedetail_tab_1 = QWidget()
self.wastedetail_tab_1.setObjectName("wastedetail_tab_1")
self.verticalLayout_9 = QVBoxLayout(self.wastedetail_tab_1)
self.verticalLayout_9.setObjectName("verticalLayout_9")
self.verticalLayout_8 = QVBoxLayout()
self.verticalLayout_8.setObjectName("verticalLayout_8")
self.horizontalLayout_12 = QHBoxLayout()
self.horizontalLayout_12.setObjectName("horizontalLayout_12")
self.waste_fromdate_label = QLabel(self.wastedetail_tab_1)
self.waste_fromdate_label.setObjectName('waste_fromdate_label')
self.horizontalLayout_12.addWidget(self.waste_fromdate_label)
self.waste_fromdate_dateedit = QDateEdit(self.wastedetail_tab_1)
self.waste_fromdate_dateedit.setCalendarPopup(True)
self.waste_fromdate_dateedit.setObjectName("waste_fromdate_dateedit")
self.horizontalLayout_12.addWidget(self.waste_fromdate_dateedit)
self.waste_todate_label = QLabel(self.wastedetail_tab_1)
self.waste_todate_label.setObjectName('waste_todate_label')
self.horizontalLayout_12.addWidget(self.waste_todate_label)
self.waste_todate_dateedit = QDateEdit(self.wastedetail_tab_1)
self.waste_todate_dateedit.setCalendarPopup(True)
self.waste_todate_dateedit.setObjectName("waste_todate_dateedit")
self.waste_todate_dateedit.setMaximumDate(QDate.currentDate())
self.horizontalLayout_12.addWidget(self.waste_todate_dateedit)
spacerItem28 = QSpacerItem(40, 20, QSizePolicy.Expanding,
QSizePolicy.Minimum)
self.horizontalLayout_12.addItem(spacerItem28)
self.waste_table_search_button = QPushButton(self.wastedetail_tab_1)
self.waste_table_search_button.setObjectName(
"waste_table_search_button")
self.horizontalLayout_12.addWidget(self.waste_table_search_button)
self.verticalLayout_8.addLayout(self.horizontalLayout_12)
self.waste_table = QTableWidget(self.wastedetail_tab_1)
self.waste_table.setObjectName("waste_table")
self.waste_table.setColumnCount(6)
self.waste_table.setRowCount(0)
item = QTableWidgetItem()
self.waste_table.setHorizontalHeaderItem(0, item)
item = QTableWidgetItem()
self.waste_table.setHorizontalHeaderItem(1, item)
item = QTableWidgetItem()
self.waste_table.setHorizontalHeaderItem(2, item)
item = QTableWidgetItem()
self.waste_table.setHorizontalHeaderItem(3, item)
item = QTableWidgetItem()
self.waste_table.setHorizontalHeaderItem(4, item)
item = QTableWidgetItem()
self.waste_table.setHorizontalHeaderItem(5, item)
self.waste_table.horizontalHeader().setCascadingSectionResizes(False)
self.waste_table.horizontalHeader().setStretchLastSection(True)
self.waste_table.verticalHeader().setVisible(False)
self.waste_table.verticalHeader().setCascadingSectionResizes(False)
self.waste_table.setSortingEnabled(True)
self.verticalLayout_8.addWidget(self.waste_table)
self.verticalLayout_9.addLayout(self.verticalLayout_8)
##retranslate
self.waste_fromdate_label.setText(
QApplication.translate("MainWindow", "From Date", None,
QApplication.UnicodeUTF8))
self.waste_fromdate_dateedit.setDisplayFormat(
QApplication.translate("MainWindow", "dd/MM/yyyy", None,
QApplication.UnicodeUTF8))
self.waste_todate_label.setText(
QApplication.translate("MainWindow", "To Date", None,
QApplication.UnicodeUTF8))
self.waste_todate_dateedit.setDisplayFormat(
QApplication.translate("MainWindow", "dd/MM/yyyy", None,
QApplication.UnicodeUTF8))
self.waste_table_search_button.setText(
QApplication.translate("MainWindow", "Search", None,
QApplication.UnicodeUTF8))
self.waste_table.horizontalHeaderItem(0).setText(
QApplication.translate("MainWindow", "Code", None,
QApplication.UnicodeUTF8))
self.waste_table.horizontalHeaderItem(1).setText(
QApplication.translate("MainWindow", "Item", None,
QApplication.UnicodeUTF8))
self.waste_table.horizontalHeaderItem(2).setText(
QApplication.translate("MainWindow", "Category", None,
QApplication.UnicodeUTF8))
self.waste_table.horizontalHeaderItem(3).setText(
QApplication.translate("MainWindow", "Units", None,
QApplication.UnicodeUTF8))
self.waste_table.horizontalHeaderItem(4).setText(
QApplication.translate("MainWindow", "Quantity", None,
QApplication.UnicodeUTF8))
self.waste_table.horizontalHeaderItem(5).setText(
QApplication.translate("MainWindow", "Reason", None,
QApplication.UnicodeUTF8))
###signals and slots && other stuffs
# self.mainwindow = Ui_MainWindow # just for the ease of finding the attributes in pycharm
self.item = WasteIngredients()
self.waste_fromdate_dateedit.setDate(QDate.currentDate())
self.waste_todate_dateedit.setDate(QDate.currentDate())
self.waste_table.setEditTriggers(QAbstractItemView.NoEditTriggers)
self.waste_table_search_button.clicked.connect(self.search_discard)
self.wastedetail_tab_1.setFocusPolicy(Qt.StrongFocus)
self.wastedetail_tab_1.focusInEvent = self.load_rows
def load_rows(self, event=None):
"""calling add row to restructure the table every time container resizes"""
self.add_row_to_table()
def search_discard(self):
"""
searches the discard from_date and to_date
:return:none
"""
f_date = self.waste_fromdate_dateedit.text()
from_date = datetime.strptime(f_date, '%d/%m/%Y')
t_date = self.waste_todate_dateedit.text()
to_date = datetime.strptime(t_date, '%d/%m/%Y')
to_date = to_date + timedelta(hours=23, minutes=59, seconds=59)
dataobj = self.item.find_itemdiscard(from_date=from_date,
to_date=to_date)
self.add_row_to_table(*dataobj)
def add_row_to_table(self, *args):
"""
complex stuff of auto complete to be added to each combo box
:return:
"""
table = self.waste_table
if args:
table.clearContents()
table.setRowCount(0)
table.setRowCount(len(args))
for i, j in enumerate(args):
code = QTableWidgetItem(j['code'])
table.setItem(i, 0, code)
item = QTableWidgetItem(j['item'])
table.setItem(i, 1, item)
category = QTableWidgetItem(j['category'])
table.setItem(i, 2, category)
unit = QTableWidgetItem(str(j['units']))
table.setItem(i, 3, unit)
quantity = QTableWidgetItem(str(j['quantity']))
table.setItem(i, 4, quantity)
reason = QTableWidgetItem(j['reason_for_discard'])
table.setItem(i, 5, reason)
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))
table.horizontalHeader().setStretchLastSection(True)
class WasteDish():
"""Dish waste management tab"""
global logger
def __init__(self):
###
logger.info('Inside WasteDish')
self.wastedetail_tab_1 = QWidget()
self.wastedetail_tab_1.setObjectName("wastedetail_tab_1")
self.verticalLayout_9 = QVBoxLayout(self.wastedetail_tab_1)
self.verticalLayout_9.setObjectName("verticalLayout_9")
self.verticalLayout_8 = QVBoxLayout()
self.verticalLayout_8.setObjectName("verticalLayout_8")
self.horizontalLayout_12 = QHBoxLayout()
self.horizontalLayout_12.setObjectName("horizontalLayout_12")
self.waste_fromdate_label = QLabel(self.wastedetail_tab_1)
self.waste_fromdate_label.setObjectName('waste_fromdate_label')
self.horizontalLayout_12.addWidget(self.waste_fromdate_label)
self.waste_fromdate_dateedit = QDateEdit(self.wastedetail_tab_1)
self.waste_fromdate_dateedit.setCalendarPopup(True)
self.waste_fromdate_dateedit.setObjectName("waste_fromdate_dateedit")
self.horizontalLayout_12.addWidget(self.waste_fromdate_dateedit)
self.waste_todate_label = QLabel(self.wastedetail_tab_1)
self.waste_todate_label.setObjectName('waste_todate_label')
self.horizontalLayout_12.addWidget(self.waste_todate_label)
self.waste_todate_dateedit = QDateEdit(self.wastedetail_tab_1)
self.waste_todate_dateedit.setCalendarPopup(True)
self.waste_todate_dateedit.setObjectName("waste_todate_dateedit")
self.waste_todate_dateedit.setMaximumDate(QDate.currentDate())
self.horizontalLayout_12.addWidget(self.waste_todate_dateedit)
spacerItem28 = QSpacerItem(40, 20, QSizePolicy.Expanding,
QSizePolicy.Minimum)
self.horizontalLayout_12.addItem(spacerItem28)
self.waste_table_search_button = QPushButton(self.wastedetail_tab_1)
self.waste_table_search_button.setObjectName(
"waste_table_search_button")
self.horizontalLayout_12.addWidget(self.waste_table_search_button)
self.verticalLayout_8.addLayout(self.horizontalLayout_12)
self.waste_table = QTableWidget(self.wastedetail_tab_1)
self.waste_table.setObjectName("waste_table")
self.waste_table.setColumnCount(5)
self.waste_table.setRowCount(0)
item = QTableWidgetItem()
self.waste_table.setHorizontalHeaderItem(0, item)
item = QTableWidgetItem()
self.waste_table.setHorizontalHeaderItem(1, item)
item = QTableWidgetItem()
self.waste_table.setHorizontalHeaderItem(2, item)
item = QTableWidgetItem()
self.waste_table.setHorizontalHeaderItem(3, item)
item = QTableWidgetItem()
self.waste_table.setHorizontalHeaderItem(4, item)
self.waste_table.horizontalHeader().setCascadingSectionResizes(False)
self.waste_table.horizontalHeader().setStretchLastSection(True)
self.waste_table.verticalHeader().setVisible(False)
self.waste_table.verticalHeader().setCascadingSectionResizes(False)
self.waste_table.setSortingEnabled(True)
self.verticalLayout_8.addWidget(self.waste_table)
self.horizontalLayout_13 = QHBoxLayout()
self.horizontalLayout_13.setObjectName("horizontalLayout_13")
self.waste_table_newrow_button = QPushButton(self.wastedetail_tab_1)
self.waste_table_newrow_button.setObjectName(
"waste_table_newrow_button")
self.horizontalLayout_13.addWidget(self.waste_table_newrow_button)
spacerItem29 = QSpacerItem(612, 20, QSizePolicy.Expanding,
QSizePolicy.Minimum)
self.horizontalLayout_13.addItem(spacerItem29)
self.waste_table_discard_button = QPushButton(self.wastedetail_tab_1)
self.waste_table_discard_button.setObjectName(
"waste_table_discard_button")
self.horizontalLayout_13.addWidget(self.waste_table_discard_button)
self.verticalLayout_8.addLayout(self.horizontalLayout_13)
self.verticalLayout_9.addLayout(self.verticalLayout_8)
##retranslate
self.waste_fromdate_label.setText(
QApplication.translate("MainWindow", "From Date", None,
QApplication.UnicodeUTF8))
self.waste_fromdate_dateedit.setDisplayFormat(
QApplication.translate("MainWindow", "dd/MM/yyyy", None,
QApplication.UnicodeUTF8))
self.waste_todate_label.setText(
QApplication.translate("MainWindow", "To Date", None,
QApplication.UnicodeUTF8))
self.waste_todate_dateedit.setDisplayFormat(
QApplication.translate("MainWindow", "dd/MM/yyyy", None,
QApplication.UnicodeUTF8))
self.waste_table_search_button.setText(
QApplication.translate("MainWindow", "Search", None,
QApplication.UnicodeUTF8))
self.waste_table.horizontalHeaderItem(0).setText(
QApplication.translate("MainWindow", "Code", None,
QApplication.UnicodeUTF8))
self.waste_table.horizontalHeaderItem(1).setText(
QApplication.translate("MainWindow", "Item", None,
QApplication.UnicodeUTF8))
self.waste_table.horizontalHeaderItem(2).setText(
QApplication.translate("MainWindow", "Category", None,
QApplication.UnicodeUTF8))
self.waste_table.horizontalHeaderItem(3).setText(
QApplication.translate("MainWindow", "Quantity", None,
QApplication.UnicodeUTF8))
self.waste_table.horizontalHeaderItem(4).setText(
QApplication.translate("MainWindow", "Reason", None,
QApplication.UnicodeUTF8))
self.waste_table_newrow_button.setText(
QApplication.translate("MainWindow", "New Row", None,
QApplication.UnicodeUTF8))
self.waste_table_discard_button.setText(
QApplication.translate("MainWindow", "Discard Item", None,
QApplication.UnicodeUTF8))
self.wastedetail_tab_1.setTabOrder(self.waste_fromdate_dateedit,
self.waste_todate_dateedit)
self.wastedetail_tab_1.setTabOrder(self.waste_todate_dateedit,
self.waste_table_search_button)
###signals and slots && other stuffs
# self.mainwindow = Ui_MainWindow # just for the ease of finding the attributes in pycharm
self.schedule = SchedulePurchase()
self.suplier = BusinessParty(category='Supplier')
self.add = AddStockInventory()
self.item = WasteMenu()
self.waste_fromdate_dateedit.setDate(QDate.currentDate())
self.waste_todate_dateedit.setDate(QDate.currentDate())
self.waste_table.setEditTriggers(QAbstractItemView.NoEditTriggers)
self.waste_table_newrow_button.clicked.connect(self.add_new_blank_rows)
self.waste_table_discard_button.clicked.connect(self.discard)
self.waste_table_search_button.clicked.connect(self.search_discard)
self.wastedetail_tab_1.setFocusPolicy(Qt.StrongFocus)
self.wastedetail_tab_1.focusInEvent = self.load_rows
def load_rows(self, event=None):
pass
def add_new_blank_rows(self):
"""
deletes the schedules in the database
"""
table = self.waste_table
item = table.item(0, 0)
if item:
message = QMessageBox.critical(QMessageBox(), 'Warning!',
'This will remove all the entries',
QMessageBox.Ok | QMessageBox.Cancel)
if message == QMessageBox.Ok:
table.setRowCount(0)
table.clearContents()
self.add_row_to_table('new')
self.waste_table_discard_button.setVisible(True)
elif not item:
self.waste_table_discard_button.setVisible(True)
self.add_row_to_table('new')
def add_row_to_table(self, *args):
"""
complex stuff of auto complete to be added to each combo box
:return:
"""
table = self.waste_table
if args:
if args[0] != 'new':
table.clearContents()
table.setRowCount(0)
table.setRowCount(len(args))
for i, j in enumerate(args):
code = QTableWidgetItem(j['code'])
table.setItem(i, 0, code)
item = QTableWidgetItem(j['item'])
table.setItem(i, 1, item)
category = QTableWidgetItem(j['category'])
table.setItem(i, 2, category)
quantity = QTableWidgetItem(str(j['quantity']))
table.setItem(i, 3, quantity)
reason = QTableWidgetItem(j['reason_for_discard'])
table.setItem(i, 4, reason)
if args[0] == 'new':
row = table.rowCount() + 1
table.setRowCount(row)
codeline = QLineEdit()
codeline.editingFinished.connect(
lambda: self.get_details_of_code(row))
table.setCellWidget(row - 1, 0, codeline)
itemcombo = QComboBox()
self.fill_item_list(itemcombo)
itemcombo.currentIndexChanged.connect(
lambda: self.get_details_of_item(row))
table.setCellWidget(row - 1, 1, itemcombo)
category = QTableWidgetItem()
table.setItem(row - 1, 2, category)
quantity = QLineEdit()
table.setCellWidget(row - 1, 3, quantity)
combo = QComboBox()
combo.addItem("Cancelled")
combo.addItem("Mishandling")
combo.addItem("Excess")
table.setCellWidget(row - 1, 4, combo)
table.setColumnWidth(0, (table.width() / 5))
table.setColumnWidth(1, (table.width() / 5))
table.setColumnWidth(2, (table.width() / 5))
table.setColumnWidth(3, (table.width() / 5))
table.horizontalHeader().setStretchLastSection(
True
) # important to resize last section else blank space after last column
def fill_item_list(self, combo):
"""
fill the item combo box
:param combo: the combobox object
:return: none
"""
itemfield = combo
itemfield.setStyleSheet("QAbstractItemView{"
"background: #4B77BE;"
"}")
self.item.populate_item(itemfield)
def get_details_of_code(self, rowcount):
"""
fills the item, category and units based on the code
:param rowcount: the row count
:return: none
"""
row = rowcount - 1
table = self.waste_table
codeline = table.cellWidget(row, 0)
data = self.item.get_details_of_code(codeline.text())
item = table.cellWidget(row, 1)
index = item.findText(data['item'])
item.setCurrentIndex(index)
category = table.item(row, 2)
category.setText(data['category'])
def get_details_of_item(self, rowcount):
"""
fills the code, category and units based on the item
:param rowcount: the row count
:return: none
"""
row = rowcount - 1
table = self.waste_table
itemcombo = table.cellWidget(row, 1)
data = self.item.get_details_of_item(itemcombo.currentText())
code = table.cellWidget(row, 0)
code.setText(data['code'])
category = table.item(row, 2)
category.setText(data['category'])
def discard(self):
"""
saves the details in db before printing
"""
logger.info('WasteDish discard initiated')
table = self.waste_table
data = self.get_data()
if data:
for i in data:
status = self.item.discard(i)
if status:
model_index = table.indexFromItem(i['model_item'])
row = model_index.row()
table.removeRow(row)
else:
msg = QMessageBox.critical(QMessageBox(), "Error!!",
"The item cannot be discarded",
QMessageBox.Ok)
if msg == QMessageBox.Ok:
return False
def search_discard(self):
"""
searches the discard from_date and to_date
:return:none
"""
table = self.waste_table
item = table.cellWidget(0, 0)
if item:
msg = QMessageBox.critical(QMessageBox(), 'Warning!',
'This will delete all the rows added',
QMessageBox.Ok | QMessageBox.Cancel)
if msg == QMessageBox.Cancel:
return False
f_date = self.waste_fromdate_dateedit.text()
from_date = datetime.strptime(f_date, '%d/%m/%Y')
t_date = self.waste_todate_dateedit.text()
to_date = datetime.strptime(t_date, '%d/%m/%Y')
to_date = to_date + timedelta(hours=23, minutes=59, seconds=59)
dataobj = self.item.find_itemdiscard(from_date=from_date,
to_date=to_date)
self.add_row_to_table(*dataobj)
self.waste_table_discard_button.setVisible(False)
def get_data(self):
"""
:return: fetches all the data for printing
"""
table = self.waste_table
rows = table.rowCount()
dataobj = []
for i in range(rows):
dictionary = {}
item = table.cellWidget(i, 0) if table.cellWidget(
i, 0) is not None else table.item(i, 0)
dictionary['code'] = item.text()
if dictionary['code'] == '':
break
item = table.cellWidget(i, 1).currentText() if table.cellWidget(
i, 1) is not None else table.item(i, 1).text()
dictionary['item'] = item
item = table.cellWidget(i, 2) if table.cellWidget(
i, 2) is not None else table.item(i, 2)
dictionary['category'] = item.text()
item = table.cellWidget(i, 3) if table.cellWidget(
i, 3) is not None else table.item(i, 3)
dictionary['quantity'] = item.text()
if dictionary['quantity'] == '':
self.show_error('Quantity')
return False
item = table.cellWidget(i, 4).currentText() if table.cellWidget(
i, 4) is not None else table.item(i, 4).text()
dictionary['reason_for_discard'] = item
dictionary['model_item'] = table.item(i, 2)
dataobj.append(dictionary)
return dataobj
def show_error(self, text):
"""
:return: pops up an error
"""
QMessageBox.critical(QMessageBox(), "Fail!!",
"Please Enter %s properly" % text, QMessageBox.Ok)
def __init__(self):
url = QLineEdit()
quality = QLineEdit()
urlLabel = QLabel('Url')
qualityLabel = QLabel('Quality')
messages = QTextEdit()
messagesLabel = QLabel('Messages')
links = QTableWidget(0, 2)
linksLabel = QLabel('Links')
clearMessages = QPushButton('Clear Messages')
checkIfOnline = QPushButton('Check If Online')
addSelectedLink = QPushButton('Add Link')
removeSelectedLink = QPushButton('Remove Selected Link')
messages.setReadOnly(True)
links.setHorizontalHeaderLabels(['Url', 'Status'])
links.horizontalHeader().setResizeMode(QHeaderView.Stretch)
links.horizontalHeader().setResizeMode(1, QHeaderView.Fixed)
# set the events
url.returnPressed.connect(self.select_stream_from_entry)
quality.returnPressed.connect(self.select_stream_from_entry)
links.itemDoubleClicked.connect(self.select_stream_from_link)
clearMessages.clicked.connect(self.clear_messages)
checkIfOnline.clicked.connect(self.check_if_online)
addSelectedLink.clicked.connect(self.add_selected_link)
removeSelectedLink.clicked.connect(self.remove_selected_link)
#set shortcut
checkIfOnline.setShortcut(QKeySequence(Qt.Key_F5))
# set the layouts
mainLayout = QGridLayout()
# first row
mainLayout.addWidget(urlLabel, 0, 0, 1, 1) # spans 1 column
mainLayout.addWidget(qualityLabel, 0, 1, 1, 1) # spans 1 column
mainLayout.addWidget(linksLabel, 0, 2, 1, 3) # spans 3 columns
# second row (links widget occupies 2 rows and 2 columns)
mainLayout.addWidget(url, 1, 0, 1, 1) # spans 1 column
mainLayout.addWidget(quality, 1, 1, 1, 1) # spans 1 column
mainLayout.addWidget(links, 1, 2, 2, 3) # spans 3 columns
# third row (messages widget occupies 2 columns)
mainLayout.addWidget(messages, 2, 0, 1, 2)
# fourth row
mainLayout.addWidget(messagesLabel, 3, 0)
mainLayout.addWidget(clearMessages, 3, 1)
mainLayout.addWidget(checkIfOnline, 3, 2)
mainLayout.addWidget(addSelectedLink, 3, 3)
mainLayout.addWidget(removeSelectedLink, 3, 4)
window = QWidget()
window.setLayout(mainLayout)
window.setWindowTitle('Live Streamer')
window.resize(700, 350)
window.show()
self.url_ui = url
self.quality_ui = quality
self.messages_ui = messages
self.links_ui = links
self.window_ui = window
self.links = set()
self.data_file = os.path.join(os.path.expanduser("~"), ".config",
"livestreamer-ui", "data.txt")
folder = os.path.dirname(self.data_file)
if not os.path.exists(folder):
os.makedirs(folder)
