cb_table.py

from mysql_connection_manager import MysqlConnectionManager


class CBTable(object):
    def __init__(self, table_name):
        self.__manager = MysqlConnectionManager.getInstance()
        self.__table_name = table_name
        self._headers = []
        self.__rows = []
        self.__table_model = self.__manager.get_table(table_name)  # needs refactor
        if self.__table_model is not None:
            self._headers = map(lambda x: str(x), self.__table_model.__table__.columns.keys())
            if table_name == 'book':
                self._headers = [col.replace('author_id', 'author_marking') for col in self._headers]
            self.__rows = self.__manager.read_all(table_name)  # needs refactor

    def insert_and_commit(self, data):
        self.__manager.insert_and_commit(data)

    @property
    def headers(self):
        return self._headers

    @headers.setter
    def headers(self, val):
        self._headers = val

    @property
    def rows(self):
        return self.__manager.read_all(self.__table_name)

    @rows.setter
    def rows(self, val):
        self.__rows = val

    def exec_query(self, query):
        return self.__manager.execute_query(query)

    def filter_query(self, binary_expression):
        return self.__manager.session.query(self.__table_model).filter(binary_expression)

    @property
    def table_model(self):
        return self.__table_model