def get_sorted_story_edges()

in clutrr/relations/puzzle.py [0:0]

• 11 lines of code
• 7 McCabe index (conditional complexity)

    def get_sorted_story_edges(self, stype='story'):
        """
        Overlay changed node ids onto story edges
        :param stype:
        :return:
        """
        if stype == 'story':
            edges_tc = copy.copy(self.story)
        elif stype == 'fact':
            edges_tc = copy.copy([fact.fact_edges for fact in self.facts])
            edges_tc = [y for x in edges_tc for y in x]
        else:
            raise NotImplementedError("stype not implemented")
        edge_keys_changed_id = [(self.story_sort_dict[key[0]],
                                 self.story_sort_dict[key[1]]) for key in edges_tc]
        return edge_keys_changed_id