profile: upgrade the "profile" context manager to a full class So far we have been able to use a simple decorator for this. However using the current context manager makes the scope of the profiling in dispatch constrainted and the time frame to decide to enable profiling quite limited (using "maybeprofile") This is the first step toward the ability to enable the profiling from within the profiling scope. eg:: with maybeprofiling(ui) as profiler: ... bar.foo(): ... if options['profile']: profiler.start() ... fooz() ... My target usecase is adding support for "--profile" to alias definitions with effect. These are to be used with "profiling.output=blackbox" to gather data about operation that get slow from time to time (eg: pull being minutes instead of seconds from time to time). Of course, in such case, the scope of the profiling would be smaller since profiler would be started after running extensions 'reposetup' (and other potentially costly logic), but these are not relevant for my target usecase (multiple second commits, multiple tens of seconds pull). Currently adding '--profile' to a command through alias requires to re-spin a Mercurial binary (using "!$HG" in alias), which as a significant performance impact, especially in context where startup performance is being worked on... An alternative approach would be to stop using the context manager in dispatch and move back to a try/finally setup.

     1

'''

     2

Examples of useful python hooks for Mercurial.

     3

'''

     4

from __future__ import absolute_import

     5

from mercurial import (

     6

    patch,

     7

    util,

     8

)

     9

    10

def diffstat(ui, repo, **kwargs):

    11

    '''Example usage:

    12

    13

    [hooks]

    14

    commit.diffstat = python:/path/to/this/file.py:diffstat

    15

    changegroup.diffstat = python:/path/to/this/file.py:diffstat

    16

    '''

    17

    if kwargs.get('parent2'):

    18

        return

    19

    node = kwargs['node']

    20

    first = repo[node].p1().node()

    21

    if 'url' in kwargs:

    22

        last = repo['tip'].node()

    23

    else:

    24

        last = node

    25

    diff = patch.diff(repo, first, last)

    26

    ui.write(patch.diffstat(util.iterlines(diff)))