comparison setup.py @ 33114:8b20338b989e

setup: prefer using the system hg to interact with the local repository

Add a findhg() function that tries to be smarter about figuring out how to run
hg for examining the local repository.  It first tries running "hg" from the
user's PATH, with the default HGRCPATH settings intact, but with HGPLAIN
enabled.  This will generally use the same version of mercurial and the same
settings used to originally clone the repository, and should have a higher
chance of working successfully than trying to run the hg script from the local
repository.  If that fails findhg() falls back to the existing behavior of
running the local hg script.

33113:fc290a39590d

                         stderr=subprocess.PIPE, env=env)
    out, err = p.communicate()
    return p.returncode, out, err

class hgcommand(object):
    def __init__(self):
        self.cmd = [sys.executable, 'hg']
        self.env = gethgenv()

    def run(self, args):
        cmd = self.cmd + args
        returncode, out, err = runcmd(cmd, self.env)
        # If root is executing setup.py, but the repository is owned by

            printf("stderr from '%s':" % (' '.join(cmd)), file=sys.stderr)
            printf(b'\n'.join([b'  ' + e for e in err]), file=sys.stderr)
            return ''
        return out

def gethgenv():
    # Execute hg out of this directory with a custom environment which takes
    # care to not use any hgrc files and do no localization.
    env = {'HGMODULEPOLICY': 'py',
           'HGRCPATH': '',
           'LANGUAGE': 'C',

        env['SystemRoot'] = os.environ['SystemRoot']

version = ''

if os.path.isdir('.hg'):
    hg = hgcommand()
    cmd = ['log', '-r', '.', '--template', '{tags}\n']
    numerictags = [t for t in hg.run(cmd).split() if t[0:1].isdigit()]
    hgid = hg.run(['id', '-i']).strip()
    if not hgid:
        # Bail out if hg is having problems interacting with this repository,

                f.write('# empty\n')

        # here no extension enabled, disabled() lists up everything
        code = ('import pprint; from mercurial import extensions; '
                'pprint.pprint(extensions.disabled())')
        returncode, out, err = runcmd([sys.executable, '-c', code], gethgenv())
        if err or returncode != 0:
            raise DistutilsExecError(err)

        with open(self._indexfilename, 'w') as f:
            f.write('# this file is autogenerated by setup.py\n')

33114:8b20338b989e

                         stderr=subprocess.PIPE, env=env)
    out, err = p.communicate()
    return p.returncode, out, err

class hgcommand(object):
    def __init__(self, cmd, env):
        self.cmd = cmd
        self.env = env

    def run(self, args):
        cmd = self.cmd + args
        returncode, out, err = runcmd(cmd, self.env)
        # If root is executing setup.py, but the repository is owned by

            printf("stderr from '%s':" % (' '.join(cmd)), file=sys.stderr)
            printf(b'\n'.join([b'  ' + e for e in err]), file=sys.stderr)
            return ''
        return out

def findhg():
    """Try to figure out how we should invoke hg for examining the local
    repository contents.

    Returns an hgcommand object."""
    # By default, prefer the "hg" command in the user's path.  This was
    # presumably the hg command that the user used to create this repository.
    #
    # This repository may require extensions or other settings that would not
    # be enabled by running the hg script directly from this local repository.
    hgenv = os.environ.copy()
    # Use HGPLAIN to disable hgrc settings that would change output formatting,
    # and disable localization for the same reasons.
    hgenv['HGPLAIN'] = '1'
    hgenv['LANGUAGE'] = 'C'
    hgcmd = ['hg']
    # Run a simple "hg log" command just to see if using hg from the user's
    # path works and can successfully interact with this repository.
    check_cmd = ['log', '-r.', '-Ttest']
    try:
        retcode, out, err = runcmd(hgcmd + check_cmd, hgenv)
    except EnvironmentError:
        retcode = -1
    if retcode == 0:
        return hgcommand(hgcmd, hgenv)

    # Fall back to trying the local hg installation.
    hgenv = localhgenv()
    # Don't source any system hgrc files when using the local hg.
    hgenv['HGRCPATH'] = ''
    hgcmd = [sys.executable, 'hg']
    try:
        retcode, out, err = runcmd(hgcmd + check_cmd, hgenv)
    except EnvironmentError:
        retcode = -1
    if retcode == 0:
        return hgcommand(hgcmd, hgenv)

    raise SystemExit('Unable to find a working hg binary to extract the '
                     'version from the repository tags')

def localhgenv():
    """Get an environment dictionary to use for invoking or importing
    mercurial from the local repository."""
    # Execute hg out of this directory with a custom environment which takes
    # care to not use any hgrc files and do no localization.
    env = {'HGMODULEPOLICY': 'py',
           'HGRCPATH': '',
           'LANGUAGE': 'C',

        env['SystemRoot'] = os.environ['SystemRoot']

version = ''

if os.path.isdir('.hg'):
    hg = findhg()
    cmd = ['log', '-r', '.', '--template', '{tags}\n']
    numerictags = [t for t in hg.run(cmd).split() if t[0:1].isdigit()]
    hgid = hg.run(['id', '-i']).strip()
    if not hgid:
        # Bail out if hg is having problems interacting with this repository,

                f.write('# empty\n')

        # here no extension enabled, disabled() lists up everything
        code = ('import pprint; from mercurial import extensions; '
                'pprint.pprint(extensions.disabled())')
        returncode, out, err = runcmd([sys.executable, '-c', code],
                                      localhgenv())
        if err or returncode != 0:
            raise DistutilsExecError(err)

        with open(self._indexfilename, 'w') as f:
            f.write('# this file is autogenerated by setup.py\n')