Example #1
0
 def load(self, name):
     sql = "SELECT * FROM solution WHERE name='%s'" % name
     q = QSqlQueryModel()
     q.setQuery(sql)
     if(q.rowCount()):
         server_model = ServerModel()
         return server_model.findByIds(q.record(0).value("servers").toString())
     else:
         return []
Example #2
0
 def load(self, name):
     sql = "SELECT * FROM solution WHERE name='%s'" % name
     q = QSqlQueryModel()
     q.setQuery(sql)
     if (q.rowCount()):
         server_model = ServerModel()
         return server_model.findByIds(
             q.record(0).value("servers").toString())
     else:
         return []
Example #3
0
    def __init__(self, parent=None):
        """
        Constructor
        
        @param parent reference to the parent widget (QWidget)
        """
        super(Server, self).__init__(parent)
        self.setupUi(self)

        self.server_id = None
        self.platform_model = PlatformModel()
        self.server_model = ServerModel()

        self.initData()
Example #4
0
    def __init__(self, parent=None):
        """
        Constructor
        
        @param parent reference to the parent widget (QWidget)
        """
        self.server_model = ServerModel()
        self.platform_model = PlatformModel()
        self.solution_model = SolutionModel()

        self.upload_thread = UploadThread()
        self.upload_thread.finished.connect(self.on_upload_thread_finished)
        self.upload_thread.information.connect(
            self.on_upload_thread_information)

        super(MainWindow, self).__init__(parent)
        self.setupUi(self)
        self.initUi()
        self.initData()
        self.createContextMenu()
Example #5
0
    def __init__(self, parent=None):
        """
        Constructor
        
        @param parent reference to the parent widget (QWidget)
        """
        super(Server, self).__init__(parent)
        self.setupUi(self)

        self.server_id = None
        self.platform_model = PlatformModel()
        self.server_model = ServerModel()

        self.initData()
Example #6
0
File: main.py Project: BlueAaron/-
 def __init__(self, parent=None):
     """
     Constructor
     
     @param parent reference to the parent widget (QWidget)
     """
     self.server_model = ServerModel()
     self.platform_model = PlatformModel()
     self.solution_model = SolutionModel()
     
     self.upload_thread = UploadThread()
     self.upload_thread.finished.connect(self.on_upload_thread_finished)
     self.upload_thread.information.connect(self.on_upload_thread_information)
     
     super(MainWindow,self).__init__(parent)
     self.setupUi(self)      
     self.initUi()
     self.initData()
     self.createContextMenu()
Example #7
0
class Server(QDialog, Ui_Server):
    saved = pyqtSignal()

    def __init__(self, parent=None):
        """
        Constructor
        
        @param parent reference to the parent widget (QWidget)
        """
        super(Server, self).__init__(parent)
        self.setupUi(self)

        self.server_id = None
        self.platform_model = PlatformModel()
        self.server_model = ServerModel()

        self.initData()

    def initData(self):
        self.platform.addItems(self.platform_model.list())

    @pyqtSlot()
    def on_add_platform_clicked(self):
        """
        Slot documentation goes here.
        """
        text, ok = QtGui.QInputDialog.getText(self, u"添加", u"平台名称:")
        if ok == True:
            if self.platform_model.check(text) > 0:
                QtGui.QMessageBox.critical(self, u"提示", u"平台名称已存在!")
            elif self.platform_model.add(text):
                self.platform.addItem(text)
                self.platform.setCurrentIndex(self.platform.count() - 1)
            else:
                QtGui.QMessageBox.critical(self, u"提示", u"添加平台失败")

    @pyqtSlot()
    def on_edit_platform_clicked(self):
        """
        Slot documentation goes here.
        """
        if self.platform.count() == 0:
            QtGui.QMessageBox.information(self, u"提示", u"没有平台可以编辑!")
        else:
            old_text = self.platform.currentText()
            new_text, ok = QtGui.QInputDialog.getText(self, u"添加", u"平台名称:", 0, old_text)
            if ok == True:
                if self.platform_model.check(new_text) > 0:
                    QtGui.QMessageBox.critical(self, u"提示", u"平台名称已存在!")
                elif self.platform_model.update(new_text, old_text):
                    index = self.platform.currentIndex()
                    self.platform.setItemText(index, new_text)
                else:
                    QtGui.QMessageBox.critical(self, u"提示", u"修改平台失败")

    @pyqtSlot()
    def on_delete_platform_clicked(self):
        """
        Slot documentation goes here.
        """
        if self.platform.count() == 0:
            QtGui.QMessageBox.information(self, u"提示", u"没有平台可以删除!")
            return

        answer = QtGui.QMessageBox.question(self, u"提示", u"确定删除当前选中平台?", QtGui.QMessageBox.Yes, QtGui.QMessageBox.No)
        if answer == QtGui.QMessageBox.Yes:
            text = self.platform.currentText()
            if self.server_model.countByPlatform(text) > 0:
                QtGui.QMessageBox.warning(self, u"提示", u"删除平台之前请先删除该平台下面的服务器!")
            else:
                if self.platform_model.delete(text):
                    self.platform.removeItem(self.platform.currentIndex())
                else:
                    QtGui.QMessageBox.critical(self, u"提示", u"删除平台失败")

    @pyqtSlot()
    def on_save_clicked(self):
        """
        Slot documentation goes here.
        """
        if self.validator() == True:
            # platform,server_name,account,password,host,port,path
            fields = {
                "server_id": self.server_id,
                "platform": self.platform.currentText(),
                "server_name": self.server_name.text(),
                "account": self.account.text(),
                "password": self.password.text(),
                "host": self.host.text(),
                "port": self.port.text(),
                "path": self.path.text(),
            }
            if self.server_id == None:
                if self.server_model.add(fields):
                    QtGui.QMessageBox.information(self, u"提示", u"添加服务器成功!")
                    self.saved.emit()
                    self.close()
                else:
                    QtGui.QMessageBox.information(self, u"提示", u"添加服务器失败!")
            else:
                if self.server_model.update(self.server_id, fields):
                    QtGui.QMessageBox.information(self, u"提示", u"修改服务器成功!")
                    self.saved.emit()
                    self.close()
                else:
                    QtGui.QMessageBox.information(self, u"提示", u"修改服务器失败!")

    def validator(self):
        if len(self.server_name.text().trimmed()) < 1:
            QtGui.QMessageBox.information(self, u"提示", u"请输入名称!")
            self.server_name.setFocus()
            return False

        if self.platform.count() < 1:
            QtGui.QMessageBox.information(self, u"提示", u"请添加平台!")
            self.add_platform.clicked.emit(True)
            return False

        if len(self.host.text().trimmed()) < 1:
            QtGui.QMessageBox.information(self, u"提示", u"请输入主机!")
            self.host.setFocus()
            return False

        if len(self.port.text().trimmed()) < 1:
            QtGui.QMessageBox.information(self, u"提示", u"请输入端口!")
            self.port.setFocus()
            return False

        if len(self.path.text().trimmed()) < 1:
            QtGui.QMessageBox.information(self, u"提示", u"请输入路径!")
            self.path.setFocus()
            return False

        if len(self.account.text().trimmed()) < 1:
            QtGui.QMessageBox.information(self, "提示", "请输入账号!")
            self.account.setFocus()
            return False

        if len(self.password.text().trimmed()) < 1:
            QtGui.QMessageBox.information(self, u"提示", u"请输入密码!")
            self.password.setFocus()
            return False

        return True

    def setData(self, data):
        if data != False:
            if data["id"] != None:
                self.server_id = data["id"]
                self.groupBox.setTitle(u"修改")
            else:
                self.groupBox.setTitle(u"复制")

            index = self.platform.findText(data["platform"])
            self.platform.setCurrentIndex(index)
            self.server_name.setText(data["server_name"])
            self.account.setText(data["account"])
            self.password.setText(data["password"])
            self.host.setText(data["host"])
            self.port.setText(data["port"])
            self.path.setText(data["path"])
Example #8
0
class Server(QDialog, Ui_Server):
    saved = pyqtSignal()

    def __init__(self, parent=None):
        """
        Constructor
        
        @param parent reference to the parent widget (QWidget)
        """
        super(Server, self).__init__(parent)
        self.setupUi(self)

        self.server_id = None
        self.platform_model = PlatformModel()
        self.server_model = ServerModel()

        self.initData()

    def initData(self):
        self.platform.addItems(self.platform_model.list())

    @pyqtSlot()
    def on_add_platform_clicked(self):
        """
        Slot documentation goes here.
        """
        text, ok = QtGui.QInputDialog.getText(self, u"添加", u"平台名称:")
        if ok == True:
            if self.platform_model.check(text) > 0:
                QtGui.QMessageBox.critical(self, u"提示", u"平台名称已存在!")
            elif self.platform_model.add(text):
                self.platform.addItem(text)
                self.platform.setCurrentIndex(self.platform.count() - 1)
            else:
                QtGui.QMessageBox.critical(self, u"提示", u"添加平台失败")

    @pyqtSlot()
    def on_edit_platform_clicked(self):
        """
        Slot documentation goes here.
        """
        if self.platform.count() == 0:
            QtGui.QMessageBox.information(self, u"提示", u"没有平台可以编辑!")
        else:
            old_text = self.platform.currentText()
            new_text, ok = QtGui.QInputDialog.getText(self, u"添加", u"平台名称:", 0,
                                                      old_text)
            if ok == True:
                if self.platform_model.check(new_text) > 0:
                    QtGui.QMessageBox.critical(self, u"提示", u"平台名称已存在!")
                elif self.platform_model.update(new_text, old_text):
                    index = self.platform.currentIndex()
                    self.platform.setItemText(index, new_text)
                else:
                    QtGui.QMessageBox.critical(self, u"提示", u"修改平台失败")

    @pyqtSlot()
    def on_delete_platform_clicked(self):
        """
        Slot documentation goes here.
        """
        if self.platform.count() == 0:
            QtGui.QMessageBox.information(self, u"提示", u"没有平台可以删除!")
            return

        answer = QtGui.QMessageBox.question(self, u"提示", u"确定删除当前选中平台?",
                                            QtGui.QMessageBox.Yes,
                                            QtGui.QMessageBox.No)
        if answer == QtGui.QMessageBox.Yes:
            text = self.platform.currentText()
            if self.server_model.countByPlatform(text) > 0:
                QtGui.QMessageBox.warning(self, u"提示", u"删除平台之前请先删除该平台下面的服务器!")
            else:
                if self.platform_model.delete(text):
                    self.platform.removeItem(self.platform.currentIndex())
                else:
                    QtGui.QMessageBox.critical(self, u"提示", u"删除平台失败")

    @pyqtSlot()
    def on_save_clicked(self):
        """
        Slot documentation goes here.
        """
        if self.validator() == True:
            #platform,server_name,account,password,host,port,path
            fields = {
                "server_id": self.server_id,
                "platform": self.platform.currentText(),
                "server_name": self.server_name.text(),
                "account": self.account.text(),
                "password": self.password.text(),
                "host": self.host.text(),
                "port": self.port.text(),
                "path": self.path.text()
            }
            if self.server_id == None:
                if self.server_model.add(fields):
                    QtGui.QMessageBox.information(self, u"提示", u"添加服务器成功!")
                    self.saved.emit()
                    self.close()
                else:
                    QtGui.QMessageBox.information(self, u"提示", u"添加服务器失败!")
            else:
                if self.server_model.update(self.server_id, fields):
                    QtGui.QMessageBox.information(self, u"提示", u"修改服务器成功!")
                    self.saved.emit()
                    self.close()
                else:
                    QtGui.QMessageBox.information(self, u"提示", u"修改服务器失败!")

    def validator(self):
        if len(self.server_name.text().trimmed()) < 1:
            QtGui.QMessageBox.information(self, u"提示", u"请输入名称!")
            self.server_name.setFocus()
            return False

        if self.platform.count() < 1:
            QtGui.QMessageBox.information(self, u"提示", u"请添加平台!")
            self.add_platform.clicked.emit(True)
            return False

        if len(self.host.text().trimmed()) < 1:
            QtGui.QMessageBox.information(self, u"提示", u"请输入主机!")
            self.host.setFocus()
            return False

        if len(self.port.text().trimmed()) < 1:
            QtGui.QMessageBox.information(self, u"提示", u"请输入端口!")
            self.port.setFocus()
            return False

        if len(self.path.text().trimmed()) < 1:
            QtGui.QMessageBox.information(self, u"提示", u"请输入路径!")
            self.path.setFocus()
            return False

        if len(self.account.text().trimmed()) < 1:
            QtGui.QMessageBox.information(self, "提示", "请输入账号!")
            self.account.setFocus()
            return False

        if len(self.password.text().trimmed()) < 1:
            QtGui.QMessageBox.information(self, u"提示", u"请输入密码!")
            self.password.setFocus()
            return False

        return True

    def setData(self, data):
        if data != False:
            if data["id"] != None:
                self.server_id = data["id"]
                self.groupBox.setTitle(u"修改")
            else:
                self.groupBox.setTitle(u"复制")

            index = self.platform.findText(data["platform"])
            self.platform.setCurrentIndex(index)
            self.server_name.setText(data["server_name"])
            self.account.setText(data["account"])
            self.password.setText(data["password"])
            self.host.setText(data["host"])
            self.port.setText(data["port"])
            self.path.setText(data["path"])
Example #9
0
class MainWindow(QMainWindow, Ui_MainWindow):
    """
    Class documentation goes here.
    """
    def __init__(self, parent=None):
        """
        Constructor
        
        @param parent reference to the parent widget (QWidget)
        """
        self.server_model = ServerModel()
        self.platform_model = PlatformModel()
        self.solution_model = SolutionModel()

        self.upload_thread = UploadThread()
        self.upload_thread.finished.connect(self.on_upload_thread_finished)
        self.upload_thread.information.connect(
            self.on_upload_thread_information)

        super(MainWindow, self).__init__(parent)
        self.setupUi(self)
        self.initUi()
        self.initData()
        self.createContextMenu()

    def initUi(self):
        self.file_list.setColumnWidth(0, 400)
        self.file_list.setColumnWidth(2, 150)

        self.selected_servers.setColumnWidth(2, 80)
        self.server_list.setColumnWidth(2, 80)

        self.server_list.hideColumn(0)
        self.selected_servers.hideColumn(0)

        self.server_list.keyPressEvent = self.on_server_list_keyPressEvent
        self.selected_servers.keyPressEvent = self.on_selected_servers_keyPressEvent

        self.statusBar.showMessage(u"状态")

    def initData(self):
        self.platform.addItems(self.platform_model.list())
        self.upload_solution.addItems(self.solution_model.list())

    def createContextMenu(self):
        self.server_list_contextmenu = QtGui.QMenu(self)
        self.server_list_contextmenu.addAction(self.action_move2left)
        self.server_list_contextmenu.addSeparator()
        self.server_list_contextmenu.addAction(self.action_add_server)
        self.server_list_contextmenu.addAction(self.action_edit_server)
        self.server_list_contextmenu.addAction(self.action_copy_server)
        self.server_list_contextmenu.addSeparator()
        self.server_list_contextmenu.addAction(self.action_delete_server)

    @pyqtSlot(QPoint)
    def on_server_list_customContextMenuRequested(self, point):
        """
        Slot documentation goes here.
        """
        point.setY(point.y() + 25)
        if self.server_list.rowCount():
            point.setX(point.x() + 15)
        self.server_list_contextmenu.exec_(self.server_list.mapToGlobal(point))

    @pyqtSlot()
    def on_start_upload_clicked(self):
        """
        Slot documentation goes here.
        """
        if self.file_list.rowCount() < 1:
            QtGui.QMessageBox.information(self, u"提示", u"请先添加要上传的文件!")
            return

        if self.selected_servers.rowCount() < 1:
            QtGui.QMessageBox.information(self, u"提示", u"请先添加服务器到上传列表中!")
            return

        answer = QtGui.QMessageBox.warning(self, u"提示",
                                           u"上传将覆盖服务器设置路径下的同名文件,确定上传吗?",
                                           QtGui.QMessageBox.Yes,
                                           QtGui.QMessageBox.No)
        if answer == QtGui.QMessageBox.No:
            return

        if self.upload_thread.isRunning() == False:
            files = [
                self.file_list.item(i, 0).text()
                for i in range(self.file_list.rowCount())
            ]
            server_ids = [
                str(self.selected_servers.item(i, 0).text())
                for i in range(self.selected_servers.rowCount())
            ]
            servers = self.server_model.findByIds(",".join(server_ids))
            self.upload_thread.setData(files, servers)
            self.upload_thread.start()

            self.stop_upload.setEnabled(True)
            self.start_upload.setEnabled(False)

    @pyqtSlot()
    def on_stop_upload_clicked(self):
        """
        Slot documentation goes here.
        """

        if self.upload_thread.isRunning() == True:
            answer = QtGui.QMessageBox.warning(self, u"提示", u"确定终止当前上传任务?",
                                               QtGui.QMessageBox.Yes,
                                               QtGui.QMessageBox.No)
            if answer == QtGui.QMessageBox.No:
                return

            self.start_upload.setEnabled(True)
            self.stop_upload.setEnabled(False)
            self.upload_thread.terminate()

    def on_upload_thread_finished(self):
        self.start_upload.setEnabled(True)
        self.stop_upload.setEnabled(False)

    @pyqtSlot(str)
    def on_upload_thread_information(self, info):
        time = QTime(QTime.currentTime())
        self.textBrowser.append(u"[%s] %s" % (time.toString(), info))

    @pyqtSlot(str)
    def on_upload_solution_currentIndexChanged(self, p0):
        """
        Slot documentation goes here.
        """
        if self.upload_solution.currentIndex() > 0:
            self.setSelectedServersData(self.solution_model.load(p0))
            self.remove_solution.setEnabled(True)
            self.refresh_server_list.clicked.emit(True)
        else:
            self.setSelectedServersData([])
            self.remove_solution.setEnabled(False)

    @pyqtSlot(str)
    def on_platform_currentIndexChanged(self, p0):
        """
        Slot documentation goes here.
        """
        self.setServerListData(self.server_model.findByPlatform(p0))

    @pyqtSlot()
    def on_move2left_clicked(self):
        """
        Slot documentation goes here.
        """
        rows = list(
            set([index.row() for index in self.server_list.selectedIndexes()]))
        for i in range(len(rows)):
            row = rows[i] - i
            index = self.selected_servers.rowCount()
            self.selected_servers.insertRow(index)
            self.selected_servers.setItem(
                index, 0,
                self.server_list.item(row, 0).clone())
            self.selected_servers.setItem(
                index, 1,
                self.server_list.item(row, 1).clone())
            self.selected_servers.setItem(
                index, 2,
                self.server_list.item(row, 2).clone())
            self.selected_servers.setItem(
                index, 3,
                self.server_list.item(row, 3).clone())
            self.server_list.removeRow(row)

    @pyqtSlot()
    def on_move2right_clicked(self):
        """
        Slot documentation goes here.
        """
        rows = list(
            set([
                index.row()
                for index in self.selected_servers.selectedIndexes()
            ]))
        for i in range(len(rows)):
            row = rows[i] - i
            self.selected_servers.removeRow(row)
        self.refresh_server_list.clicked.emit(True)

    @pyqtSlot()
    def on_menu_about_triggered(self):
        """
        Slot documentation goes here.
        """
        QtGui.QMessageBox.about(self, u"FTP Client", u"Powered by ZLH!")

    @pyqtSlot()
    def on_menu_clear_log_triggered(self):
        self.textBrowser.clear()

    @pyqtSlot()
    def on_add_solution_clicked(self):
        """
        Slot documentation goes here.
        """
        if self.selected_servers.rowCount() < 1:
            QtGui.QMessageBox.information(self, u"提示", u"请先添加服务器到上传列表中!")
            return

        text, ok = QtGui.QInputDialog.getText(self, u"添加", u"上传方案名称:")
        if ok == True:
            if self.solution_model.check(text) > 0:
                QtGui.QMessageBox.critical(self, u"提示", u"上传方案名称已存在!")
                return
            servers = [
                str(self.selected_servers.item(i, 0).text())
                for i in range(self.selected_servers.rowCount())
            ]
            if self.solution_model.add(text, ",".join(servers)):
                self.upload_solution.addItem(text)
                self.upload_solution.setCurrentIndex(
                    self.upload_solution.count() - 1)
                QtGui.QMessageBox.information(self, u"提示", u"操作成功!")
            else:
                QtGui.QMessageBox.critical(self, u"提示", u"操作失败!")

    @pyqtSlot()
    def on_remove_solution_clicked(self):
        """
        Slot documentation goes here.
        """
        if self.upload_solution.currentIndex() == 0:
            QtGui.QMessageBox.information(self, u"FTP Client", u"请选择要删除的上传方案!")
            return

        answer = QtGui.QMessageBox.question(self, u"提示", u"确定删除当前选中上传方案?",
                                            QtGui.QMessageBox.Yes,
                                            QtGui.QMessageBox.No)
        if answer == QtGui.QMessageBox.Yes:
            name = self.upload_solution.currentText()
            if self.solution_model.delete(name):
                self.upload_solution.removeItem(
                    self.upload_solution.currentIndex())
                self.selected_servers.setRowCount(0)
                self.refresh_server_list.clicked.emit(True)
                QtGui.QMessageBox.information(self, u"提示", u"操作成功!")

    @pyqtSlot()
    def on_refresh_server_list_clicked(self):
        """
        Slot documentation goes here.
        """
        platform = self.platform.currentText()
        self.setServerListData(self.server_model.findByPlatform(platform))

    @pyqtSlot()
    def on_server_list_itemSelectionChanged(self):
        rows = self.server_list.selectedIndexes()
        if (rows):
            self.move2left.setEnabled(True)
            self.action_move2left.setEnabled(True)
            self.action_edit_server.setEnabled(True)
            self.action_copy_server.setEnabled(True)
            self.action_delete_server.setEnabled(True)
        else:
            self.move2left.setEnabled(False)
            self.action_move2left.setEnabled(False)
            self.action_edit_server.setEnabled(False)
            self.action_copy_server.setEnabled(False)
            self.action_delete_server.setEnabled(False)

    @pyqtSlot()
    def on_selected_servers_itemSelectionChanged(self):
        rows = self.selected_servers.selectedIndexes()
        if rows:
            self.move2right.setEnabled(True)
        else:
            self.move2right.setEnabled(False)

    def on_server_list_keyPressEvent(self, event):
        if event.key() == Qt.Key_Escape:
            self.server_list.clearSelection()
        elif event.key() == Qt.Key_Left:
            self.move2left.clicked.emit(True)

    def on_selected_servers_keyPressEvent(self, event):
        if event.key() == Qt.Key_Escape:
            self.selected_servers.clearSelection()
        elif event.key() == Qt.Key_Delete or event.key() == Qt.Key_Right:
            self.move2right.clicked.emit(True)

    @pyqtSlot()
    def on_action_add_server_triggered(self):
        """
        Slot documentation goes here.
        """
        self.server = Server()
        self.server.saved.connect(self.on_server_saved)
        self.server.show()

    @pyqtSlot()
    def on_action_edit_server_triggered(self):
        """
        Slot documentation goes here.
        """
        server_id = self.server_list.item(self.server_list.currentRow(),
                                          0).text()

        self.server = Server()
        self.server.saved.connect(self.on_server_saved)
        self.server.setData(self.server_model.item(server_id))
        self.server.show()

    @pyqtSlot()
    def on_action_copy_server_triggered(self):
        """
        Slot documentation goes here.
        """
        server_id = self.server_list.item(self.server_list.currentRow(),
                                          0).text()
        server_data = self.server_model.item(server_id)
        server_data["id"] = None

        self.server = Server()
        self.server.saved.connect(self.on_server_saved)
        self.server.setData(server_data)
        self.server.show()

    @pyqtSlot()
    def on_action_delete_server_triggered(self):
        """
        Slot documentation goes here.
        """
        index = self.server_list.currentRow()
        platform = self.server_list.item(index, 1).text()
        server = self.server_list.item(index, 2).text()
        answer = QtGui.QMessageBox.question(
            self, u"提示", u"确定删除 %s %s?" % (platform, server),
            QtGui.QMessageBox.Yes, QtGui.QMessageBox.No)
        if answer == QtGui.QMessageBox.Yes:
            server_id = self.server_list.item(index, 0).text()
            if self.server_model.delete(server_id):
                self.server_list.removeRow(index)
                QtGui.QMessageBox.information(self, u"提示", u"操作成功!")
            else:
                QtGui.QMessageBox.critical(self, u"提示", u"操作失败!")

    def on_server_saved(self):
        self.refresh_server_list.clicked.emit(True)

    @pyqtSlot()
    def on_action_move2left_triggered(self):
        self.move2left.clicked.emit(True)

    def setServerListData(self, data):
        self.server_list.setRowCount(0)
        for row in data:
            if self.inSelectedServers(row["id"]) == False:
                index = self.server_list.rowCount()
                self.server_list.insertRow(index)
                self.server_list.setItem(index, 0, QTableWidgetItem(row["id"]))
                self.server_list.setItem(index, 1,
                                         QTableWidgetItem(row["platform"]))
                self.server_list.setItem(index, 2,
                                         QTableWidgetItem(row["server_name"]))
                self.server_list.setItem(index, 3,
                                         QTableWidgetItem(row["path"]))

    def setSelectedServersData(self, data):
        self.selected_servers.setRowCount(0)
        for row in data:
            index = self.selected_servers.rowCount()
            self.selected_servers.insertRow(index)
            self.selected_servers.setItem(index, 0,
                                          QTableWidgetItem(row["id"]))
            self.selected_servers.setItem(index, 1,
                                          QTableWidgetItem(row["platform"]))
            self.selected_servers.setItem(index, 2,
                                          QTableWidgetItem(row["server_name"]))
            self.selected_servers.setItem(index, 3,
                                          QTableWidgetItem(row["path"]))

        self.refresh_server_list.clicked.emit(True)

    def inSelectedServers(self, server_id):
        for i in range(self.selected_servers.rowCount()):
            if self.selected_servers.item(i, 0).text() == server_id:
                return True
        return False
Example #10
0
File: main.py Project: BlueAaron/-
class MainWindow(QMainWindow, Ui_MainWindow):
    
    """
    Class documentation goes here.
    """
    def __init__(self, parent=None):
        """
        Constructor
        
        @param parent reference to the parent widget (QWidget)
        """
        self.server_model = ServerModel()
        self.platform_model = PlatformModel()
        self.solution_model = SolutionModel()
        
        self.upload_thread = UploadThread()
        self.upload_thread.finished.connect(self.on_upload_thread_finished)
        self.upload_thread.information.connect(self.on_upload_thread_information)
        
        super(MainWindow,self).__init__(parent)
        self.setupUi(self)      
        self.initUi()
        self.initData()
        self.createContextMenu()
        
        
    def initUi(self):        
        self.file_list.setColumnWidth(0, 400)
        self.file_list.setColumnWidth(2, 150)
        
        self.selected_servers.setColumnWidth(2, 80)
        self.server_list.setColumnWidth(2, 80)
        
        self.server_list.hideColumn(0)        
        self.selected_servers.hideColumn(0)
        
        self.server_list.keyPressEvent = self.on_server_list_keyPressEvent
        self.selected_servers.keyPressEvent = self.on_selected_servers_keyPressEvent
        
        self.statusBar.showMessage(u"状态")
        
    def initData(self):
        self.platform.addItems(self.platform_model.list())        
        self.upload_solution.addItems(self.solution_model.list())   
                
    def createContextMenu(self):        
        self.server_list_contextmenu = QtGui.QMenu(self)  
        self.server_list_contextmenu.addAction(self.action_move2left) 
        self.server_list_contextmenu.addSeparator()
        self.server_list_contextmenu.addAction(self.action_add_server)  
        self.server_list_contextmenu.addAction(self.action_edit_server)  
        self.server_list_contextmenu.addAction(self.action_copy_server)  
        self.server_list_contextmenu.addSeparator()
        self.server_list_contextmenu.addAction(self.action_delete_server)  
       
    @pyqtSlot(QPoint)
    def on_server_list_customContextMenuRequested(self, point):
        """
        Slot documentation goes here.
        """
        point.setY(point.y() + 25)
        if self.server_list.rowCount():
            point.setX(point.x() + 15)
        self.server_list_contextmenu.exec_(self.server_list.mapToGlobal(point))    
        
    @pyqtSlot()
    def on_start_upload_clicked(self):
        """
        Slot documentation goes here.
        """               
        if self.file_list.rowCount() < 1:
            QtGui.QMessageBox.information(self, u"提示", u"请先添加要上传的文件!")
            return
            
        if self.selected_servers.rowCount() < 1:
            QtGui.QMessageBox.information(self, u"提示", u"请先添加服务器到上传列表中!")
            return
            
        answer = QtGui.QMessageBox.warning(self, u"提示", u"上传将覆盖服务器设置路径下的同名文件,确定上传吗?", QtGui.QMessageBox.Yes, QtGui.QMessageBox.No)
        if answer == QtGui.QMessageBox.No:
            return
            
        if self.upload_thread.isRunning() == False:
            files = [self.file_list.item(i, 0).text() for i in range(self.file_list.rowCount())]
            server_ids = [str(self.selected_servers.item(i, 0).text()) for i in range(self.selected_servers.rowCount())]
            servers = self.server_model.findByIds(",".join(server_ids))
            self.upload_thread.setData(files, servers)            
            self.upload_thread.start();
            
            self.stop_upload.setEnabled(True)
            self.start_upload.setEnabled(False)
              
    @pyqtSlot()
    def on_stop_upload_clicked(self):
        """
        Slot documentation goes here.
        """
                    
        if self.upload_thread.isRunning() == True:
            answer = QtGui.QMessageBox.warning(self, u"提示", u"确定终止当前上传任务?", QtGui.QMessageBox.Yes, QtGui.QMessageBox.No)
            if answer == QtGui.QMessageBox.No:
                return
        
            self.start_upload.setEnabled(True)
            self.stop_upload.setEnabled(False)
            self.upload_thread.terminate();
    
    def on_upload_thread_finished(self):
        self.start_upload.setEnabled(True)
        self.stop_upload.setEnabled(False)
    
    @pyqtSlot(str)
    def on_upload_thread_information(self, info):
        time = QTime(QTime.currentTime())
        self.textBrowser.append(u"[%s] %s" % (time.toString(), info))
    
    @pyqtSlot(str)
    def on_upload_solution_currentIndexChanged(self, p0):
        """
        Slot documentation goes here.
        """
        if self.upload_solution.currentIndex() >  0:
            self.setSelectedServersData(self.solution_model.load(p0))
            self.remove_solution.setEnabled(True)
            self.refresh_server_list.clicked.emit(True)
        else:
            self.setSelectedServersData([])
            self.remove_solution.setEnabled(False)
        
    @pyqtSlot(str)
    def on_platform_currentIndexChanged(self, p0):
        """
        Slot documentation goes here.
        """
        self.setServerListData(self.server_model.findByPlatform(p0))
    
    @pyqtSlot()
    def on_move2left_clicked(self):
        """
        Slot documentation goes here.
        """        
        rows = list(set([index.row() for index in self.server_list.selectedIndexes()]))
        for i in range(len(rows)):
            row = rows[i]-i
            index = self.selected_servers.rowCount()
            self.selected_servers.insertRow(index)
            self.selected_servers.setItem(index, 0, self.server_list.item(row, 0).clone())
            self.selected_servers.setItem(index, 1, self.server_list.item(row, 1).clone())
            self.selected_servers.setItem(index, 2, self.server_list.item(row, 2).clone())
            self.selected_servers.setItem(index, 3, self.server_list.item(row, 3).clone())
            self.server_list.removeRow(row)
    
    @pyqtSlot()
    def on_move2right_clicked(self):
        """
        Slot documentation goes here.
        """
        rows = list(set([index.row() for index in self.selected_servers.selectedIndexes()]))
        for i in range(len(rows)):
            row = rows[i]-i
            self.selected_servers.removeRow(row)
        self.refresh_server_list.clicked.emit(True)
             
    @pyqtSlot()
    def on_menu_about_triggered(self):
        """
        Slot documentation goes here.
        """
        QtGui.QMessageBox.about(self, u"FTP Client", u"Powered by ZLH!")
    
    @pyqtSlot()
    def on_menu_clear_log_triggered(self):
        self.textBrowser.clear()
        
    @pyqtSlot()
    def on_add_solution_clicked(self):
        """
        Slot documentation goes here.
        """
        if self.selected_servers.rowCount() < 1:
            QtGui.QMessageBox.information(self, u"提示", u"请先添加服务器到上传列表中!")
            return
        
        text, ok = QtGui.QInputDialog.getText(self, u"添加", u"上传方案名称:")
        if ok == True:
            if self.solution_model.check(text) > 0:
                QtGui.QMessageBox.critical(self, u"提示", u"上传方案名称已存在!")
                return
            servers = [str(self.selected_servers.item(i, 0).text()) for i in range(self.selected_servers.rowCount())]
            if self.solution_model.add(text, ",".join(servers)):
                self.upload_solution.addItem(text)
                self.upload_solution.setCurrentIndex(self.upload_solution.count()-1)
                QtGui.QMessageBox.information(self, u"提示", u"操作成功!")
            else:
                QtGui.QMessageBox.critical(self, u"提示", u"操作失败!")
    
    @pyqtSlot()
    def on_remove_solution_clicked(self):
        """
        Slot documentation goes here.
        """
        if self.upload_solution.currentIndex() ==  0:
            QtGui.QMessageBox.information(self, u"FTP Client", u"请选择要删除的上传方案!")
            return
            
        answer = QtGui.QMessageBox.question(self, u"提示", u"确定删除当前选中上传方案?", QtGui.QMessageBox.Yes, QtGui.QMessageBox.No)
        if answer == QtGui.QMessageBox.Yes:
            name = self.upload_solution.currentText()
            if self.solution_model.delete(name):
                self.upload_solution.removeItem(self.upload_solution.currentIndex())
                self.selected_servers.setRowCount(0)
                self.refresh_server_list.clicked.emit(True)
                QtGui.QMessageBox.information(self, u"提示", u"操作成功!")
        
    @pyqtSlot()
    def on_refresh_server_list_clicked(self):
        """
        Slot documentation goes here.
        """
        platform = self.platform.currentText()
        self.setServerListData(self.server_model.findByPlatform(platform))
         
    @pyqtSlot()
    def on_server_list_itemSelectionChanged(self):
        rows = self.server_list.selectedIndexes()
        if(rows):
            self.move2left.setEnabled(True)
            self.action_move2left.setEnabled(True)
            self.action_edit_server.setEnabled(True)
            self.action_copy_server.setEnabled(True)
            self.action_delete_server.setEnabled(True)
        else:
            self.move2left.setEnabled(False)
            self.action_move2left.setEnabled(False)
            self.action_edit_server.setEnabled(False)
            self.action_copy_server.setEnabled(False)
            self.action_delete_server.setEnabled(False)
   
    @pyqtSlot()
    def on_selected_servers_itemSelectionChanged(self):
        rows = self.selected_servers.selectedIndexes()
        if rows:
            self.move2right.setEnabled(True)
        else:
            self.move2right.setEnabled(False)
        
    def on_server_list_keyPressEvent(self, event):
        if event.key() == Qt.Key_Escape:
            self.server_list.clearSelection()
        elif event.key() == Qt.Key_Left:
            self.move2left.clicked.emit(True)
            
    def on_selected_servers_keyPressEvent(self, event):
        if event.key() == Qt.Key_Escape:
            self.selected_servers.clearSelection()
        elif event.key() == Qt.Key_Delete or event.key() == Qt.Key_Right:
            self.move2right.clicked.emit(True)        
            
    @pyqtSlot()
    def on_action_add_server_triggered(self):
        """
        Slot documentation goes here.
        """
        self.server = Server()
        self.server.saved.connect(self.on_server_saved)
        self.server.show()
    
    @pyqtSlot()
    def on_action_edit_server_triggered(self):
        """
        Slot documentation goes here.
        """
        server_id = self.server_list.item(self.server_list.currentRow(), 0).text()     
        
        self.server = Server()   
        self.server.saved.connect(self.on_server_saved)
        self.server.setData(self.server_model.item(server_id))     
        self.server.show() 
    
    @pyqtSlot()
    def on_action_copy_server_triggered(self):
        """
        Slot documentation goes here.
        """
        server_id = self.server_list.item(self.server_list.currentRow(), 0).text()  
        server_data = self.server_model.item(server_id)      
        server_data["id"] = None
        
        self.server = Server()   
        self.server.saved.connect(self.on_server_saved)
        self.server.setData(server_data)     
        self.server.show() 
    
    @pyqtSlot()
    def on_action_delete_server_triggered(self):
        """
        Slot documentation goes here.
        """
        index = self.server_list.currentRow()
        platform = self.server_list.item(index, 1).text()
        server = self.server_list.item(index, 2).text()
        answer = QtGui.QMessageBox.question(self, u"提示", u"确定删除 %s %s?" % (platform, server), QtGui.QMessageBox.Yes, QtGui.QMessageBox.No)
        if answer == QtGui.QMessageBox.Yes:            
            server_id = self.server_list.item(index, 0).text()      
            if self.server_model.delete(server_id):
                self.server_list.removeRow(index)
                QtGui.QMessageBox.information(self, u"提示", u"操作成功!")                
            else:
                QtGui.QMessageBox.critical(self, u"提示", u"操作失败!")
      
    def on_server_saved(self):
        self.refresh_server_list.clicked.emit(True)
        
    @pyqtSlot()
    def on_action_move2left_triggered(self):
        self.move2left.clicked.emit(True)
        
    def setServerListData(self, data):
        self.server_list.setRowCount(0)
        for row in data:
            if self.inSelectedServers(row["id"]) == False:
                index = self.server_list.rowCount()
                self.server_list.insertRow(index)
                self.server_list.setItem(index, 0, QTableWidgetItem(row["id"]))
                self.server_list.setItem(index, 1, QTableWidgetItem(row["platform"]))
                self.server_list.setItem(index, 2, QTableWidgetItem(row["server_name"]))
                self.server_list.setItem(index, 3, QTableWidgetItem(row["path"]))
            
    def setSelectedServersData(self, data):    
        self.selected_servers.setRowCount(0)
        for row in data:
            index = self.selected_servers.rowCount()
            self.selected_servers.insertRow(index)
            self.selected_servers.setItem(index, 0, QTableWidgetItem(row["id"]))
            self.selected_servers.setItem(index, 1, QTableWidgetItem(row["platform"]))
            self.selected_servers.setItem(index, 2, QTableWidgetItem(row["server_name"]))
            self.selected_servers.setItem(index, 3, QTableWidgetItem(row["path"]))
             
        self.refresh_server_list.clicked.emit(True)
                
    def inSelectedServers(self, server_id):
        for i in range(self.selected_servers.rowCount()):
            if self.selected_servers.item(i, 0).text() == server_id:
                return True
        return False