diff options
author | Georg Brandl <georg@python.org> | 2017-01-22 18:35:50 +0100 |
---|---|---|
committer | Georg Brandl <georg@python.org> | 2017-01-22 18:35:50 +0100 |
commit | 6af1485becae0a9847c1f49e7dd0e6040a412c9c (patch) | |
tree | 4c16d6d77da6d7c3ba271805cdcc862a2881698a /pygments/modeline.py | |
parent | 304ab38c9a49800c86c5ccc97e77f37f17c1480f (diff) | |
download | pygments-6af1485becae0a9847c1f49e7dd0e6040a412c9c.tar.gz |
modeline: work for the first N lines and add a test to keep it that way
Diffstat (limited to 'pygments/modeline.py')
-rw-r--r-- | pygments/modeline.py | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/pygments/modeline.py b/pygments/modeline.py index 2200f1cf..43c1784f 100644 --- a/pygments/modeline.py +++ b/pygments/modeline.py @@ -35,9 +35,10 @@ def get_filetype_from_buffer(buf, max_lines=5): ret = get_filetype_from_line(l) if ret: return ret - for l in lines[max_lines:-1:-1]: - ret = get_filetype_from_line(l) - if ret: - return ret + for i in range(max_lines, -1, -1): + if i < len(lines): + ret = get_filetype_from_line(lines[i]) + if ret: + return ret return None |