summaryrefslogtreecommitdiffstats
path: root/crmsh/clidisplay.py
blob: fdcc50d3bff6d227ed62f9dadeb7a97880653887 (plain)
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
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
# Copyright (C) 2008-2011 Dejan Muhamedagic <dmuhamedagic@suse.de>
# See COPYING for license information.

"""
Display output for various syntax elements.
"""

from contextlib import contextmanager

from . import config


# Enable colors/upcasing
_pretty = True


def enable_pretty():
    global _pretty
    _pretty = True


def disable_pretty():
    global _pretty
    _pretty = False


@contextmanager
def nopretty(cond=True):
    if cond:
        disable_pretty()
    try:
        yield
    finally:
        if cond:
            enable_pretty()


def colors_enabled():
    return 'color' in config.color.style and _pretty


def _colorize(s, colors):
    if s and colors_enabled():
        return ''.join(('${%s}' % clr.upper()) for clr in colors) + s + '${NORMAL}'
    return s


def error(s):
    return _colorize(s, config.color.error)


def ok(s):
    return _colorize(s, config.color.ok)


def info(s):
    return _colorize(s, config.color.info)


def warn(s):
    return _colorize(s, config.color.warn)


def keyword(s):
    if "uppercase" in config.color.style:
        s = s.upper()
    if "color" in config.color.style:
        s = _colorize(s, config.color.keyword)
    return s


def prompt(s):
    if colors_enabled():
        s = "${RLIGNOREBEGIN}${GREEN}${BOLD}${RLIGNOREEND}" + s
        return s + "${RLIGNOREBEGIN}${NORMAL}${RLIGNOREEND}"
    return s


def prompt_noreadline(s):
    if colors_enabled():
        return "${GREEN}${BOLD}" + s + "${NORMAL}"
    return s


def help_header(s):
    return _colorize(s, config.color.help_header)


def help_keyword(s):
    return _colorize(s, config.color.help_keyword)


def help_topic(s):
    return _colorize(s, config.color.help_topic)


def help_block(s):
    return _colorize(s, config.color.help_block)


def ident(s):
    return _colorize(s, config.color.identifier)


def attr_name(s):
    return _colorize(s, config.color.attr_name)


def attr_value(s):
    return _colorize(s, config.color.attr_value)


def rscref(s):
    return _colorize(s, config.color.resource_reference)


def idref(s):
    return _colorize(s, config.color.id_reference)


def score(s):
    return _colorize(s, config.color.score)


def ticket(s):
    return _colorize(s, config.color.ticket)


# vim:ts=4:sw=4:et: