don't read whole file to detect encoding

2020-03-24 22:19:01 -07:00
parent 10daf4c45f
commit d60e9fd0f6
1 changed files with 8 additions and 8 deletions
--- a/yamllint/cli.py
+++ b/yamllint/cli.py
@@ -31,14 +31,14 @@ from yamllint.linter import PROBLEM_LEVELS

 def determine_encoding(file):
    with io.open(file, 'rb') as raw_file:
-        data = raw_file.read()
-        if data.startswith(codecs.BOM_UTF16_LE):
-            encoding = 'utf-16-le'
-        elif data.startswith(codecs.BOM_UTF16_BE):
-            encoding = 'utf-16-be'
-        else:
-            encoding = 'utf-8'
-        return encoding
+        data = raw_file.read(4)
+    if data.startswith(codecs.BOM_UTF16_LE):
+        encoding = 'utf-16-le'
+    elif data.startswith(codecs.BOM_UTF16_BE):
+        encoding = 'utf-16-be'
+    else:
+        encoding = 'utf-8'
+    return encoding


 def find_files_recursively(items, conf):