mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
a11c1bbdd0
Signed-off-by: Edward Z. Yang <ezyangfb.com> Pull Request resolved: https://github.com/pytorch/pytorch/pull/76089 Approved by: https://github.com/albanD
38 lines
884 B
Python
Executable File
38 lines
884 B
Python
Executable File
#!/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())
|