HYTOS

# coding: utf-8

""" This is Data Transfer dialog module """

import os

import sys

from PyQt5.QtCore import *

from PyQt5.QtGui import *

from PyQt5.QtWidgets import *

from AppDocData import AppDocData, MessageType

sys.path.insert(0, os.path.dirname(os.path.realpath(__file__)) + '\\UI')

import DataTransfer_UI

class QDataTransferDialog(QDialog):

    """ This Data Trasnfer dialog class """

    def __init__(self, parent):

        QDialog.__init__(self, parent)

        self.ui = DataTransfer_UI.Ui_DataTransferDialog()

        self.ui.setupUi(self)

        self.ui.lineEditPassword.setEchoMode(QLineEdit.Password) 

        self.ui.pushButtonTestConnection.clicked.connect(self.on_test_connection_clicked)

        self.ui.treeWidgetDrawingList.setHeaderHidden(False)

        self.ui.treeWidgetDrawingList.header().setStretchLastSection(False)

        self.ui.treeWidgetDrawingList.setHeaderLabels([self.tr('Name'), self.tr('DateTime')])

        self.ui.treeWidgetDrawingList.header().setSectionResizeMode(0, QHeaderView.ResizeToContents)

        self.ui.treeWidgetDrawingList.header().setSectionResizeMode(1, QHeaderView.ResizeToContents)

        self.load_data()

    def load_data(self):

        """ load mssql server information and drawing list """

        from AppDocData import AppDocData

        configs = AppDocData.instance().getConfigs('Data Transfer')

        for config in configs:

            if 'Server' == config.key:

                self.ui.lineEditServer.setText(config.value)

            elif 'User' == config.key:

                self.ui.lineEditUser.setText(config.value)

            elif 'Password' == config.key:

                self.ui.lineEditPassword.setText(config.value)

        self.load_drawing_list()

    def load_drawing_list(self):

        """ load drawing list """

        try:

            app_doc_data = AppDocData.instance()

            drawings = app_doc_data.getDrawings()

            self.ui.treeWidgetDrawingList.clear()

            self.ui.treeWidgetDrawingList.root = QTreeWidgetItem(self.ui.treeWidgetDrawingList, [self.tr('P&ID Drawings'), ''])

            self.ui.treeWidgetDrawingList.root.setFlags(self.ui.treeWidgetDrawingList.root.flags() | Qt.ItemIsTristate | Qt.ItemIsUserCheckable)

            self.ui.treeWidgetDrawingList.root.setCheckState(0, Qt.Unchecked)

            for drawing in drawings:

                item = QTreeWidgetItem(self.ui.treeWidgetDrawingList.root, [drawing.name, drawing.datetime])

                item.setIcon(0, QIcon(':newPrefix/image.png'))

                item.setFlags(item.flags() | Qt.ItemIsUserCheckable)

                item.setCheckState(0, Qt.Unchecked)

                item.setData(0, Qt.UserRole, drawing)

            self.ui.treeWidgetDrawingList.root.setText(0, self.tr('P&ID Drawings')+'({})'.format(self.ui.treeWidgetDrawingList.root.childCount()))

            self.ui.treeWidgetDrawingList.expandItem(self.ui.treeWidgetDrawingList.root)

            self.ui.treeWidgetDrawingList.setSortingEnabled(True)

            self.ui.treeWidgetDrawingList.root.sortChildren(0, Qt.AscendingOrder)

            self.ui.treeWidgetDrawingList.resizeColumnToContents(0)

        except Exception as ex:

            from App import App

            message = 'error occurred({}) in {}:{}'.format(ex, sys.exc_info()[-1].tb_frame.f_code.co_filename, sys.exc_info()[-1].tb_lineno)

            App.mainWnd().addMessage.emit(MessageType.Error, message)

    def on_test_connection_clicked(self):

        """ test if connection is success or not """

        from AppDocData import AppDocData

        from AppDatabase import AppDatabase

        server = self.ui.lineEditServer.text()

        user = self.ui.lineEditUser.text()

        password = self.ui.lineEditPassword.text()

        if server and user and password:

            app_doc_data = AppDocData.instance()

            database = AppDatabase('MSSQL', self.ui.lineEditServer.text(), self.ui.lineEditUser.text(), self.ui.lineEditPassword.text(), db_path=app_doc_data.project.getName())

            try:

                conn = database.connect()

                with conn:

                    QMessageBox.information(self, self.tr('Information'), self.tr('Test connection is success'))

            except Exception as ex:

                mb = QMessageBox()

                mb.setIcon(QMessageBox.Critical)

                mb.setWindowTitle(self.tr('Error'))

                mb.setText('{}'.format(ex))

                mb.exec_()

    def accept(self):

        """ save sql information to database """

        from AppDocData import Config

        from AppDatabase import AppDatabase

        try:

            app_doc_data = AppDocData.instance()

            configs = []

            server = self.ui.lineEditServer.text()

            user = self.ui.lineEditUser.text()

            password = self.ui.lineEditPassword.text()

            configs.append(Config('Data Transfer', 'Server', server))

            configs.append(Config('Data Transfer', 'User', user))

            configs.append(Config('Data Transfer', 'Password', password))

            AppDocData.instance().saveConfigs(configs)

            database = AppDatabase('MSSQL', server, user, password, db_path=app_doc_data.project.getName())

            checked = [self.ui.treeWidgetDrawingList.root.child(i) for i in range(self.ui.treeWidgetDrawingList.root.childCount()) \

                if self.ui.treeWidgetDrawingList.root.child(i).checkState(0) == Qt.Checked]

            self.ui.progressBarDataTransfer.setMaximum(len(checked))

            for item in checked:

                drawing = item.data(0, Qt.UserRole)

                drawing.transfer(database)

                self.ui.progressBarDataTransfer.setValue(self.ui.progressBarDataTransfer.value() + 1)

                QApplication.processEvents()

            QDialog.accept(self)

        except Exception as ex:

            from App import App

            message = 'error occurred({}) in {}:{}'.format(ex, sys.exc_info()[-1].tb_frame.f_code.co_filename, sys.exc_info()[-1].tb_lineno)

            App.mainWnd().addMessage.emit(MessageType.Error, message)