db/tbs_db.py

from  utils.load_module import load_module
    
class TbsDb():

    def __init__(self, db_module_filename, class_name):
        self.db_module = load_module(db_module_filename, class_name)

    def get_list(self, query):
        response = self.execute_query(query)
        if response['success']==False:
            return response
        return response['result'][0]
        # return self.execute_query(query)
    
    def add_item(self, query):
        return self.execute_query(query)
    
    def get_item(self, query):
        return self.execute_query(query)
    
    def execute_query(self, query):
        return self.db_module.execute_query(query)