projects
/
python_utils.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
Fix a bug in sanity check image.
[python_utils.git]
/
ansi.py
diff --git
a/ansi.py
b/ansi.py
index 03f8fd27473c4e295cdafbb9e7122b4cec296453..a49760037e31c6c01e84a225e7147d588f8f5934 100755
(executable)
--- a/
ansi.py
+++ b/
ansi.py
@@
-4,20
+4,20
@@
setting the text color, background, etc... using ANSI escape
sequences."""
setting the text color, background, etc... using ANSI escape
sequences."""
+import contextlib
import difflib
import io
import logging
import re
import sys
from abc import abstractmethod
import difflib
import io
import logging
import re
import sys
from abc import abstractmethod
-from typing import Any, Callable, Dict, Iterable, Optional, Tuple
+from typing import Any, Callable, Dict, Iterable,
Literal,
Optional, Tuple
from overrides import overrides
import logging_utils
import string_utils
from overrides import overrides
import logging_utils
import string_utils
-
logger = logging.getLogger(__name__)
# https://en.wikipedia.org/wiki/ANSI_escape_code
logger = logging.getLogger(__name__)
# https://en.wikipedia.org/wiki/ANSI_escape_code
@@
-1885,10
+1885,9
@@
def bg(
return bg_24bit(red, green, blue)
return bg_24bit(red, green, blue)
-class StdoutInterceptor(io.TextIOBase):
- """An interceptor for data written to stdout. Use as a context.
+class StdoutInterceptor(io.TextIOBase
, contextlib.AbstractContextManager
):
+ """An interceptor for data written to stdout. Use as a context.
"""
- """
def __init__(self):
super().__init__()
self.saved_stdout: io.TextIO = None
def __init__(self):
super().__init__()
self.saved_stdout: io.TextIO = None
@@
-1903,10
+1902,10
@@
class StdoutInterceptor(io.TextIOBase):
sys.stdout = self
return self
sys.stdout = self
return self
- def __exit__(self, *args) ->
Optional[bool
]:
+ def __exit__(self, *args) ->
Literal[False
]:
sys.stdout = self.saved_stdout
print(self.buf)
sys.stdout = self.saved_stdout
print(self.buf)
- return
Non
e
+ return
Fals
e
class ProgrammableColorizer(StdoutInterceptor):
class ProgrammableColorizer(StdoutInterceptor):
@@
-1914,6
+1913,7
@@
class ProgrammableColorizer(StdoutInterceptor):
something (usually add color to) the match.
"""
something (usually add color to) the match.
"""
+
def __init__(
self,
patterns: Iterable[Tuple[re.Pattern, Callable[[Any, re.Pattern], str]]],
def __init__(
self,
patterns: Iterable[Tuple[re.Pattern, Callable[[Any, re.Pattern], str]]],
@@
-1929,6
+1929,7
@@
class ProgrammableColorizer(StdoutInterceptor):
if __name__ == '__main__':
if __name__ == '__main__':
+
def main() -> None:
import doctest
def main() -> None:
import doctest