|
|
"""A custom pygments lexer for IPython code cells.
|
|
|
|
|
|
Informs The pygments highlighting library of the quirks of IPython's superset
|
|
|
of Python -- magic commands, !shell commands, etc.
|
|
|
"""
|
|
|
#-----------------------------------------------------------------------------
|
|
|
# Copyright (c) 2013, the IPython Development Team.
|
|
|
#
|
|
|
# Distributed under the terms of the Modified BSD License.
|
|
|
#
|
|
|
# The full license is in the file COPYING.txt, distributed with this software.
|
|
|
#-----------------------------------------------------------------------------
|
|
|
|
|
|
#-----------------------------------------------------------------------------
|
|
|
# Imports
|
|
|
#-----------------------------------------------------------------------------
|
|
|
|
|
|
# Third-party imports
|
|
|
from pygments.lexers import PythonLexer, BashLexer
|
|
|
from pygments.lexer import bygroups, using
|
|
|
from pygments.token import Keyword, Operator, Text
|
|
|
|
|
|
#-----------------------------------------------------------------------------
|
|
|
# Class declarations
|
|
|
#-----------------------------------------------------------------------------
|
|
|
|
|
|
class IPythonLexer(PythonLexer):
|
|
|
"""
|
|
|
Pygments Lexer for use with IPython code. Inherits from
|
|
|
PythonLexer and adds information about IPython specific
|
|
|
keywords (i.e. magic commands, shell commands, etc.)
|
|
|
"""
|
|
|
|
|
|
#Basic properties
|
|
|
name = 'IPython'
|
|
|
aliases = ['ip', 'ipython']
|
|
|
filenames = ['*.ipy']
|
|
|
|
|
|
#Highlighting information
|
|
|
tokens = PythonLexer.tokens.copy()
|
|
|
tokens['root'] = [
|
|
|
(r'(\%+)(\w+)\s+(\.*)(\n)', bygroups(Operator, Keyword,
|
|
|
using(BashLexer), Text)),
|
|
|
(r'(\%+)(\w+)\b', bygroups(Operator, Keyword)),
|
|
|
(r'^(!)(.+)(\n)', bygroups(Operator, using(BashLexer), Text)),
|
|
|
] + tokens['root']
|
|
|
|
