Files
pre-commit-hooks/pre_commit_hooks/check_merge_conflict.py
2016-11-30 09:56:42 -08:00

46 lines
1.1 KiB
Python

from __future__ import print_function
import argparse
import os.path
CONFLICT_PATTERNS = [
b'<<<<<<< ',
b'======= ',
b'=======\n',
b'>>>>>>> '
]
WARNING_MSG = 'Merge conflict string "{0}" found in {1}:{2}'
def is_in_merge():
return (
os.path.exists(os.path.join('.git', 'MERGE_MSG')) and
os.path.exists(os.path.join('.git', 'MERGE_HEAD'))
)
def detect_merge_conflict(argv=None):
parser = argparse.ArgumentParser()
parser.add_argument('filenames', nargs='*')
args = parser.parse_args(argv)
if not is_in_merge():
return 0
retcode = 0
for filename in args.filenames:
with open(filename, 'rb') as inputfile:
for i, line in enumerate(inputfile):
for pattern in CONFLICT_PATTERNS:
if line.startswith(pattern):
print(WARNING_MSG.format(
pattern.decode(), filename, i + 1,
))
retcode = 1
return retcode
if __name__ == '__main__':
exit(detect_merge_conflict())