HYTOS

# -*- coding: utf-8 -*-

# Form implementation generated from reading ui file 'ProjectDialog.ui'

#

# Created by: PyQt5 UI code generator 5.6

#

# WARNING! All changes made in this file will be lost!

from PyQt5 import QtCore, QtGui, QtWidgets

from PyQt5.QtWidgets import *

import os

from AppDocData import AppDocData

import Valve_Manual_UI

import math

def is_not_blank(s):

    return bool(s and s.strip())

class QValve_Manual(QDialog):

    def __init__(self):

        QDialog.__init__(self)

        self.ui = Valve_Manual_UI.Ui_Valve_ManualDialog()

        self.ui.setupUi(self)

        self._item = None

        self.ui.lineEdit_Pressure_Drop.setValidator(QtGui.QDoubleValidator(self.ui.lineEdit_Pressure_Drop))

        self.ui.lineEdit_Elevation.setValidator(QtGui.QDoubleValidator(self.ui.lineEdit_Elevation))

        self.initialize()

    def show_dialog(self, item):

        self.setWindowFlags(self.windowFlags() & ~QtCore.Qt.WindowContextHelpButtonHint)

        self._item = item

        self.ui.lineEdit_TagNo.setFocus()

        self.set_controls()

        self.load_data()

        return self.exec_()

    def initialize(self):

        self.ui.label_Img.setVisible(False)

    def set_controls(self):

        self.ui.label_Img.setVisible(True)

    def load_data(self):

        """ load tag no and nozzle data """

        tag_no = self._item.tag_no

        if tag_no != 'None' and is_not_blank(tag_no):

            self.ui.lineEdit_TagNo.setText(tag_no)

        app_doc_data = AppDocData.instance()

        drawing = app_doc_data.activeDrawing

        if drawing:

            for attr in drawing.attrs:

                if attr[0] == 'Units':

                    self.ui.label_PressureUnit.setText(attr[1]['Pressure'])

                    self.ui.label_ElevationUnit.setText(attr[1]['Length'])

        matches = [connector.data for connector in self._item.connectors if connector.data]

        if matches:

            pressure_drop = matches[0].pressure_drop

            if pressure_drop:

                self.ui.lineEdit_Pressure_Drop.setText(str(pressure_drop))

            elevation = matches[0].elevation

            if elevation:

                self.ui.lineEdit_Elevation.setText(str(elevation))

    def accept(self):

        """ set tag no and nozzle data """

        from EngineeringConnectorItem import NozzleData

        tag_no = self.ui.lineEdit_TagNo.text()

        if is_not_blank(tag_no):

            self._item.tag_no = tag_no

            for connector in self._item.connectors:

                if not connector.data:

                    connector.data = NozzleData()

                pressure_drop = self.ui.lineEdit_Pressure_Drop.text()

                if pressure_drop:

                    connector.data.pressure_drop = float(pressure_drop)

                else:

                    connector.data.pressure_drop = None

                elevation = self.ui.lineEdit_Elevation.text()

                if elevation:

                    connector.data.elevation = float(elevation)

                else:

                    connector.data.elevation = None

            QDialog.accept(self)

        else:

            QMessageBox.warning(self, self.tr('Notice'), self.tr('Please Input [Tag No.]'))

    def reject(self):

        QDialog.reject(self)