66756de12d3f76d48774434bb210cab39a9590c0,graphbrain/agents/txt_parser.py,TxtParser,input_file,#TxtParser#Any#,24

Before Change

    def input_file(self, file_name):
        pos = 0
        with open(file_name, "r") as f:
            text = f.read().replace("\n", " ")
            parses = self.parser.parse(text)
            for parse in parses:
                main_edge = parse["main_edge"]

After Change

    def input_file(self, file_name):
        pos = 0
        for paragraph in paragraphs(file_name):
            parses = self.parser.parse(paragraph)
            for parse in parses:
                main_edge = parse["main_edge"]

                // add main edge
                if main_edge:
                    self.hg.add_to_sequence(self.sequence, pos, main_edge)
                    self.edges += 1
                    pos += 1

                    // attach text to edge
                    self.hg.set_attribute(main_edge, "text", parse["text"])

                    // add extra edges
                    for edge in parse["extra_edges"]:
                        self.add(edge)

    def report(self):
        rep_str = ("edges found: {}".format(self.edges))
        return "{}\n\n{}".format(rep_str, super().report())

Instances