HYTOS

# coding: utf-8

import os

import sys

import sqlite3

from SingletonInstance import SingletonInstane

'''

    @brief      HMB data

'''

class HMBData:

    def __init__(self, uid=None):

        self._uid = uid

        self._stream_no = None

        self._name = None

        self._unit = None

        self._value = None

        self.isDeleted = False

    '''

        @author humkyung

        @date   2018.07.12

    '''

    @property

    def uid(self):

        return self._uid

    '''

        @author humkyung

        @date   2018.07.12

    '''

    @uid.setter

    def uid(self, value):

        self._uid = value

    '''

        @author humkyung

        @date   2018.07.12

    '''

    @property

    def streamNo(self):

        return self._stream_no

    '''

        @author humkyung

        @date   2018.07.12

    '''

    @streamNo.setter

    def streamNo(self, value):

        self._stream_no = value

    '''

        @author humkyung

        @date   2018.07.12

    '''

    @property

    def name(self):

        return self._name

    '''

        @author humkyung

        @date   2018.07.12

    '''

    @name.setter

    def name(self, value):

        self._name = value

    '''

        @author humkyung

        @date   2018.07.12

    '''

    @property

    def unit(self):

        return self._unit

    '''

        @author humkyung

        @date   2018.07.12

    '''

    @unit.setter

    def unit(self, value):

        self._unit = value

    '''

        @author humkyung

        @date   2018.07.12

    '''

    @property

    def value(self):

        return self._value

    '''

        @author humkyung

        @date   2018.07.12

    '''

    @value.setter

    def value(self, value):

        self._value = value

    '''

        @brief      create hmb data from database record

        @author     humkyung

        @date       2018.07.12

    '''

    @staticmethod

    def fromRow(row):

        hmb = HMBData()

        hmb._uid = row[0]

        hmb._stream_no = row[1]

        hmb._name = row[2]

        hmb._unit = row[3]

        hmb._value = row[4]

        return hmb

class HMBTable:

    '''

        @brief  constructor

        @author humkyung

        @date   2018.07.12

    '''

    def __init__(self):

        self._hmbs = None

    '''

        @brief      load hmb data from database

        @author     humkyung

        @date       2018.07.12

    '''

    def loadData(self):

        from AppDocData import AppDocData

        try:

            if self._hmbs is None:

                self._hmbs = []

                appDocData = AppDocData.instance()

                dbPath = os.path.join(appDocData.getCurrentProject().getDbFilePath() , 'ITI_PID.db')

                conn = sqlite3.connect(dbPath)

                cursor = conn.cursor()

                sql = 'SELECT UID,STREAM_NO,NAME,UNIT,VALUE FROM HMB ORDER BY STREAM_NO'

                cursor.execute(sql)

                rows = cursor.fetchall()

                for row in rows:

                    hmb = HMBData.fromRow(row)

                    self._hmbs.append(hmb)

        # Catch the exception

        except Exception as ex:

            # Roll back any change if something goes wrong

            conn.rollback()

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

        finally:

            conn.close()

        return self._hmbs

    '''

        @brief      save hmb data 

        @author     humkyung

        @date       2018.07.12

    '''

    def saveData(self):

        import uuid

        from AppDocData import AppDocData

        try:

            if self._hmbs is not None:

                appDocData = AppDocData.instance()

                # Creates or opens a file called mydb with a SQLite3 DB

                dbPath = os.path.join(appDocData.getCurrentProject().getDbFilePath(), "ITI_PID.db")

                conn = sqlite3.connect(dbPath)

                # Get a cursor object

                cursor = conn.cursor()

                for data in self._hmbs:

                    if data.isDeleted == False:

                        if data.uid is None:

                            sql = "insert or replace into HMB values(?,?,?,?,?)"

                            param = (str(uuid.uuid4()), data.streamNo, data.name, data.unit, data.value)

                            cursor.execute(sql, param)

                        else:

                            sql = "update HMB set STREAM_NO=?,NAME=?,UNIT=?,VALUE=? where UID=?"

                            param = (data.streamNo, data.name, data.unit, data.value, data.uid)

                            cursor.execute(sql, param)

                    else:

                        sql = "delete from HMB where uid=?"

                        param = (data.uid,)

                        cursor.execute(sql, param)

                conn.commit()

        # Catch the exception

        except Exception as ex:

            # Roll back any change if something goes wrong

            conn.rollback()

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

        finally:

            # Close the db connection

            conn.close()

    '''

        @brief      return stream no collection which are not duplicated 

        @author     humkyung

        @date       2018.07.12

    '''

    def streamNos(self):

        return set([hmb.streamNo for hmb in self._hmbs if hmb.isDeleted == False]) if self._hmbs is not None else {}

    '''

        @brief      return given index's data

        @author     humkyung

        @date       2018.07.12

    '''

    def dataOfStreamNo(self, streamNo):

        return [hmb for hmb in self._hmbs if hmb.streamNo == streamNo and hmb.isDeleted == False] if self._hmbs is not None else None

    '''

        @brief      append hmb data

        @author     humkyung

        @date       2018.07.12

    '''

    def append(self, hmbData):

        if self._hmbs is None:

            self._hmbs = []

        self._hmbs.append(hmbData)

    '''

        @brief      reset

        @author     humkyung

        @date       2018.07.12

    '''

    def reset(self):

        if self._hmbs is not None:

            del self._hmbs

        self._hmbs = None