comparison mercurial/statprof.py @ 49578:aab3d4c05720 stable

profile: prevent a crash when line number is unknown

For some unknown reason, `self.lineno` can be None. The previous code crashed in
such case, we now ignore the case, as we do for other error in this function.

We also fallback to using `-1` in the output when this lack of line number
makes it to the display code.

The reason of unknown line-numbers is… unknown.

49576:d12446766a35

            cls.cache[k] = v
            return v

    def getsource(self, length):
        if self.source is None:
            lineno = self.lineno - 1
            try:
                with open(self.path, b'rb') as fp:
                    for i, line in enumerate(fp):
                        if i == lineno:
                            self.source = line.strip()
                            break

            codepattern = b'%s%s %d: %s%s'
            codestring = codepattern % (
                prefix,
                b'line'.rjust(spacing_len),
                site.lineno,
                b''.ljust(max(0, 4 - len(str(site.lineno)))),
                site.getsource(30),
            )

            finalstring = liststring + codestring

49578:aab3d4c05720

            cls.cache[k] = v
            return v

    def getsource(self, length):
        if self.source is None:
            try:
                lineno = self.lineno - 1  # lineno can be None
                with open(self.path, b'rb') as fp:
                    for i, line in enumerate(fp):
                        if i == lineno:
                            self.source = line.strip()
                            break

            codepattern = b'%s%s %d: %s%s'
            codestring = codepattern % (
                prefix,
                b'line'.rjust(spacing_len),
                site.lineno if site.lineno is not None else -1,
                b''.ljust(max(0, 4 - len(str(site.lineno)))),
                site.getsource(30),
            )

            finalstring = liststring + codestring