pre-commit/pre_commit/output.py at main · pre-commit/pre-commit · GitHub

33 lines (25 loc) · 911 Bytes

from __future__ import annotations

import contextlib

import sys

from typing import Any

from typing import IO

def write(s: str, stream: IO[bytes] = sys.stdout.buffer) -> None:

stream.write(s.encode())

stream.flush()

def write_line_b(

s: bytes | None = None,

stream: IO[bytes] = sys.stdout.buffer,

logfile_name: str | None = None,

) -> None:

with contextlib.ExitStack() as exit_stack:

output_streams = [stream]

if logfile_name:

stream = exit_stack.enter_context(open(logfile_name, 'ab'))

output_streams.append(stream)

for output_stream in output_streams:

if s is not None:

output_stream.write(s)

output_stream.write(b'\n')

output_stream.flush()

def write_line(s: str | None = None, **kwargs: Any) -> None:

write_line_b(s.encode() if s is not None else s, **kwargs)

Expand file tree