1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
# -*- coding: utf-8 -*-
#
# Copyright (C) 2010 Wikkid Developers
#
# This software is licensed under the GNU Affero General Public License
# version 3 (see the file LICENSE).
"""The formatter registry is able to return formatters by name."""
import re
from wikkid.formatter.pygmentsformatter import PygmentsFormatter
from wikkid.formatter.restformatter import RestructuredTextFormatter
# Both textile and markdown are optional.
try:
from wikkid.formatter.markdownformatter import MarkdownFormatter
has_markdown = True
except ImportError:
has_markdown = False
try:
from wikkid.formatter.textileformatter import TextileFormatter
has_textile = True
except ImportError:
has_textile = False
class FormatterRegistry(object):
"""Has a dictionary of formatters based on name."""
def __init__(self):
self.formatters = {
'pygments': PygmentsFormatter(),
'rest': RestructuredTextFormatter(),
}
if has_markdown:
self.formatters['markdown'] = MarkdownFormatter()
if has_textile:
self.formatters['textile'] = TextileFormatter()
def __getitem__(self, formatter):
return self.formatters[formatter]
formatter_registry = FormatterRegistry()
def get_formatter(name):
"""Get a formatter by name."""
return formatter_registry[name]
FORMAT_MATCHER = re.compile(r'^#\s+(\w+).*$')
def get_wiki_formatter(content: str, default_formatter: str) -> str:
"""Choose a wiki formatter based on the first line of content.
Args:
content: The content of the file.
default_formatter: The name of the default formatter.
The first line of the content may specify a formatter using the form:
# formatter-name
For example:
# pygments
The first line must start with a # and the first word must specify
a formatter name. If niether of those match, the default_formatter
is returned. If the default_formatter doesn't exist, a key error
is raised.
"""
end_of_line = content.find('\n')
match = FORMAT_MATCHER.match(content[:end_of_line])
if match is not None:
try:
name = match.group(1).lower()
formatter = formatter_registry[name]
return content[end_of_line + 1:], formatter
except KeyError:
# Fall through to returning the default.
pass
return content, formatter_registry[default_formatter]
|
