def _get_next_level(self, current): if self.direction == 'down': return Category.get_all_children(current, version=self.version) elif self.direction == 'up': return Category.get_all_parents(current, version=self.version) else: raise Exception("Unknown direction %s" % self.direction)