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 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()
