def get_model(cls, table): if isinstance(table, int): obj = cls.get(cls.c.id == table) else: obj = cls.get(cls.c.table_name == table) if obj: return _get_model(obj.table_name)
def get_model(cls, table): if isinstance(table, int): obj = cls.get(cls.c.id == table) else: obj = cls.get(cls.c.table_name == table) if obj: return _get_model(obj.table_name) else: raise NotFound("Can't find model of table [%s]" % str(table))