third-party/pre-commit
1
0
Fork 0
mirror of https://github.com/pre-commit/pre-commit.git synced 2026-02-17 08:14:42 +04:00
Code Issues Projects Releases Packages Wiki Activity Actions

Refactor xargs.partition: _command_length usage

Browse source
This commit is contained in:
George Y. Kussumoto 2018-10-06 20:04:17 -03:00
parent bb6b1c33ae
commit 333ea75e45
1 changed files with 7 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

11
pre_commit/xargs.py
View file

@ -13,9 +13,8 @@ def _get_platform_max_length():
return 4 * 1024
def _get_command_length(command, arg):
parts = command + (arg,)
full_cmd = ' '.join(parts)
def _command_length(*cmd):
full_cmd = ' '.join(cmd)
# win32 uses the amount of characters, more details at:
# https://blogs.msdn.microsoft.com/oldnewthing/20031210-00/?p=41553/
@ -38,17 +37,21 @@ def partition(cmd, varargs, _max_length=None):
# Reversed so arguments are in order
varargs = list(reversed(varargs))
total_length = _command_length(*cmd)
while varargs:
arg = varargs.pop()
if _get_command_length(cmd + tuple(ret_cmd), arg) <= _max_length:
arg_length = _command_length(arg) + 1
if total_length + arg_length <= _max_length:
ret_cmd.append(arg)
total_length += arg_length
elif not ret_cmd:
raise ArgumentTooLongError(arg)
else:
# We've exceeded the length, yield a command
ret.append(cmd + tuple(ret_cmd))
ret_cmd = []
total_length = _command_length(*cmd)
varargs.append(arg)
ret.append(cmd + tuple(ret_cmd))