iframe-proxy | Sunbelt Computer Software

History

105 lines (82 loc) · 3.06 KB

Raw

100

101

102

103

104

105

from __future__ import unicode_literals

import functools

import logging

import os

import os.path

import re

from pre_commit.errors import FatalError

from pre_commit.util import CalledProcessError

from pre_commit.util import cmd_output

from pre_commit.util import memoize_by_cwd

logger = logging.getLogger('pre_commit')

def get_root():

try:

return cmd_output('git', 'rev-parse', '--show-toplevel')[1].strip()

except CalledProcessError:

raise FatalError(

'git failed. Is it installed, and are you in a Git repository '

'directory?'

)

def get_git_dir(git_root):

return os.path.normpath(os.path.join(

git_root,

cmd_output('git', 'rev-parse', '--git-dir', cwd=git_root)[1].strip(),

))

def is_in_merge_conflict():

git_dir = get_git_dir('.')

return (

os.path.exists(os.path.join(git_dir, 'MERGE_MSG')) and

os.path.exists(os.path.join(git_dir, 'MERGE_HEAD'))

)

def parse_merge_msg_for_conflicts(merge_msg):

# Conflicted files start with tabs

return [

line.lstrip('#').strip()

for line in merge_msg.splitlines()

# '#\t' for git 2.4.1

if line.startswith(('\t', '#\t'))

]

@memoize_by_cwd

def get_conflicted_files():

logger.info('Checking merge-conflict files only.')

# Need to get the conflicted files from the MERGE_MSG because they could

# have resolved the conflict by choosing one side or the other

merge_msg = open(os.path.join(get_git_dir('.'), 'MERGE_MSG')).read()

merge_conflict_filenames = parse_merge_msg_for_conflicts(merge_msg)

# This will get the rest of the changes made after the merge.

# If they resolved the merge conflict by choosing a mesh of both sides

# this will also include the conflicted files

tree_hash = cmd_output('git', 'write-tree')[1].strip()

merge_diff_filenames = cmd_output(

'git', 'diff', '-m', tree_hash, 'HEAD', 'MERGE_HEAD', '--name-only',

)[1].splitlines()

return set(merge_conflict_filenames) | set(merge_diff_filenames)

@memoize_by_cwd

def get_staged_files():

return cmd_output(

'git', 'diff', '--staged', '--name-only',

# Everything except for D

'--diff-filter=ACMRTUXB'

)[1].splitlines()

@memoize_by_cwd

def get_all_files():

return cmd_output('git', 'ls-files')[1].splitlines()

def get_files_matching(all_file_list_strategy):

@functools.wraps(all_file_list_strategy)

@memoize_by_cwd

def wrapper(include_expr, exclude_expr):

include_regex = re.compile(include_expr)

exclude_regex = re.compile(exclude_expr)

return set(

filename

for filename in all_file_list_strategy()

if (

include_regex.search(filename) and

not exclude_regex.search(filename) and

os.path.lexists(filename)

)

return wrapper

get_staged_files_matching = get_files_matching(get_staged_files)

get_all_files_matching = get_files_matching(get_all_files)

get_conflicted_files_matching = get_files_matching(get_conflicted_files)

Sunbelt Computer Software

PL/B Language Development and Support

Navigation Menu

Sunbelt Computer Software

PL/B Language Development and Support

FilesExpand file tree

git.py

Latest commit

History

git.py

File metadata and controls