Remove dependency on plumbum

This commit is contained in:
Anthony Sottile
2015-03-20 13:52:21 -07:00
parent c9b6161fab
commit 713fab4bc7
12 changed files with 99 additions and 55 deletions

View File

@ -2,10 +2,26 @@ from __future__ import absolute_import
from __future__ import print_function
from __future__ import unicode_literals
from plumbum import local
import subprocess
class CalledProcessError(RuntimeError):
pass
def added_files():
return set(local['git'](
'diff', '--staged', '--name-only', '--diff-filter', 'A',
return set(cmd_output(
'git', 'diff', '--staged', '--name-only', '--diff-filter', 'A',
).splitlines())
def cmd_output(*cmd, **kwargs):
retcode = kwargs.pop('retcode', 0)
popen_kwargs = {'stdout': subprocess.PIPE, 'stderr': subprocess.PIPE}
popen_kwargs.update(kwargs)
proc = subprocess.Popen(cmd, **popen_kwargs)
stdout, stderr = proc.communicate()
stdout, stderr = stdout.decode('UTF-8'), stderr.decode('UTF-8')
if retcode is not None and proc.returncode != retcode:
raise CalledProcessError(cmd, retcode, proc.returncode, stdout, stderr)
return stdout