comparison hgext/blackbox.py @ 40644:a9393d7600f3

blackbox: extract logger class from ui wrapper

This moves most functions to new blackboxlogger class. The ui wrapper will
be removed later.

40643:56694b4d41b0

                           newpath='%s.%d' % (path, i))
                rotate(oldpath=path,
                       newpath=maxfiles > 0 and path + '.1')
    return vfs(name, 'a')

def wrapui(ui):
    class blackboxui(ui.__class__):
        @property
        def _bbvfs(self):
            vfs = None
            repo = getattr(self, '_bbrepo', None)
            if repo:
                vfs = repo.vfs
                if not vfs.isdir('.'):
                    vfs = None
            return vfs

        @util.propertycache
        def track(self):
            return self.configlist('blackbox', 'track')

        def debug(self, *msg, **opts):
            super(blackboxui, self).debug(*msg, **opts)
            if self.debugflag:
                self.log('debug', '%s', ''.join(msg))

        def log(self, event, *msg, **opts):
            global _lastlogger
            super(blackboxui, self).log(event, *msg, **opts)

            if not '*' in self.track and not event in self.track:
                return

            if self._bbvfs:
                _lastlogger = self

                # a repo, so just default to the last blackbox logger that
                # was seen.
                pass
            else:
                return
            _lastlogger._log(self, event, msg, opts)

        def _log(self, ui, event, msg, opts):
            if getattr(self, '_bbinlog', False):
                # recursion and failure guard
                return

                self._bbinlog = False

        def setrepo(self, repo):
            self._bbrepo = repo

    ui.__class__ = blackboxui
    uimod.ui = blackboxui

def uisetup(ui):
    wrapui(ui)

    # It doesn't have a .hg directory to put a blackbox in, so we don't do
    # the blackbox setup for it.
    if not repo.local():
        return

    if util.safehasattr(ui, 'setrepo'):
        logger = ui
        logger.setrepo(repo)

        # Set _lastlogger even if ui.log is not called. This gives blackbox a
        # fallback place to log.
        global _lastlogger

40644:a9393d7600f3

                           newpath='%s.%d' % (path, i))
                rotate(oldpath=path,
                       newpath=maxfiles > 0 and path + '.1')
    return vfs(name, 'a')

if True:
    class blackboxlogger(object):
        def __init__(self, ui):
            self.track = ui.configlist('blackbox', 'track')

        @property
        def _bbvfs(self):
            vfs = None
            repo = getattr(self, '_bbrepo', None)
            if repo:
                vfs = repo.vfs
                if not vfs.isdir('.'):
                    vfs = None
            return vfs

        def log(self, ui, event, msg, opts):
            global _lastlogger
            if not '*' in self.track and not event in self.track:
                return

            if self._bbvfs:
                _lastlogger = self

                # a repo, so just default to the last blackbox logger that
                # was seen.
                pass
            else:
                return
            _lastlogger._log(ui, event, msg, opts)

        def _log(self, ui, event, msg, opts):
            if getattr(self, '_bbinlog', False):
                # recursion and failure guard
                return

                self._bbinlog = False

        def setrepo(self, repo):
            self._bbrepo = repo

def wrapui(ui):
    class blackboxui(ui.__class__):
        def __init__(self, src=None):
            super(blackboxui, self).__init__(src)
            if src and r'_bblogger' in src.__dict__:
                self._bblogger = src._bblogger

        # trick to initialize logger after configuration is loaded, which
        # can be replaced later with blackboxlogger(ui) in uisetup(), where
        # both user and repo configurations should be available.
        @util.propertycache
        def _bblogger(self):
            return blackboxlogger(self)

        def debug(self, *msg, **opts):
            super(blackboxui, self).debug(*msg, **opts)
            if self.debugflag:
                self.log('debug', '%s', ''.join(msg))

        def log(self, event, *msg, **opts):
            super(blackboxui, self).log(event, *msg, **opts)
            self._bblogger.log(self, event, msg, opts)

    ui.__class__ = blackboxui
    uimod.ui = blackboxui

def uisetup(ui):
    wrapui(ui)

    # It doesn't have a .hg directory to put a blackbox in, so we don't do
    # the blackbox setup for it.
    if not repo.local():
        return

    logger = getattr(ui, '_bblogger', None)
    if logger:
        logger.setrepo(repo)

        # Set _lastlogger even if ui.log is not called. This gives blackbox a
        # fallback place to log.
        global _lastlogger