HYTOS

    def add_column(self, column_name, column_type, table_name):

        try:

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

            conn = sqlite3.connect(self.activeDrawing.path)

            conn.execute('PRAGMA foreign_keys = ON')

            # Get a cursor object

            cursor = conn.cursor()

            sql = "Alter Table " + table_name + " Add Column '" + column_name + "' " + column_type

            cursor.execute(sql)

            conn.commit()

        # Catch the exception

        except Exception as ex:

            from App import App

            # Roll back any change if something goes wrong

            conn.rollback()

            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)

        finally:

            # Close the db connection

            conn.close()

    def get_column_names(self, table_name):

        res = []

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

        conn = sqlite3.connect(self.activeDrawing.path)

        with conn:

            try:

                # Get a cursor object

                cursor = conn.cursor()

                sql = "PRAGMA Table_Info ('" + table_name + "')"

                cursor.execute(sql)

                rows = cursor.fetchall()

                for row in rows:

                    res.append(row[1])

            # Catch the exception

            except Exception as ex:

                from App import App

                # Roll back any change if something goes wrong

                conn.rollback()

                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)

        return res

            find_column_names = ['Homo_Den', 'Homo_Vel']

            column_names = app_doc_data.get_column_names('PressureVariation')

            for column_name in find_column_names:

                if column_name not in column_names:

                    app_doc_data.add_column(column_name, 'REAL', 'PressureVariation')