Initial colorful stubs

Author: fliiiix

pyrightconfig.stricter.json

@@ -30,6 +30,7 @@
         "stubs/boltons",
         "stubs/braintree",
         "stubs/cffi",
+        "stubs/colorful",
         "stubs/dateparser",
         "stubs/defusedxml",
         "stubs/docker",

stubs/colorful/METADATA.toml

@@ -0,0 +1,2 @@
+version = "0.5.*"
+upstream_repository = "https://github.com/timofurrer/colorful"

stubs/colorful/colorful/__init__.pyi

@@ -0,0 +1,12 @@
+MODIFIERS: dict[str, tuple[int, int]]
+MODIFIER_RESET_OFFSET: int
+FOREGROUND_COLOR_OFFSET: int
+BACKGROUND_COLOR_OFFSET: int
+COLOR_CLOSE_OFFSET: int
+CSI: str
+ANSI_ESCAPE_CODE: str
+NEST_PLACEHOLDER: str
+
+def round(value: float) -> int: ...
+def rgb_to_ansi256(r: int, g: int, b: int) -> int: ...
+def rgb_to_ansi16(r: int, g: int, b: int, use_bright: bool = False) -> int: ...

stubs/colorful/colorful/ansi.pyi

@@ -0,0 +1,4 @@
+def parse_colors(path: str) -> dict[str, str]: ...
+def parse_rgb_txt_file(path: str) -> dict[str, str]: ...
+def parse_json_color_file(path: str) -> dict[str, str]: ...
+def sanitize_color_palette(colorpalette: dict[str, str | tuple[int, int, int]]) -> dict[str, tuple[int, int, int]]: ...

stubs/colorful/colorful/colors.pyi

@@ -0,0 +1,100 @@
+from _typeshed import SupportsWrite
+
+# This module defines a function "str()", which is why "str" can't be used
+# as a type annotation or type alias.
+from builtins import str as _str
+from collections.abc import Iterator
+from typing import Any, Literal
+from typing_extensions import LiteralString, Self, TypeAlias
+
+# Custom type helpers
+_ColorModeType: TypeAlias = Literal[0, 8, 16, 256, 16777215]
+_PaletteType: TypeAlias = dict[_str, _str | tuple[int, int, int]]
+_SanitizedPaletteType: TypeAlias = dict[_str, tuple[int, int, int]]
+_StyleType: TypeAlias = tuple[_str, _str]
+
+DEFAULT_RGB_TXT_PATH: _str
+COLOR_PALETTE: dict[_str, _str]
+COLORNAMES_COLORS_PATH: _str
+
+class ColorfulError(Exception): ...
+class ColorfulAttributeError(AttributeError, ColorfulError): ...
+
+def translate_rgb_to_ansi_code(red: int, green: int, blue: int, offset: int, colormode: _ColorModeType) -> _str: ...
+def translate_colorname_to_ansi_code(
+    colorname: _str, offset: int, colormode: _ColorModeType, colorpalette: _PaletteType
+) -> _str: ...
+def resolve_modifier_to_ansi_code(modifiername: _str, colormode: _ColorModeType) -> _str: ...
+def translate_style(style: _str, colormode: _ColorModeType, colorpalette: _PaletteType) -> _str: ...
+def style_string(string: _str, ansi_style: _StyleType, colormode: _ColorModeType, nested: bool = False) -> _str: ...
+
+class ColorfulString:
+    orig_string: _str
+    styled_string: _str
+    colorful_ctx: Colorful
+    def __init__(self, orig_string: _str, styled_string: _str, colorful_ctx: Colorful) -> None: ...
+    def __len__(self) -> int: ...
+    def __iter__(self) -> Iterator[_str]: ...
+    def __add__(self, other: _str | ColorfulString) -> Self: ...
+    def __iadd__(self, other: _str | ColorfulString) -> Self: ...
+    def __radd__(self, other: _str | ColorfulString) -> Self: ...
+    def __mul__(self, other: _str) -> Self: ...
+    def __format__(self, format_spec: _str) -> _str: ...
+    def __getattr__(self, name: _str) -> Any: ...
+
+class Colorful:
+    NO_COLORS: int
+    ANSI_8_COLORS: int
+    ANSI_16_COLORS: int
+    ANSI_256_COLORS: int
+    TRUE_COLORS: int
+    COLORNAMES_COLORS = COLORNAMES_COLORS_PATH
+    close_fg_color: _str
+    close_bg_color: _str
+    no_bold: _str
+    no_dimmed: _str
+    no_italic: _str
+    no_underlined: _str
+    no_blinkslow: _str
+    no_blinkrapid: _str
+    no_inversed: _str
+    no_concealed: _str
+    no_struckthrough: _str
+    colormode: _ColorModeType
+    def __init__(self, colormode: _ColorModeType | None = None, colorpalette: _str | _PaletteType | None = None) -> None: ...
+    @property
+    def colorpalette(self) -> _SanitizedPaletteType | None: ...
+    @colorpalette.setter
+    def colorpalette(self, colorpalette: _str | _PaletteType) -> None: ...
+    def setup(
+        self,
+        colormode: _ColorModeType | None = None,
+        colorpalette: _str | _PaletteType | None = None,
+        extend_colors: bool = False,
+    ) -> None: ...
+    def disable(self) -> None: ...
+    def use_8_ansi_colors(self) -> None: ...
+    def use_16_ansi_colors(self) -> None: ...
+    def use_256_ansi_colors(self) -> None: ...
+    def use_true_colors(self) -> None: ...
+    def use_palette(self, colorpalette: _str | _PaletteType) -> None: ...
+    def update_palette(self, colorpalette: _str | _PaletteType) -> None: ...
+    def use_style(self, style_name: _str) -> None: ...
+    def format(self, string: _str, *args: LiteralString, **kwargs: LiteralString) -> _str: ...
+    def str(self, string: _str) -> ColorfulString: ...
+    def print(
+        self, *objects: object, sep: _str = " ", end: _str = "\n", file: SupportsWrite[_str] | None = None, flush: bool = False
+    ) -> None: ...
+
+    class ColorfulStyle:
+        colormode: _ColorModeType
+        colorful_ctx: Colorful
+        def __init__(self, style: _StyleType, colormode: _ColorModeType, colorful_ctx: Colorful) -> None: ...
+        def evaluate(self, string: _str, nested: bool = False) -> ColorfulString: ...
+        def __and__(self, other: Self) -> Self: ...
+        def __call__(self, string: _str, nested: bool = False) -> ColorfulString: ...
+        def __or__(self, other) -> ColorfulString: ...
+        def __eq__(self, other: object) -> bool: ...
+        def __hash__(self) -> int: ...
+
+    def __getattr__(self, name: _str) -> Any: ...

stubs/colorful/colorful/core.pyi

@@ -0,0 +1,2 @@
+SOLARIZED: dict[str, str]
+MONOKAI: dict[str, str]

stubs/colorful/colorful/styles.pyi

@@ -0,0 +1,7 @@
+NO_COLORS: int
+ANSI_8_COLORS: int
+ANSI_16_COLORS: int
+ANSI_256_COLORS: int
+TRUE_COLORS: int
+
+def detect_color_support(env: dict[str, str]) -> int: ...

stubs/colorful/colorful/terminal.pyi

@@ -0,0 +1,2 @@
+def hex_to_rgb(value: str) -> tuple[int, int, int]: ...
+def check_hex(value: str) -> None: ...