tools/linter/trailing_newlines.py

#!/usr/bin/env python3

import fileinput
import os
import sys

NEWLINE, = b'\n'


def correct_trailing_newlines(filename: str) -> bool:
    with open(filename, 'rb') as f:
        a = len(f.read(2))
        if a == 0:
            return True
        elif a == 1:
            # file is wrong whether or not the only byte is a newline
            return False
        else:
            f.seek(-2, os.SEEK_END)
            b, c = f.read(2)
            # no ASCII byte is part of any non-ASCII character in UTF-8
            return b != NEWLINE and c == NEWLINE


def main() -> int:
    # mimic git grep exit code behavior
    exit_code = 1
    for line in fileinput.input():
        stripped = line.rstrip()
        if not correct_trailing_newlines(stripped):
            exit_code = 0
            print(stripped)
    return exit_code


if __name__ == '__main__':
    sys.exit(main())