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 Project import Project

from AppDocData import AppDocData

import Strainer_T_UI

import math

class QStrainer_T(QDialog):

    def __init__(self):

        QDialog.__init__(self)

        self.ui = Strainer_T_UI.Ui_Strainer_T_Dialog()

        self.ui.setupUi(self)

        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 showDialog(self, item):

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

        self._item = item

        self.ui.lineEdit_TagNo.setFocus()

        self.set_controls(item)

        self.load_data()

        self.exec_()

    def initialize(self):

        self.ui.label_Img.setVisible(False)

    def set_controls(self, item):    

        self.ui.label_Img.setVisible(True)

    def load_data(self):

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

        from Drawing import Drawing

        appDocData = AppDocData.instance()

        drawing = appDocData.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.nozzle_data for connector in self._item.connectors if connector.nozzle_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

        for connector in self._item.connectors:

            if not connector.nozzle_data: 

                connector.nozzle_data = NozzleData()

            pressure_drop = self.ui.lineEdit_Pressure_Drop.text()

            if pressure_drop:

                connector.nozzle_data.pressure_drop = float(pressure_drop)    

            else: 

                connector.nozzle_data.pressure_drop = None

            elevation = self.ui.lineEdit_Elevation.text()

            if elevation:

                connector.nozzle_data.elevation = float(elevation)  

            else:

                connector.nozzle_data.elevation = None

        QDialog.accept(self)

    def reject(self):

        QDialog.reject(self)