view contrib/python-hook-examples.py @ 38371:fb4813304c5f

py3: add `and None` to suppress return values of .write() calls

.write() calls don't return anything on Python2, so we need to make sure we
suppress that on py3 too.

This makes the test pass on Python 3.

Differential Revision: https://phab.mercurial-scm.org/D3793

'''
Examples of useful python hooks for Mercurial.
'''
from __future__ import absolute_import
from mercurial import (
    patch,
    util,
)

def diffstat(ui, repo, **kwargs):
    '''Example usage:

    [hooks]
    commit.diffstat = python:/path/to/this/file.py:diffstat
    changegroup.diffstat = python:/path/to/this/file.py:diffstat
    '''
    if kwargs.get('parent2'):
        return
    node = kwargs['node']
    first = repo[node].p1().node()
    if 'url' in kwargs:
        last = repo['tip'].node()
    else:
        last = node
    diff = patch.diff(repo, first, last)
    ui.write(patch.diffstat(util.iterlines(diff)))