iframe-proxy | Sunbelt Computer Software

History

63 lines (55 loc) · 2.08 KB

Raw

from __future__ import unicode_literals

import contextlib

import io

import logging

import time

from pre_commit.util import CalledProcessError

logger = logging.getLogger('pre_commit')

@contextlib.contextmanager

def staged_files_only(cmd_runner):

"""Clear any unstaged changes from the git working directory inside this

context.

Args:

cmd_runner - PrefixedCommandRunner

"""

# Determine if there are unstaged files

retcode, diff_stdout_binary, _ = cmd_runner.run(

[

'git', 'diff', '--ignore-submodules', '--binary', '--exit-code',

'--no-color', '--no-ext-diff',

retcode=None,

encoding=None,

)

if retcode and diff_stdout_binary.strip():

patch_filename = cmd_runner.path('patch{}'.format(int(time.time())))

logger.warning('Unstaged files detected.')

logger.info(

'Stashing unstaged files to {}.'.format(patch_filename),

)

# Save the current unstaged changes as a patch

with io.open(patch_filename, 'wb') as patch_file:

patch_file.write(diff_stdout_binary)

# Clear the working directory of unstaged changes

cmd_runner.run(['git', 'checkout', '--', '.'])

try:

yield

finally:

# Try to apply the patch we saved

try:

cmd_runner.run(('git', 'apply', patch_filename), encoding=None)

except CalledProcessError:

logger.warning(

'Stashed changes conflicted with hook auto-fixes... '

'Rolling back fixes...'

)

# We failed to apply the patch, presumably due to fixes made

# by hooks.

# Roll back the changes made by hooks.

cmd_runner.run(['git', 'checkout', '--', '.'])

cmd_runner.run(('git', 'apply', patch_filename), encoding=None)

logger.info('Restored changes from {}.'.format(patch_filename))

else:

# There weren't any staged files so we don't need to do anything

# special

yield

Sunbelt Computer Software

PL/B Language Development and Support

Navigation Menu

Sunbelt Computer Software

PL/B Language Development and Support

FilesExpand file tree

staged_files_only.py

Latest commit

History

staged_files_only.py

File metadata and controls