File: ansi_print.py

package info (click to toggle)
pypy3 7.0.0%2Bdfsg-3
  • links: PTS, VCS
  • area: main
  • in suites: buster
  • size: 111,848 kB
  • sloc: python: 1,291,746; ansic: 74,281; asm: 5,187; cpp: 3,017; sh: 2,533; makefile: 544; xml: 243; lisp: 45; csh: 21; awk: 4
file content (79 lines) | stat: -rw-r--r-- 2,264 bytes parent folder | download | duplicates (7)
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
"""
A simple color logger.
"""

import sys
from py.io import ansi_print
from rpython.tool.ansi_mandelbrot import Driver


isatty = lambda: True
mandelbrot_driver = Driver()
wrote_dot = False     # global shared state


def _make_method(subname, colors):
    #
    def logger_method(self, text):
        global wrote_dot
        if self.output_disabled:
            return
        text = "[%s%s] %s" % (self.name, subname, text)
        if isatty():
            col = colors
        else:
            col = ()
        if wrote_dot:
            text = '\n' + text
        ansi_print(text, col)
        wrote_dot = False
    #
    return logger_method


class AnsiLogger(object):
    output_disabled = False

    def __init__(self, name):
        self.name = name

    # these methods write "[name:method] text" to the terminal, with color codes
    red      = _make_method('', (31,))
    bold     = _make_method('', (1,))
    WARNING  = _make_method(':WARNING', (31,))
    event    = _make_method('', (1,))
    ERROR    = _make_method(':ERROR', (1, 31))
    Error    = _make_method(':Error', (1, 31))
    info     = _make_method(':info', (35,))
    stub     = _make_method(':stub', (34,))

    # some more methods used by sandlib
    call      = _make_method(':call', (34,))
    result    = _make_method(':result', (34,))
    exception = _make_method(':exception', (34,))
    vpath     = _make_method(':vpath', (35,))
    timeout   = _make_method('', (1, 31))

    # directly calling the logger writes "[name] text" with no particular color
    __call__ = _make_method('', ())

    # calling unknown method names writes "[name:method] text" without color
    def __getattr__(self, name):
        if name[0].isalpha():
            method = _make_method(':' + name, ())
            setattr(self.__class__, name, method)
            return getattr(self, name)
        raise AttributeError(name)

    def dot(self):
        """Output a mandelbrot dot to the terminal."""
        if not isatty():
            return
        global wrote_dot
        if not wrote_dot:
            mandelbrot_driver.reset()
            wrote_dot = True
        mandelbrot_driver.dot()

    def debug(self, info):
        """For messages that are dropped.  Can be monkeypatched in tests."""