diff options
Diffstat (limited to 'venv/lib/python3.9/site-packages/pygments/modeline.py')
-rw-r--r-- | venv/lib/python3.9/site-packages/pygments/modeline.py | 43 |
1 files changed, 43 insertions, 0 deletions
diff --git a/venv/lib/python3.9/site-packages/pygments/modeline.py b/venv/lib/python3.9/site-packages/pygments/modeline.py new file mode 100644 index 00000000..7b6f6a32 --- /dev/null +++ b/venv/lib/python3.9/site-packages/pygments/modeline.py @@ -0,0 +1,43 @@ +""" + pygments.modeline + ~~~~~~~~~~~~~~~~~ + + A simple modeline parser (based on pymodeline). + + :copyright: Copyright 2006-2023 by the Pygments team, see AUTHORS. + :license: BSD, see LICENSE for details. +""" + +import re + +__all__ = ['get_filetype_from_buffer'] + + +modeline_re = re.compile(r''' + (?: vi | vim | ex ) (?: [<=>]? \d* )? : + .* (?: ft | filetype | syn | syntax ) = ( [^:\s]+ ) +''', re.VERBOSE) + + +def get_filetype_from_line(l): + m = modeline_re.search(l) + if m: + return m.group(1) + + +def get_filetype_from_buffer(buf, max_lines=5): + """ + Scan the buffer for modelines and return filetype if one is found. + """ + lines = buf.splitlines() + for l in lines[-1:-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 |