pre-commit/pre-commit-hooks
1
0
Fork 0
mirror of https://github.com/pre-commit/pre-commit-hooks.git synced 2026-03-29 10:16:52 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Refactored literals

Browse source
This commit is contained in:
Edoardo Bezzeccheri 2025-05-07 08:44:27 +00:00
parent 5c514f85cc
commit 068f38ba74
1 changed files with 8 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

12
pre_commit_hooks/end_of_file_fixer.py
View file

@ -5,6 +5,10 @@ import os
from collections.abc import Sequence
from typing import IO
LF = b'\n'
CR = b'\r'
CRLF = b'\r\n'
def fix_file(file_obj: IO[bytes]) -> int:
# Test for newline at end of file
@ -15,13 +19,13 @@ def fix_file(file_obj: IO[bytes]) -> int:
return 0
last_character = file_obj.read(1)
# last_character will be '' for an empty file
if last_character not in {b'\n', b'\r'} and last_character != b'':
if last_character not in {LF, CR} and last_character != b'':
# Needs this seek for windows, otherwise IOError
file_obj.seek(0, os.SEEK_END)
file_obj.write(b'\n')
file_obj.write(LF)
return 1
while last_character in {b'\n', b'\r'}:
while last_character in {LF, CR}:
# Deal with the beginning of the file
if file_obj.tell() == 1:
# If we've reached the beginning of the file and it is all
@ -38,7 +42,7 @@ def fix_file(file_obj: IO[bytes]) -> int:
# newlines. If we find extraneous newlines, then backtrack and trim them.
position = file_obj.tell()
remaining = file_obj.read()
for sequence in (b'\n', b'\r\n', b'\r'):
for sequence in (LF, CRLF, CR):
if remaining == sequence:
return 0
elif remaining.startswith(sequence):