forked from github/pelican
89b28fd36b
Starting with python 3.6 warnings are issued for invalid escape sequences in regular expressions. This commit corrects all DeprecationWarning's via properly declaring the offending regular expressions as raw strings. Resolves #2095.
95 lines
3 KiB
Python
95 lines
3 KiB
Python
# -*- coding: utf-8 -*-
|
|
from __future__ import print_function, unicode_literals
|
|
|
|
import re
|
|
|
|
from docutils import nodes, utils
|
|
from docutils.parsers.rst import Directive, directives, roles
|
|
|
|
from pygments import highlight
|
|
from pygments.formatters import HtmlFormatter
|
|
from pygments.lexers import TextLexer, get_lexer_by_name
|
|
|
|
import six
|
|
|
|
import pelican.settings as pys
|
|
|
|
|
|
class Pygments(Directive):
|
|
""" Source code syntax highlighting.
|
|
"""
|
|
required_arguments = 1
|
|
optional_arguments = 0
|
|
final_argument_whitespace = True
|
|
option_spec = {
|
|
'anchorlinenos': directives.flag,
|
|
'classprefix': directives.unchanged,
|
|
'hl_lines': directives.unchanged,
|
|
'lineanchors': directives.unchanged,
|
|
'linenos': directives.unchanged,
|
|
'linenospecial': directives.nonnegative_int,
|
|
'linenostart': directives.nonnegative_int,
|
|
'linenostep': directives.nonnegative_int,
|
|
'lineseparator': directives.unchanged,
|
|
'linespans': directives.unchanged,
|
|
'nobackground': directives.flag,
|
|
'nowrap': directives.flag,
|
|
'tagsfile': directives.unchanged,
|
|
'tagurlformat': directives.unchanged,
|
|
}
|
|
has_content = True
|
|
|
|
def run(self):
|
|
self.assert_has_content()
|
|
try:
|
|
lexer = get_lexer_by_name(self.arguments[0])
|
|
except ValueError:
|
|
# no lexer found - use the text one instead of an exception
|
|
lexer = TextLexer()
|
|
|
|
# Fetch the defaults
|
|
if pys.PYGMENTS_RST_OPTIONS is not None:
|
|
for k, v in six.iteritems(pys.PYGMENTS_RST_OPTIONS):
|
|
# Locally set options overrides the defaults
|
|
if k not in self.options:
|
|
self.options[k] = v
|
|
|
|
if ('linenos' in self.options and
|
|
self.options['linenos'] not in ('table', 'inline')):
|
|
if self.options['linenos'] == 'none':
|
|
self.options.pop('linenos')
|
|
else:
|
|
self.options['linenos'] = 'table'
|
|
|
|
for flag in ('nowrap', 'nobackground', 'anchorlinenos'):
|
|
if flag in self.options:
|
|
self.options[flag] = True
|
|
|
|
# noclasses should already default to False, but just in case...
|
|
formatter = HtmlFormatter(noclasses=False, **self.options)
|
|
parsed = highlight('\n'.join(self.content), lexer, formatter)
|
|
return [nodes.raw('', parsed, format='html')]
|
|
|
|
|
|
directives.register_directive('code-block', Pygments)
|
|
directives.register_directive('sourcecode', Pygments)
|
|
|
|
|
|
_abbr_re = re.compile(r'\((.*)\)$', re.DOTALL)
|
|
|
|
|
|
class abbreviation(nodes.Inline, nodes.TextElement):
|
|
pass
|
|
|
|
|
|
def abbr_role(typ, rawtext, text, lineno, inliner, options={}, content=[]):
|
|
text = utils.unescape(text)
|
|
m = _abbr_re.search(text)
|
|
if m is None:
|
|
return [abbreviation(text, text)], []
|
|
abbr = text[:m.start()].strip()
|
|
expl = m.group(1)
|
|
return [abbreviation(abbr, abbr, explanation=expl)], []
|
|
|
|
|
|
roles.register_local_role('abbr', abbr_role)
|