main commit
All checks were successful
continuous-integration/drone/push Build is passing

This commit is contained in:
2025-10-16 16:30:25 +09:00
parent 91c7e04474
commit 537e7b363f
1146 changed files with 45926 additions and 77196 deletions

View File

@@ -1,7 +1,6 @@
"""Terminals and colors."""
from __future__ import annotations
import base64
import codecs
import os
import platform
import sys
@@ -9,8 +8,6 @@ from functools import reduce
__all__ = ('colored',)
from typing import Any
BLACK, RED, GREEN, YELLOW, BLUE, MAGENTA, CYAN, WHITE = range(8)
OP_SEQ = '\033[%dm'
RESET_SEQ = '\033[0m'
@@ -29,7 +26,7 @@ _IMG_PRE = '\033Ptmux;\033\033]' if TERM_IS_SCREEN else '\033]'
_IMG_POST = '\a\033\\' if TERM_IS_SCREEN else '\a'
def fg(s: int) -> str:
def fg(s):
return COLOR_SEQ % s
@@ -44,11 +41,11 @@ class colored:
... c.green('dog ')))
"""
def __init__(self, *s: object, **kwargs: Any) -> None:
self.s: tuple[object, ...] = s
self.enabled: bool = not IS_WINDOWS and kwargs.get('enabled', True)
self.op: str = kwargs.get('op', '')
self.names: dict[str, Any] = {
def __init__(self, *s, **kwargs):
self.s = s
self.enabled = not IS_WINDOWS and kwargs.get('enabled', True)
self.op = kwargs.get('op', '')
self.names = {
'black': self.black,
'red': self.red,
'green': self.green,
@@ -59,10 +56,10 @@ class colored:
'white': self.white,
}
def _add(self, a: object, b: object) -> str:
return f"{a}{b}"
def _add(self, a, b):
return str(a) + str(b)
def _fold_no_color(self, a: Any, b: Any) -> str:
def _fold_no_color(self, a, b):
try:
A = a.no_color()
except AttributeError:
@@ -72,113 +69,109 @@ class colored:
except AttributeError:
B = str(b)
return f"{A}{B}"
return ''.join((str(A), str(B)))
def no_color(self) -> str:
def no_color(self):
if self.s:
return str(reduce(self._fold_no_color, self.s))
return ''
def embed(self) -> str:
def embed(self):
prefix = ''
if self.enabled:
prefix = self.op
return f"{prefix}{reduce(self._add, self.s)}"
return ''.join((str(prefix), str(reduce(self._add, self.s))))
def __str__(self) -> str:
def __str__(self):
suffix = ''
if self.enabled:
suffix = RESET_SEQ
return f"{self.embed()}{suffix}"
return str(''.join((self.embed(), str(suffix))))
def node(self, s: tuple[object, ...], op: str) -> colored:
def node(self, s, op):
return self.__class__(enabled=self.enabled, op=op, *s)
def black(self, *s: object) -> colored:
def black(self, *s):
return self.node(s, fg(30 + BLACK))
def red(self, *s: object) -> colored:
def red(self, *s):
return self.node(s, fg(30 + RED))
def green(self, *s: object) -> colored:
def green(self, *s):
return self.node(s, fg(30 + GREEN))
def yellow(self, *s: object) -> colored:
def yellow(self, *s):
return self.node(s, fg(30 + YELLOW))
def blue(self, *s: object) -> colored:
def blue(self, *s):
return self.node(s, fg(30 + BLUE))
def magenta(self, *s: object) -> colored:
def magenta(self, *s):
return self.node(s, fg(30 + MAGENTA))
def cyan(self, *s: object) -> colored:
def cyan(self, *s):
return self.node(s, fg(30 + CYAN))
def white(self, *s: object) -> colored:
def white(self, *s):
return self.node(s, fg(30 + WHITE))
def __repr__(self) -> str:
def __repr__(self):
return repr(self.no_color())
def bold(self, *s: object) -> colored:
def bold(self, *s):
return self.node(s, OP_SEQ % 1)
def underline(self, *s: object) -> colored:
def underline(self, *s):
return self.node(s, OP_SEQ % 4)
def blink(self, *s: object) -> colored:
def blink(self, *s):
return self.node(s, OP_SEQ % 5)
def reverse(self, *s: object) -> colored:
def reverse(self, *s):
return self.node(s, OP_SEQ % 7)
def bright(self, *s: object) -> colored:
def bright(self, *s):
return self.node(s, OP_SEQ % 8)
def ired(self, *s: object) -> colored:
def ired(self, *s):
return self.node(s, fg(40 + RED))
def igreen(self, *s: object) -> colored:
def igreen(self, *s):
return self.node(s, fg(40 + GREEN))
def iyellow(self, *s: object) -> colored:
def iyellow(self, *s):
return self.node(s, fg(40 + YELLOW))
def iblue(self, *s: colored) -> colored:
def iblue(self, *s):
return self.node(s, fg(40 + BLUE))
def imagenta(self, *s: object) -> colored:
def imagenta(self, *s):
return self.node(s, fg(40 + MAGENTA))
def icyan(self, *s: object) -> colored:
def icyan(self, *s):
return self.node(s, fg(40 + CYAN))
def iwhite(self, *s: object) -> colored:
def iwhite(self, *s):
return self.node(s, fg(40 + WHITE))
def reset(self, *s: object) -> colored:
return self.node(s or ('',), RESET_SEQ)
def reset(self, *s):
return self.node(s or [''], RESET_SEQ)
def __add__(self, other: object) -> str:
return f"{self}{other}"
def __add__(self, other):
return str(self) + str(other)
def supports_images() -> bool:
try:
return sys.stdin.isatty() and bool(os.environ.get('ITERM_PROFILE'))
except AttributeError:
return False
def supports_images():
return sys.stdin.isatty() and ITERM_PROFILE
def _read_as_base64(path: str) -> str:
with open(path, mode='rb') as fh:
def _read_as_base64(path):
with codecs.open(path, mode='rb') as fh:
encoded = base64.b64encode(fh.read())
return encoded.decode('ascii')
return encoded if isinstance(encoded, str) else encoded.decode('ascii')
def imgcat(path: str, inline: int = 1, preserve_aspect_ratio: int = 0, **kwargs: Any) -> str:
def imgcat(path, inline=1, preserve_aspect_ratio=0, **kwargs):
return '\n%s1337;File=inline=%d;preserveAspectRatio=%d:%s%s' % (
_IMG_PRE, inline, preserve_aspect_ratio,
_read_as_base64(path), _IMG_POST)