def setUp(self): self.clear_cache() try: self.hbase_tables_created = True for hbase_model_class in HBaseModel.__subclasses__(): hbase_model_class.create_table() except Exception: self.tearDown() raise
def get_model_class(cls, model_class_name): for subclass in HBaseModel.__subclasses__(): if subclass.__name__ == model_class_name: return subclass raise Exception('HBaseModel {} not found'.format(model_class_name))
def tearDown(self): if not self.hbase_tables_created: return for hbase_model_class in HBaseModel.__subclasses__(): hbase_model_class.drop_table()