make_bootstrap.py

# -*- coding: utf-8 -*-
from __future__ import print_function
from __future__ import unicode_literals

import io
import textwrap
import virtualenv


def main():
    contents = virtualenv.create_bootstrap_script(textwrap.dedent(
        '''
        import distutils.spawn
        import os
        import os.path
        import subprocess

        def adjust_options(options, args):
            args[:] = [
                os.path.join(os.environ['HOME'], '.pre-commit-venv')
            ]

        def after_install(options, home_dir):
            subprocess.check_call([
                os.path.join(home_dir, 'bin', 'pip'),
                'install', 'pre-commit',
            ])

            bin_dir = os.path.join(os.environ['HOME'], 'bin')
            script_src = os.path.join(home_dir, 'bin', 'pre-commit')
            script_dest = os.path.join(bin_dir, 'pre-commit')
            print('*' * 79)
            print('Installing pre-commit to {0}'.format(script_dest))
            print('*' * 79)

            if not os.path.exists(bin_dir):
                os.mkdir(bin_dir)

            # os.symlink is not idempotent
            if os.path.exists(script_dest):
                os.remove(script_dest)

            os.symlink(script_src, script_dest)

            if not distutils.spawn.find_executable('pre-commit'):
                print('It looks like {0} is not on your path'.format(bin_dir))
                print('You may want to add it.')
                print('Often this does the trick: source ~/.profile')
        '''
    ))

    with io.open('install-local.py', 'w', encoding='UTF-8') as install_file:
        install_file.write(contents)


if __name__ == '__main__':
    exit(main())