From 6af1485becae0a9847c1f49e7dd0e6040a412c9c Mon Sep 17 00:00:00 2001 From: Georg Brandl Date: Sun, 22 Jan 2017 18:35:50 +0100 Subject: modeline: work for the first N lines and add a test to keep it that way --- pygments/modeline.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) (limited to 'pygments/modeline.py') 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 -- cgit v1.2.1