diff options
author | Cyril Jouve <jv.cyril@gmail.com> | 2018-06-04 23:44:44 +0200 |
---|---|---|
committer | Cyril Jouve <jv.cyril@gmail.com> | 2018-06-08 19:28:08 +0200 |
commit | b77225c13fff0f270bda09c68bc56e95051de8ea (patch) | |
tree | 001b87cf9a93695434da9d36ef15b012cd3a4f00 | |
parent | a907b4f0fa07014e2a96949a4b9b900fc0842fd7 (diff) | |
download | yapf-b77225c13fff0f270bda09c68bc56e95051de8ea.tar.gz |
avoid reading whole file in IsPythonFile
-rw-r--r-- | yapf/yapflib/file_resources.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/yapf/yapflib/file_resources.py b/yapf/yapflib/file_resources.py index 764e2d6..6e7202d 100644 --- a/yapf/yapflib/file_resources.py +++ b/yapf/yapflib/file_resources.py @@ -178,8 +178,8 @@ def IsPythonFile(filename): try: with py3compat.open_with_encoding( filename, mode='r', encoding=encoding) as fd: - first_line = fd.readlines()[0] - except (IOError, IndexError): + first_line = fd.readline(256) + except IOError: return False return re.match(r'^#!.*\bpython[23]?\b', first_line) |