summaryrefslogtreecommitdiffstats
path: root/contrib/python/Pygments/py3/pygments/formatters/bbcode.py
blob: 2d4dbbd1d6d76353a8dd3a9feb9d50ac4010b82e (plain) (blame)
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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
""" 
    pygments.formatters.bbcode 
    ~~~~~~~~~~~~~~~~~~~~~~~~~~ 
 
    BBcode formatter. 
 
    :copyright: Copyright 2006-2021 by the Pygments team, see AUTHORS.
    :license: BSD, see LICENSE for details. 
""" 
 
 
from pygments.formatter import Formatter 
from pygments.util import get_bool_opt 
 
__all__ = ['BBCodeFormatter'] 
 
 
class BBCodeFormatter(Formatter): 
    """ 
    Format tokens with BBcodes. These formatting codes are used by many 
    bulletin boards, so you can highlight your sourcecode with pygments before 
    posting it there. 
 
    This formatter has no support for background colors and borders, as there 
    are no common BBcode tags for that. 
 
    Some board systems (e.g. phpBB) don't support colors in their [code] tag, 
    so you can't use the highlighting together with that tag. 
    Text in a [code] tag usually is shown with a monospace font (which this 
    formatter can do with the ``monofont`` option) and no spaces (which you 
    need for indentation) are removed. 
 
    Additional options accepted: 
 
    `style` 
        The style to use, can be a string or a Style subclass (default: 
        ``'default'``). 
 
    `codetag` 
        If set to true, put the output into ``[code]`` tags (default: 
        ``false``) 
 
    `monofont` 
        If set to true, add a tag to show the code with a monospace font 
        (default: ``false``). 
    """ 
    name = 'BBCode' 
    aliases = ['bbcode', 'bb'] 
    filenames = [] 
 
    def __init__(self, **options): 
        Formatter.__init__(self, **options) 
        self._code = get_bool_opt(options, 'codetag', False) 
        self._mono = get_bool_opt(options, 'monofont', False) 
 
        self.styles = {} 
        self._make_styles() 
 
    def _make_styles(self): 
        for ttype, ndef in self.style: 
            start = end = '' 
            if ndef['color']: 
                start += '[color=#%s]' % ndef['color'] 
                end = '[/color]' + end 
            if ndef['bold']: 
                start += '[b]' 
                end = '[/b]' + end 
            if ndef['italic']: 
                start += '[i]' 
                end = '[/i]' + end 
            if ndef['underline']: 
                start += '[u]' 
                end = '[/u]' + end 
            # there are no common BBcodes for background-color and border 
 
            self.styles[ttype] = start, end 
 
    def format_unencoded(self, tokensource, outfile): 
        if self._code: 
            outfile.write('[code]') 
        if self._mono: 
            outfile.write('[font=monospace]') 
 
        lastval = '' 
        lasttype = None 
 
        for ttype, value in tokensource: 
            while ttype not in self.styles: 
                ttype = ttype.parent 
            if ttype == lasttype: 
                lastval += value 
            else: 
                if lastval: 
                    start, end = self.styles[lasttype] 
                    outfile.write(''.join((start, lastval, end))) 
                lastval = value 
                lasttype = ttype 
 
        if lastval: 
            start, end = self.styles[lasttype] 
            outfile.write(''.join((start, lastval, end))) 
 
        if self._mono: 
            outfile.write('[/font]') 
        if self._code: 
            outfile.write('[/code]') 
        if self._code or self._mono: 
            outfile.write('\n')