comparison hgext/blackbox.py @ 40762:37d6ee46a965

blackbox: extract global last logger to proxylogger class

So the blackboxlogger can be instantiated with a repo.

40761:691c68bc1222

    default=lambda: ['*'],
)
configitem('blackbox', 'date-format',
    default='%Y/%m/%d %H:%M:%S',
)

_lastlogger = None

def _openlogfile(ui, vfs):
    def rotate(oldpath, newpath):
        try:
            vfs.unlink(newpath)

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

class blackboxlogger(object):
    def __init__(self, ui):
        self._repo = None
        self._trackedevents = set(ui.configlist('blackbox', 'track'))

    def tracked(self, event):
        return b'*' in self._trackedevents or event in self._trackedevents

    def log(self, ui, event, msg, opts):
        global _lastlogger
        if self._bbvfs:
            _lastlogger = self
        elif _lastlogger and _lastlogger._bbvfs:
            # certain logger instances exist outside the context of
            # 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):
        default = ui.configdate('devel', 'default-date')
        date = dateutil.datestr(default, ui.config('blackbox', 'date-format'))
        user = procutil.getuser()
        pid = '%d' % procutil.getpid()
        rev = '(unknown)'

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

def uipopulate(ui):
    ui.setlogger(b'blackbox', blackboxlogger(ui))

def reposetup(ui, repo):
    # During 'hg pull' a httppeer repo is created to represent the remote repo.
    # It doesn't have a .hg directory to put a blackbox in, so we don't do
    # the blackbox setup for it.

    # Since blackbox.log is stored in the repo directory, the logger should be
    # instantiated per repository.
    logger = blackboxlogger(ui)
    ui.setlogger(b'blackbox', logger)
    if logger:
        logger.setrepo(repo)

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

    repo._wlockfreeprefix.add('blackbox.log')

@command('blackbox',
    [('l', 'limit', 10, _('the number of events to show')),

40762:37d6ee46a965

    default=lambda: ['*'],
)
configitem('blackbox', 'date-format',
    default='%Y/%m/%d %H:%M:%S',
)

def _openlogfile(ui, vfs):
    def rotate(oldpath, newpath):
        try:
            vfs.unlink(newpath)

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

class proxylogger(object):
    """Forward log events to another logger to be set later"""

    def __init__(self):
        self.logger = None

    def tracked(self, event):
        return self.logger is not None and self.logger.tracked(event)

    def log(self, ui, event, msg, opts):
        assert self.logger is not None
        self.logger.log(ui, event, msg, opts)

_lastlogger = proxylogger()

class blackboxlogger(object):
    def __init__(self, ui):
        self._repo = None
        self._trackedevents = set(ui.configlist('blackbox', 'track'))

    def tracked(self, event):
        return b'*' in self._trackedevents or event in self._trackedevents

    def log(self, ui, event, msg, opts):
        if self._bbvfs:
            _lastlogger.logger = self
        else:
            return

        default = ui.configdate('devel', 'default-date')
        date = dateutil.datestr(default, ui.config('blackbox', 'date-format'))
        user = procutil.getuser()
        pid = '%d' % procutil.getpid()
        rev = '(unknown)'

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

def uipopulate(ui):
    ui.setlogger(b'blackbox', _lastlogger)

def reposetup(ui, repo):
    # During 'hg pull' a httppeer repo is created to represent the remote repo.
    # It doesn't have a .hg directory to put a blackbox in, so we don't do
    # the blackbox setup for it.

    # Since blackbox.log is stored in the repo directory, the logger should be
    # instantiated per repository.
    logger = blackboxlogger(ui)
    ui.setlogger(b'blackbox', logger)
    logger.setrepo(repo)

    # Set _lastlogger even if ui.log is not called. This gives blackbox a
    # fallback place to log
    if _lastlogger.logger is None:
        _lastlogger.logger = logger

    repo._wlockfreeprefix.add('blackbox.log')

@command('blackbox',
    [('l', 'limit', 10, _('the number of events to show')),