PyQt实现测试工具
2017-02-17 18:08 一方书斋 阅读(928) 评论(0) 编辑 收藏 举报测试工具：
1. 基本界面实现：
# coding:utf-8 import sys import os import os.path import re import time from PyQt4 import QtCore from PyQt4 import QtGui from PyQt4 import Qt msg_file_not_exists = 'Can not find config file' msg_file_is_null = 'There is no attribute now' project_file = 'D:\workspace\py_demo\project.py' case_file = 'D:\workspace\py_demo\case.py' case_seprator = '|' def get_time(): return time.strftime("%Y-%m-%d %X", time.localtime(time.time())) def _save_project(project_index, project_info, create_time, edit_time, project_detail): all_lines = [] line_str = str(project_index) + case_seprator + project_info + case_seprator + \ create_time + case_seprator + edit_time + case_seprator + project_detail + '\n' if os.path.exists(project_file): if os.path.getsize(project_file): with open(project_file, 'r') as f: all_lines = f.readlines() if all_lines[project_index:project_index + 1]: all_lines[project_index] = line_str else: all_lines.append(line_str) else: all_lines.append(line_str) else: all_lines.append(line_str) with open(project_file, 'w') as f: f.writelines(all_lines) def _save_case(project_index, case_index, case_info, create_time, edit_time, start_run_time, finish_run_time, run_type, run_script, run_result): all_lines = [] line_str = str(project_index) + case_seprator + str( case_index) + case_seprator + case_info + case_seprator + \ create_time + case_seprator + edit_time + case_seprator + \ start_run_time + case_seprator + finish_run_time + \ case_seprator + run_type + case_seprator + run_script +\ case_seprator + run_result + '\n' if os.path.exists(case_file): if os.path.getsize(case_file): with open(case_file, 'r') as f: all_lines = f.readlines() if all_lines[case_index:case_index + 1]: all_lines[case_index] = line_str else: all_lines.append(line_str) else: all_lines.append(line_str) else: all_lines.append(line_str) with open(case_file, 'w') as f: f.writelines(all_lines) class MainWindow(QtGui.QMainWindow): def __init__(self): super(MainWindow, self).__init__() self.setWindowTitle('AutoTest') self.resize(600, 300) self.setTool = self.addToolBar('Set') self.setToolAction = Qt.QAction( QtGui.QIcon('images/open.png'), u'Set', self) self.setToolAction.setShortcut('Ctrl+S') self.setToolAction.setStatusTip(u'Config') self.setToolAction.triggered.connect(self.set_config_file) self.setTool.addAction(self.setToolAction) self.newProjectTool = self.addToolBar('NewProject') self.newProjectToolAction = Qt.QAction( QtGui.QIcon('images/selectall.png'), u'CreateProject', self) self.newProjectToolAction.triggered.connect(self.newproject) self.newProjectTool.addAction(self.newProjectToolAction) self.newCaseTool = self.addToolBar('NewCase') self.newCaseToolAction = Qt.QAction( QtGui.QIcon('images/copy.png'), u'CreateCase', self) self.newCaseToolAction.triggered.connect(self.newcase) self.newCaseTool.addAction(self.newCaseToolAction) wedgit = QtGui.QWidget() layout = QtGui.QVBoxLayout() self.tree = QtGui.QTreeWidget() self.tree.setColumnCount(4) self.tree.setHeaderLabels(['Step', 'Start Time','Finish Time','Result']) # self.tree.headerItem().setStretchLastSection(True) self.load_case() self.tree.itemChanged.connect(self.save_project) self.tree.itemDoubleClicked.connect(self.edit_project) # self.tree.setExpanded(0,True) self.log_text = QtGui.QTextBrowser() layout.addWidget(self.tree) layout.addWidget(self.log_text) wedgit.setLayout(layout) self.setCentralWidget(wedgit) def set_config_file(self): self.dialog = config_file() self.dialog.show() def load_case(self): if os.path.exists(project_file): if os.path.getsize(project_file): with open(project_file, 'r') as f: all_lines = f.readlines() for i, line in enumerate(all_lines): project_text = line.split(case_seprator)[1] self.project = QtGui.QTreeWidgetItem(self.tree) self.tree.setItemExpanded(self.project, True) # 设置自动扩展 self.tree.addTopLevelItem(self.project) self.project.setText(0, project_text) project_no = line.split(case_seprator)[0] if os.path.exists(case_file): if os.path.getsize(case_file): with open(case_file, 'r') as f: case_all_line = f.readlines() for i, line in enumerate(case_all_line): case_project_no = line.split( case_seprator)[0] if case_project_no == project_no: case_name = line.split( case_seprator)[1] case_result = line.split( case_seprator)[2] self.case = QtGui.QTreeWidgetItem( self.project) self.case.setText(0, case_name) self.case.setText(0, case_result) def newproject(self): self.tree.blockSignals(True) # 禁止itemchanged信号 self.project = QtGui.QTreeWidgetItem(self.tree) self.tree.addTopLevelItem(self.project) self.project.setFlags(self.project.flags() | QtCore.Qt.ItemIsEditable) self.project.setText(0, 'Here is project information') self.tree.blockSignals(False) self.save_project(self.project, 0) def newcase(self): self.tree.blockSignals(True) check_index = self.tree.indexOfTopLevelItem(self.tree.currentItem()) if check_index > -1: self.project = self.tree.currentItem() else: self.project = self.tree.currentItem().parent() self.case = QtGui.QTreeWidgetItem(self.project) self.case.setFlags(self.project.flags() | QtCore.Qt.ItemIsEditable) self.case.setText(0, 'Here is case name') self.case.setCheckState(0, QtCore.Qt.Unchecked) self.project.addChild(self.case) self.tree.blockSignals(False) self.save_project(self.case, 0) def save_case(self, item, column): print 'item text', item.text(column) print 'index', self.tree.indexFromItem(item).row() print 'parent text', item.parent().text(0) def save_project(self, item, column): check_index = self.tree.indexOfTopLevelItem(item) if check_index > -1: project_index = check_index project_info = str(item.text(column)) create_time = get_time() edit_time = get_time() _save_project(project_index, project_info, create_time, edit_time, '') else: project_index = self.tree.indexOfTopLevelItem(item.parent()) case_index = self.tree.indexFromItem(item).row() case_info = str(item.text(column)) # case_index.data() create_time=get_time() edit_time=get_time() _save_case(project_index, case_index, case_info,create_time,edit_time,'','','','','') print 'index', project_index print 'text', item.text(0) print 'column', column def edit_project(self,item,column): current_index = self.tree.indexOfTopLevelItem(item) print current_index if current_index > -1: project_dialog = project(None,current_index) project_dialog.exec_() if project_dialog.reply == QtGui.QMessageBox.Yes: #窗口之间传值 project_title = project_dialog.get_project_title() item.setText(0,project_title) else: project_index = self.tree.indexOfTopLevelItem(item.parent()) case_index = self.tree.indexFromItem(item).row() case_dialog = case(None,project_index,case_index) case_dialog.exec_() if case_dialog.reply == QtGui.QMessageBox.Yes: case_title = case_dialog.get_case_title() item.setText(0,case_title) class case(QtGui.QDialog): def __init__(self, parent, project_index, case_index): super(case, self).__init__(parent) self.project_index = project_index self.case_index = case_index self.setWindowTitle('Case') self.resize(500, 300) self.setup_ui() self.update_flag = False def closeEvent(self, event): if self.update_flag: self.reply = QtGui.QMessageBox.question( self, 'Warning', 'Some content was changed,save?', QtGui.QMessageBox.Yes, QtGui.QMessageBox.No) if self.reply == QtGui.QMessageBox.No: # event.ignore() # 选择No直接关闭窗口，event.ignore表示保留原窗口 pass else: case_edit_time = get_time() case_title = self.case_title_edit_line.text() _save_case(self.project_index, int(self.caseNo), self.case_title_edit_line.text( ), self.case_create_time, case_edit_time,'','',self.case_run_type_edit_line.currentText (),'','pending run') def set_update_flag(self): self.update_flag = True def get_case_title(self): return self.case_title_edit_line.text() def setup_ui(self): all_lines=[] with open(case_file, 'r') as f: for line in f.readlines(): if int(line.split(case_seprator)[0]) == int(self.project_index): all_lines.append(line) case_info = all_lines[self.case_index] self.caseNo = case_info.split(case_seprator)[1] case_title = case_info.split(case_seprator)[2] self.case_create_time = case_info.split(case_seprator)[3] case_edit_time = case_info.split(case_seprator)[4] case_start_time = case_info.split(case_seprator)[5] case_finish_time = case_info.split(case_seprator)[6] case_run_type = case_info.split(case_seprator)[7] case_run_script = case_info.split(case_seprator)[8] case_run_result = case_info.split(case_seprator)[9] mainlayout = QtGui.QVBoxLayout() layout1 = QtGui.QHBoxLayout() self.case_number_label = QtGui.QLabel('No:') self.case_number_read_line = QtGui.QLabel(self.caseNo) self.case_number_label.setBuddy(self.case_number_read_line) layout1.addWidget(self.case_number_label) layout1.addWidget(self.case_number_read_line) layout1.setAlignment(QtCore.Qt.AlignLeft | QtCore.Qt.AlignVCenter) layout2 = QtGui.QHBoxLayout() self.case_title_label = QtGui.QLabel('Title:') self.case_title_edit_line = QtGui.QLineEdit(case_title) self.case_title_label.setBuddy(self.case_title_edit_line) layout2.addWidget(self.case_title_label) layout2.addWidget(self.case_title_edit_line) layout2.setAlignment(QtCore.Qt.AlignLeft | QtCore.Qt.AlignVCenter) self.case_title_edit_line.textChanged.connect(self.set_update_flag) layout3 = QtGui.QHBoxLayout() self.case_create_time_label = QtGui.QLabel('CreateTime:') self.case_create_time_read_line = QtGui.QLabel( self.case_create_time) self.case_create_time_label.setBuddy( self.case_create_time_read_line) layout3.addWidget(self.case_create_time_label) layout3.addWidget(self.case_create_time_read_line) layout3.setAlignment(QtCore.Qt.AlignLeft | QtCore.Qt.AlignVCenter) layout4 = QtGui.QHBoxLayout() self.case_edit_time_label = QtGui.QLabel('EditTime:') self.case_edit_time_read_line = QtGui.QLabel(case_edit_time) self.case_edit_time_label.setBuddy(self.case_edit_time_read_line) layout4.addWidget(self.case_edit_time_label) layout4.addWidget(self.case_edit_time_read_line) layout4.setAlignment(QtCore.Qt.AlignLeft | QtCore.Qt.AlignVCenter) layout5 = QtGui.QHBoxLayout() self.case_start_time_label = QtGui.QLabel('Start Run Time:') self.case_start_time_read_line = QtGui.QLineEdit(case_start_time) self.case_start_time_label.setBuddy(self.case_start_time_read_line) layout5.addWidget(self.case_start_time_label) layout5.addWidget(self.case_start_time_read_line) layout5.setAlignment(QtCore.Qt.AlignLeft | QtCore.Qt.AlignVCenter) layout6 = QtGui.QHBoxLayout() self.case_finish_time_label = QtGui.QLabel('Finish Run Time:') self.case_finish_time_read_line = QtGui.QLineEdit(case_finish_time) self.case_finish_time_label.setBuddy(self.case_finish_time_read_line) layout6.addWidget(self.case_finish_time_label) layout6.addWidget(self.case_finish_time_read_line) layout6.setAlignment(QtCore.Qt.AlignLeft | QtCore.Qt.AlignVCenter) layout7 = QtGui.QHBoxLayout() self.case_run_type_label = QtGui.QLabel('Run Type:') self.case_run_type_edit_line = QtGui.QComboBox(self) self.case_run_type_edit_line.addItems(['Shell','Python']) self.case_run_type_label.setBuddy(self.case_run_type_edit_line) layout7.addWidget(self.case_run_type_label) layout7.addWidget(self.case_run_type_edit_line) layout7.setAlignment(QtCore.Qt.AlignLeft | QtCore.Qt.AlignVCenter) layout8 = QtGui.QHBoxLayout() self.case_run_script_label = QtGui.QLabel('Run Script:') self.case_run_script_edit_line = QtGui.QTextEdit(case_run_script) self.case_run_script_label.setBuddy(self.case_run_script_edit_line) layout8.addWidget(self.case_run_script_label) layout8.addWidget(self.case_run_script_edit_line) layout8.setAlignment(QtCore.Qt.AlignLeft | QtCore.Qt.AlignVCenter) layout9 = QtGui.QHBoxLayout() self.case_run_result_label = QtGui.QLabel('Run Result:') self.case_run_result_read_line = QtGui.QLineEdit(case_run_result) self.case_run_result_label.setBuddy(self.case_run_result_read_line) layout9.addWidget(self.case_run_result_label) layout9.addWidget(self.case_run_result_read_line) layout9.setAlignment(QtCore.Qt.AlignLeft | QtCore.Qt.AlignVCenter) mainlayout.addLayout(layout1) mainlayout.addLayout(layout2) mainlayout.addLayout(layout3) mainlayout.addLayout(layout4) mainlayout.addLayout(layout5) mainlayout.addLayout(layout6) mainlayout.addLayout(layout7) mainlayout.addLayout(layout8) mainlayout.addLayout(layout9) self.setLayout(mainlayout) class project(QtGui.QDialog): def __init__(self, parent,project_index): super(project, self).__init__(parent) self.project_index = project_index self.setWindowTitle('Project') self.resize(500, 300) self.setup_ui() self.update_flag = False def closeEvent(self, event): if self.update_flag: self.reply = QtGui.QMessageBox.question( self, 'Warning', 'Some content was changed,save?', QtGui.QMessageBox.Yes, QtGui.QMessageBox.No) if self.reply == QtGui.QMessageBox.No: event.accept() # 选择No直接关闭窗口，event.ignore表示保留原窗口 else: edit_time = get_time() project_title = self.project_title_edit_line.text() _save_project(int(self.projectNo), self.project_title_edit_line.text( ), self.project_create_time, edit_time, self.project_detail_edit_line.toPlainText()) def set_update_flag(self): self.update_flag = True def get_project_title(self): return self.project_title_edit_line.text() def setup_ui(self): with open(project_file, 'r') as f: all_lines = f.readlines() project_info = all_lines[self.project_index] self.projectNo = project_info.split(case_seprator)[0] project_title = project_info.split(case_seprator)[1] self.project_create_time = project_info.split(case_seprator)[2] project_edit_time = project_info.split(case_seprator)[3] project_detail = project_info.split(case_seprator)[4] mainlayout = QtGui.QVBoxLayout() layout1 = QtGui.QHBoxLayout() self.project_number_label = QtGui.QLabel('No:') self.project_number_read_line = QtGui.QLabel(self.projectNo) self.project_number_label.setBuddy(self.project_number_read_line) layout1.addWidget(self.project_number_label) layout1.addWidget(self.project_number_read_line) layout1.setAlignment(QtCore.Qt.AlignLeft | QtCore.Qt.AlignVCenter) layout2 = QtGui.QHBoxLayout() self.project_title_label = QtGui.QLabel('Title:') self.project_title_edit_line = QtGui.QLineEdit(project_title) self.project_title_label.setBuddy(self.project_title_edit_line) layout2.addWidget(self.project_title_label) layout2.addWidget(self.project_title_edit_line) layout2.setAlignment(QtCore.Qt.AlignLeft | QtCore.Qt.AlignVCenter) self.project_title_edit_line.textChanged.connect(self.set_update_flag) layout3 = QtGui.QHBoxLayout() self.project_create_time_label = QtGui.QLabel('CreateTime:') self.project_create_time_read_line = QtGui.QLabel( self.project_create_time) self.project_create_time_label.setBuddy( self.project_create_time_read_line) layout3.addWidget(self.project_create_time_label) layout3.addWidget(self.project_create_time_read_line) layout3.setAlignment(QtCore.Qt.AlignLeft | QtCore.Qt.AlignVCenter) layout4 = QtGui.QHBoxLayout() self.project_edit_time_label = QtGui.QLabel('EditTime:') self.project_edit_time_read_line = QtGui.QLabel(project_edit_time) self.project_edit_time_label.setBuddy(self.project_edit_time_read_line) layout4.addWidget(self.project_edit_time_label) layout4.addWidget(self.project_edit_time_read_line) layout4.setAlignment(QtCore.Qt.AlignLeft | QtCore.Qt.AlignVCenter) layout5 = QtGui.QHBoxLayout() self.project_detail_label = QtGui.QLabel('Detail:') self.project_detail_edit_line = QtGui.QTextEdit(project_detail) self.project_detail_label.setBuddy(self.project_detail_edit_line) layout5.addWidget(self.project_detail_label) layout5.addWidget(self.project_detail_edit_line) layout5.setAlignment(QtCore.Qt.AlignLeft | QtCore.Qt.AlignVCenter) self.project_detail_edit_line.textChanged.connect(self.set_update_flag) mainlayout.addLayout(layout1) mainlayout.addLayout(layout2) mainlayout.addLayout(layout3) mainlayout.addLayout(layout4) mainlayout.addLayout(layout5) self.setLayout(mainlayout) class config_file(QtGui.QDialog): def __init__(self, parent=None): super(config_file, self).__init__(parent) self.file_name = 'D:\workspace\py_demo\NotePad\config.py' self.separator_char = '=' self.config_table = QtGui.QTableWidget(0, 2) self.config_table.setSelectionMode( QtGui.QAbstractItemView.SingleSelection) # 设置为只能选中单个目标 self.config_table.setSelectionBehavior( QtGui.QAbstractItemView.SelectRows) # 设置选中单个单元格 self.config_table.setEditTriggers( QtGui.QAbstractItemView.NoEditTriggers) # 默认双击可以编辑，这里设置禁止编辑 self.config_table.setAlternatingRowColors(True) # 隔行改变颜色 self.config_table.setShowGrid(True) # 设置显示网格线 self.config_table.setSortingEnabled(False) # 设置禁止排序 self.config_table.setHorizontalHeaderLabels(['Attribute', 'Value']) self.load_config() layout = QtGui.QVBoxLayout() btnlayout = QtGui.QHBoxLayout() self.add_item_btn = QtGui.QPushButton() self.add_item_btn.setText('Add') self.add_item_btn.clicked.connect(self.add_config) self.edit_item_btn = QtGui.QPushButton() self.edit_item_btn.setText('Edit') self.edit_item_btn.clicked.connect(self.edit_config) self.delete_item_btn = QtGui.QPushButton() self.delete_item_btn.setText('Delete') self.delete_item_btn.clicked.connect(self.delete_config) btnlayout.addStretch() btnlayout.addWidget(self.add_item_btn) btnlayout.addWidget(self.edit_item_btn) btnlayout.addWidget(self.delete_item_btn) layout.addWidget(self.config_table) layout.addLayout(btnlayout) self.setLayout(layout) def load_config(self): if os.path.exists(self.file_name): if os.path.getsize(self.file_name): with open(self.file_name, 'r') as f: for line_count, line in enumerate(f.readlines()): if self.separator_char in line: attribute = line.split(self.separator_char)[ 0].strip() value = line.split(self.separator_char)[1].strip() self.config_table.insertRow(line_count) self.newAttr = QtGui.QTableWidgetItem(attribute) self.config_table.setItem( line_count, 0, self.newAttr) self.newValue = QtGui.QTableWidgetItem(value) self.config_table.setItem( line_count, 1, self.newValue) else: self.config_table.insertRow(0) self.newAttr = QtGui.QTableWidgetItem( 'There is no attribute now') self.config_table.setSpan(0, 0, 1, 2) self.newAttr.setTextAlignment( QtCore.Qt.AlignHCenter | QtCore.Qt.AlignVCenter) self.config_table.setItem(0, 0, self.newAttr) else: self.config_table.insertRow(0) self.newAttr = QtGui.QTableWidgetItem(msg_file_not_exists) self.config_table.setSpan(0, 0, 1, 2) self.newAttr.setTextAlignment( QtCore.Qt.AlignHCenter | QtCore.Qt.AlignVCenter) self.config_table.setItem(0, 0, self.newAttr) def add_config(self): if str(self.config_table.item(0, 0).text()) in (msg_file_is_null, msg_file_not_exists): current_row_index = self.config_table.currentRow() self.config_table.removeRow(current_row_index) total_count_index = self.config_table.rowCount() self.config_table.setEditTriggers( QtGui.QAbstractItemView.DoubleClicked) self.config_table.insertRow(total_count_index) def edit_config(self): self.config_table.setEditTriggers( QtGui.QAbstractItemView.DoubleClicked) def delete_config(self): current_row_index = self.config_table.currentRow() if current_row_index != -1: reply = QtGui.QMessageBox.warning( self, 'Warning', 'Could you confirm to delete the config', QtGui.QMessageBox.Yes, QtGui.QMessageBox.No) if reply == QtGui.QMessageBox.Yes: self.config_table.removeRow(current_row_index) def closeEvent(self, event): row_count = self.config_table.rowCount() all_lines = [] for row in range(row_count): if self.config_table.item(row, 0) and self.config_table.item(row, 1): line = str(self.config_table.item(row, 0).text()) + \ self.separator_char + \ str(self.config_table.item(row, 1).text()) all_lines.append(line + '\n') else: QtGui.QMessageBox.information( self, 'Error', 'Config shoud not be null') event.ignore() return with open(self.file_name, 'w') as f: f.writelines(all_lines) app = QtGui.QApplication(sys.argv) mainwindow = MainWindow() mainwindow.show() app.exec_()
