?HAMA

Edit File Size: 2.51 KB

/opt/hc_python/lib64/python3.12/site-packages/pre_commit/commands/hazmat.py

Text Base64

from __future__ import annotations

import argparse
import subprocess
from collections.abc import Sequence

from pre_commit.parse_shebang import normalize_cmd

def add_parsers(parser: argparse.ArgumentParser) -> None:
    subparsers = parser.add_subparsers(dest='tool')

cd_parser = subparsers.add_parser(
        'cd', help='cd to a subdir and run the command',
    )
    cd_parser.add_argument('subdir')
    cd_parser.add_argument('cmd', nargs=argparse.REMAINDER)

ignore_exit_code_parser = subparsers.add_parser(
        'ignore-exit-code', help='run the command but ignore the exit code',
    )
    ignore_exit_code_parser.add_argument('cmd', nargs=argparse.REMAINDER)

n1_parser = subparsers.add_parser(
        'n1', help='run the command once per filename',
    )
    n1_parser.add_argument('cmd', nargs=argparse.REMAINDER)

def _cmd_filenames(cmd: tuple[str, ...]) -> tuple[
    tuple[str, ...],
    tuple[str, ...],
]:
    for idx, val in enumerate(reversed(cmd)):
        if val == '--':
            split = len(cmd) - idx
            break
    else:
        raise SystemExit('hazmat entry must end with `--`')

return cmd[:split - 1], cmd[split:]

def cd(subdir: str, cmd: tuple[str, ...]) -> int:
    cmd, filenames = _cmd_filenames(cmd)

prefix = f'{subdir}/'
    new_filenames = []
    for filename in filenames:
        if not filename.startswith(prefix):
            raise SystemExit(f'unexpected file without {prefix=}: {filename}')
        else:
            new_filenames.append(filename.removeprefix(prefix))

cmd = normalize_cmd(cmd)
    return subprocess.call((*cmd, *new_filenames), cwd=subdir)

def ignore_exit_code(cmd: tuple[str, ...]) -> int:
    cmd = normalize_cmd(cmd)
    subprocess.call(cmd)
    return 0

def n1(cmd: tuple[str, ...]) -> int:
    cmd, filenames = _cmd_filenames(cmd)
    cmd = normalize_cmd(cmd)
    ret = 0
    for filename in filenames:
        ret |= subprocess.call((*cmd, filename))
    return ret

def impl(args: argparse.Namespace) -> int:
    args.cmd = tuple(args.cmd)
    if args.tool == 'cd':
        return cd(args.subdir, args.cmd)
    elif args.tool == 'ignore-exit-code':
        return ignore_exit_code(args.cmd)
    elif args.tool == 'n1':
        return n1(args.cmd)
    else:
        raise NotImplementedError(f'unexpected tool: {args.tool}')

def main(argv: Sequence[str] | None = None) -> int:
    parser = argparse.ArgumentParser()
    add_parsers(parser)
    args = parser.parse_args(argv)

return impl(args)

if __name__ == '__main__':
    raise SystemExit(main())

Exit & Reset Text mode: syntax highlighting auto-detects file type.

Name	Size	Perms	Modified	Actions
__pycache__ DIR	-	drwxr-xr-x	2025-12-03 08:02:54	Edit Download recursive recursive
autoupdate.py	6.99 KB	lrw-r--r--	2025-12-03 08:02:54	Edit Download recursive recursive
clean.py	429 B	lrw-r--r--	2025-12-03 08:02:54	Edit Download recursive recursive
gc.py	2.97 KB	lrw-r--r--	2025-12-03 08:02:54	Edit Download recursive recursive
hazmat.py	2.51 KB	lrw-r--r--	2025-12-03 08:02:54	Edit Download recursive recursive
hook_impl.py	9.20 KB	lrw-r--r--	2025-12-03 08:02:54	Edit Download recursive recursive
init_templatedir.py	1.11 KB	lrw-r--r--	2025-12-03 08:02:54	Edit Download recursive recursive
install_uninstall.py	5.22 KB	lrw-r--r--	2025-12-03 08:02:54	Edit Download recursive recursive
migrate_config.py	4.07 KB	lrw-r--r--	2025-12-03 08:02:54	Edit Download recursive recursive
run.py	13.83 KB	lrw-r--r--	2025-12-03 08:02:54	Edit Download recursive recursive
sample_config.py	453 B	lrw-r--r--	2025-12-03 08:02:54	Edit Download recursive recursive
try_repo.py	2.52 KB	lrw-r--r--	2025-12-03 08:02:54	Edit Download recursive recursive
validate_config.py	371 B	lrw-r--r--	2025-12-03 08:02:54	Edit Download recursive recursive
validate_manifest.py	377 B	lrw-r--r--	2025-12-03 08:02:54	Edit Download recursive recursive
__init__.py	0 B	lrw-r--r--	2025-12-03 08:02:54	Edit Download recursive recursive

Directory Contents