pre-commit-hooks/pre_commit_hooks/fix_byte_order_marker.py at main · pre-commit/pre-commit-hooks

31 lines (21 loc) · 823 Bytes

from __future__ import annotations

import argparse

from collections.abc import Sequence

def main(argv: Sequence[str] | None = None) -> int:

parser = argparse.ArgumentParser()

parser.add_argument('filenames', nargs='*', help='Filenames to check')

args = parser.parse_args(argv)

retv = 0

for filename in args.filenames:

with open(filename, 'rb') as f_b:

bts = f_b.read(3)

if bts == b'\xef\xbb\xbf':

with open(filename, newline='', encoding='utf-8-sig') as f:

contents = f.read()

with open(filename, 'w', newline='', encoding='utf-8') as f:

f.write(contents)

print(f'{filename}: removed byte-order marker')

retv = 1

return retv

if __name__ == '__main__':

raise SystemExit(main())

Provide feedback