diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2021-02-26 04:10:02 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2021-02-26 04:10:02 +0000 |
commit | 227038a454943a075c51b60988c53f77a605fa10 (patch) | |
tree | 15d438e5519a2ba9ee7ba8c6f864bbc597ea33b5 /ptpython/lexer.py | |
parent | Initial commit. (diff) | |
download | ptpython-0f157635f1e48118e2f823b632da3d813dac7dc1.tar.xz ptpython-0f157635f1e48118e2f823b632da3d813dac7dc1.zip |
Adding upstream version 3.0.16.upstream/3.0.16
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'ptpython/lexer.py')
-rw-r--r-- | ptpython/lexer.py | 28 |
1 files changed, 28 insertions, 0 deletions
diff --git a/ptpython/lexer.py b/ptpython/lexer.py new file mode 100644 index 0000000..62e470f --- /dev/null +++ b/ptpython/lexer.py @@ -0,0 +1,28 @@ +from typing import Callable, Optional + +from prompt_toolkit.document import Document +from prompt_toolkit.formatted_text import StyleAndTextTuples +from prompt_toolkit.lexers import Lexer, PygmentsLexer +from pygments.lexers import BashLexer +from pygments.lexers import Python3Lexer as PythonLexer + +__all__ = ["PtpythonLexer"] + + +class PtpythonLexer(Lexer): + """ + Lexer for ptpython input. + + If the input starts with an exclamation mark, use a Bash lexer, otherwise, + use a Python 3 lexer. + """ + + def __init__(self, python_lexer: Optional[Lexer] = None) -> None: + self.python_lexer = python_lexer or PygmentsLexer(PythonLexer) + self.system_lexer = PygmentsLexer(BashLexer) + + def lex_document(self, document: Document) -> Callable[[int], StyleAndTextTuples]: + if document.text.startswith("!"): + return self.system_lexer.lex_document(document) + + return self.python_lexer.lex_document(document) |