summaryrefslogtreecommitdiffstats
path: root/pre_commit_hooks/util.py
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2022-01-30 11:02:58 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-05-21 20:41:55 +0000
commitd4583dcad7d68d3c1503b04ec0d3364809304807 (patch)
tree82456c5d0bc77961759812ddd85414435ba89127 /pre_commit_hooks/util.py
parentInitial commit. (diff)
downloadpre-commit-hooks-d4583dcad7d68d3c1503b04ec0d3364809304807.tar.xz
pre-commit-hooks-d4583dcad7d68d3c1503b04ec0d3364809304807.zip
Adding upstream version 4.5.0+dfsg.upstream/4.5.0+dfsgupstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to '')
-rw-r--r--pre_commit_hooks/util.py32
1 files changed, 32 insertions, 0 deletions
diff --git a/pre_commit_hooks/util.py b/pre_commit_hooks/util.py
new file mode 100644
index 0000000..d6c90ae
--- /dev/null
+++ b/pre_commit_hooks/util.py
@@ -0,0 +1,32 @@
+from __future__ import annotations
+
+import subprocess
+from typing import Any
+
+
+class CalledProcessError(RuntimeError):
+ pass
+
+
+def added_files() -> set[str]:
+ cmd = ('git', 'diff', '--staged', '--name-only', '--diff-filter=A')
+ return set(cmd_output(*cmd).splitlines())
+
+
+def cmd_output(*cmd: str, retcode: int | None = 0, **kwargs: Any) -> str:
+ kwargs.setdefault('stdout', subprocess.PIPE)
+ kwargs.setdefault('stderr', subprocess.PIPE)
+ proc = subprocess.Popen(cmd, **kwargs)
+ stdout, stderr = proc.communicate()
+ stdout = stdout.decode()
+ if retcode is not None and proc.returncode != retcode:
+ raise CalledProcessError(cmd, retcode, proc.returncode, stdout, stderr)
+ return stdout
+
+
+def zsplit(s: str) -> list[str]:
+ s = s.strip('\0')
+ if s:
+ return s.split('\0')
+ else:
+ return []