mercurial/lsprofcalltree.py

demandimport: fix a crash in LazyFinder.__delattr__

I was tinkering with `with hgdemandimport.deactivated()` wrapped around loading
the keyring module, and got spew that seemed to be confirmed by PyCharm.  But I
can't believe we haven't seen this before (and phabricator uses the same
pattern):

    ** Unknown exception encountered with possibly-broken third-party extension "mercurial_keyring" 1.4.3 (keyring 23.11.0, backend unknown)
    ** which supports versions unknown of Mercurial.
    ** Please disable "mercurial_keyring" and try your action again.
    ** If that fixes the bug please report it to https://foss.heptapod.net/mercurial/mercurial_keyring/issues
    ** Python 3.9.15 (main, Oct 13 2022, 04:28:25) [GCC 7.5.0]
    ** Mercurial Distributed SCM (version 6.3.1)
    ** Extensions loaded: absorb, attorc 20220315, blackbox, eol, extdiff, fastannotate, lfs, mercurial_keyring 1.4.3 (keyring 23.11.0, backend unknown), phabblocker 20220315, phabricator 20220315, purge, rebase, schemes, share, show, strip, uncommit
    Traceback (most recent call last):
      File "/usr/local/bin/hg", line 59, in <module>
        dispatch.run()
      File "/usr/local/lib/python3.9/site-packages/mercurial/dispatch.py", line 143, in run
        status = dispatch(req)
      File "/usr/local/lib/python3.9/site-packages/mercurial/dispatch.py", line 232, in dispatch
        status = _rundispatch(req)
      File "/usr/local/lib/python3.9/site-packages/mercurial/dispatch.py", line 276, in _rundispatch
        ret = _runcatch(req) or 0
      File "/usr/local/lib/python3.9/site-packages/mercurial/dispatch.py", line 451, in _runcatch
        return _callcatch(ui, _runcatchfunc)
      File "/usr/local/lib/python3.9/site-packages/mercurial/dispatch.py", line 461, in _callcatch
        return scmutil.callcatch(ui, func)
      File "/usr/local/lib/python3.9/site-packages/mercurial/scmutil.py", line 153, in callcatch
        return func()
      File "/usr/local/lib/python3.9/site-packages/mercurial/dispatch.py", line 441, in _runcatchfunc
        return _dispatch(req)
      File "/usr/local/lib/python3.9/site-packages/mercurial/dispatch.py", line 1265, in _dispatch
        return runcommand(
      File "/usr/local/lib/python3.9/site-packages/mercurial/dispatch.py", line 899, in runcommand
        ret = _runcommand(ui, options, cmd, d)
      File "/usr/local/lib/python3.9/site-packages/mercurial/dispatch.py", line 1277, in _runcommand
        return cmdfunc()
      File "/usr/local/lib/python3.9/site-packages/mercurial/dispatch.py", line 1263, in <lambda>
        d = lambda: util.checksignature(func)(ui, *args, **strcmdopt)
      File "/usr/local/lib/python3.9/site-packages/mercurial/util.py", line 1880, in check
        return func(*args, **kwargs)
      File "/root/mercurial_keyring/mercurial_keyring/mercurial_keyring.py", line 962, in cmd_keyring_check
        user, pwd, source, final_url = handler.get_credentials(
      File "/root/mercurial_keyring/mercurial_keyring/mercurial_keyring.py", line 497, in get_credentials
        keyring_pwd = password_store.get_http_password(keyring_url, actual_user)
      File "/root/mercurial_keyring/mercurial_keyring/mercurial_keyring.py", line 287, in get_http_password
        return self._read_password_from_keyring(
      File "/root/mercurial_keyring/mercurial_keyring/mercurial_keyring.py", line 335, in _read_password_from_keyring
        keyring = import_keyring()
>> `with hgdemandimport.deactivated()` inserted here
      File "/root/mercurial_keyring/mercurial_keyring/mercurial_keyring.py", line 120, in import_keyring
        return _import_keyring()
      File "/root/mercurial_keyring/mercurial_keyring/mercurial_keyring.py", line 133, in _import_keyring
        mod, was_imported_now = meu.direct_import_ext(
      File "/usr/lib/python3.9/site-packages/mercurial_extension_utils.py", line 1381, in direct_import_ext
        __import__(module_name)
      File "<frozen importlib._bootstrap>", line 1007, in _find_and_load
      File "<frozen importlib._bootstrap>", line 986, in _find_and_load_unlocked
      File "<frozen importlib._bootstrap>", line 680, in _load_unlocked
      File "/usr/local/lib/python3.9/site-packages/hgdemandimport/demandimportpy3.py", line 46, in exec_module
        self.loader.exec_module(module)
      File "/usr/lib/python3.9/site-packages/keyring/__init__.py", line 1, in <module>
        from .core import (
      File "<frozen importlib._bootstrap>", line 1007, in _find_and_load
      File "<frozen importlib._bootstrap>", line 986, in _find_and_load_unlocked
      File "<frozen importlib._bootstrap>", line 680, in _load_unlocked
      File "/usr/local/lib/python3.9/site-packages/hgdemandimport/demandimportpy3.py", line 46, in exec_module
        self.loader.exec_module(module)
      File "/usr/lib/python3.9/site-packages/keyring/core.py", line 11, in <module>
        from . import backend, credentials
      File "<frozen importlib._bootstrap>", line 1007, in _find_and_load
      File "<frozen importlib._bootstrap>", line 986, in _find_and_load_unlocked
      File "<frozen importlib._bootstrap>", line 680, in _load_unlocked
      File "/usr/local/lib/python3.9/site-packages/hgdemandimport/demandimportpy3.py", line 46, in exec_module
        self.loader.exec_module(module)
      File "/usr/lib/python3.9/site-packages/keyring/backend.py", line 13, in <module>
        from .py312compat import metadata
      File "<frozen importlib._bootstrap>", line 1007, in _find_and_load
      File "<frozen importlib._bootstrap>", line 986, in _find_and_load_unlocked
      File "<frozen importlib._bootstrap>", line 680, in _load_unlocked
      File "/usr/local/lib/python3.9/site-packages/hgdemandimport/demandimportpy3.py", line 46, in exec_module
        self.loader.exec_module(module)
      File "/usr/lib/python3.9/site-packages/keyring/py312compat.py", line 10, in <module>
        import importlib_metadata as metadata  # type: ignore
      File "<frozen importlib._bootstrap>", line 1007, in _find_and_load
      File "<frozen importlib._bootstrap>", line 986, in _find_and_load_unlocked
      File "<frozen importlib._bootstrap>", line 680, in _load_unlocked
      File "/usr/local/lib/python3.9/site-packages/hgdemandimport/demandimportpy3.py", line 46, in exec_module
        self.loader.exec_module(module)
      File "/usr/lib/python3.9/site-packages/importlib_metadata/__init__.py", line 715, in <module>
        class MetadataPathFinder(NullFinder, DistributionFinder):
      File "/usr/lib/python3.9/site-packages/importlib_metadata/_compat.py", line 24, in install
        disable_stdlib_finder()
      File "/usr/lib/python3.9/site-packages/importlib_metadata/_compat.py", line 43, in disable_stdlib_finder
        del finder.find_distributions
      File "/usr/local/lib/python3.9/site-packages/hgdemandimport/demandimportpy3.py", line 88, in __delattr__
        return delattr(object.__getattribute__(self, "_finder"))
    TypeError: delattr expected 2 arguments, got 1

"""
lsprofcalltree.py - lsprof output which is readable by kcachegrind

Authors:
    * David Allouche <david <at> allouche.net>
    * Jp Calderone & Itamar Shtull-Trauring
    * Johan Dahlin

This software may be used and distributed according to the terms
of the GNU General Public License, incorporated herein by reference.
"""


from . import pycompat


def label(code):
    if isinstance(code, str):
        # built-in functions ('~' sorts at the end)
        return b'~' + pycompat.sysbytes(code)
    else:
        return b'%s %s:%d' % (
            pycompat.sysbytes(code.co_name),
            pycompat.sysbytes(code.co_filename),
            code.co_firstlineno,
        )


class KCacheGrind:
    def __init__(self, profiler):
        self.data = profiler.getstats()
        self.out_file = None

    def output(self, out_file):
        self.out_file = out_file
        out_file.write(b'events: Ticks\n')
        self._print_summary()
        for entry in self.data:
            self._entry(entry)

    def _print_summary(self):
        max_cost = 0
        for entry in self.data:
            totaltime = int(entry.totaltime * 1000)
            max_cost = max(max_cost, totaltime)
        self.out_file.write(b'summary: %d\n' % max_cost)

    def _entry(self, entry):
        out_file = self.out_file

        code = entry.code
        if isinstance(code, str):
            out_file.write(b'fi=~\n')
        else:
            out_file.write(b'fi=%s\n' % pycompat.sysbytes(code.co_filename))

        out_file.write(b'fn=%s\n' % label(code))

        inlinetime = int(entry.inlinetime * 1000)
        if isinstance(code, str):
            out_file.write(b'0 %d\n' % inlinetime)
        else:
            out_file.write(b'%d %d\n' % (code.co_firstlineno, inlinetime))

        # recursive calls are counted in entry.calls
        if entry.calls:
            calls = entry.calls
        else:
            calls = []

        if isinstance(code, str):
            lineno = 0
        else:
            lineno = code.co_firstlineno

        for subentry in calls:
            self._subentry(lineno, subentry)

        out_file.write(b'\n')

    def _subentry(self, lineno, subentry):
        out_file = self.out_file
        code = subentry.code
        out_file.write(b'cfn=%s\n' % label(code))
        if isinstance(code, str):
            out_file.write(b'cfi=~\n')
            out_file.write(b'calls=%d 0\n' % subentry.callcount)
        else:
            out_file.write(b'cfi=%s\n' % pycompat.sysbytes(code.co_filename))
            out_file.write(
                b'calls=%d %d\n' % (subentry.callcount, code.co_firstlineno)
            )

        totaltime = int(subentry.totaltime * 1000)
        out_file.write(b'%d %d\n' % (lineno, totaltime))