chromatic-python 0.2.0__py3-none-any.whl → 0.2.2__py3-none-any.whl

chromatic/__init__.py

@@ -7,6 +7,7 @@ from . import ascii, color, data
 from .ascii import (
     AnsiImage,
     ansi2img,
+    ansify,
     ansi_quantize,
     ascii2img,
     ascii_printable,
@@ -40,3 +41,69 @@ from .color import (
 from .data import register_user_font
 
 __all__ = []
+
+try:
+    import os
+    import sys
+    from functools import lru_cache, wraps
+    from types import ModuleType
+
+    def find_modules(path: str):
+        from setuptools import find_packages
+        from pkgutil import iter_modules
+
+        tree: dict[str, dict | ModuleType] = {__name__: {'__module__': sys.modules[__name__]}}
+        children = set()
+        for pkg in find_packages(path):
+            children.add(pkg)
+            pkg_path = path + '/' + pkg.replace('.', '/')
+            if sys.version_info.major == 2 or (
+                sys.version_info.major == 3 and sys.version_info.minor < 6
+            ):
+                for _, name, ispkg in iter_modules([pkg_path]):
+                    if not ispkg:
+                        children.add(f"{pkg}.{name}")
+            else:
+                for info in iter_modules([pkg_path]):
+                    if not info.ispkg:
+                        children.add(f"{pkg}.{info.name}")
+        for child in children:
+            name = f"{__name__}.{child}"
+            depth = tree
+            for node in name.split('.'):
+                if node not in depth:
+                    depth[node] = {}
+                depth = depth[node]
+            if name in sys.modules:
+                depth['__module__'] = sys.modules[name]
+        return tree
+
+    def publicize_modules(modulename: str, tree: dict[str, dict | ModuleType]):
+        def is_local(obj: object):
+            if isinstance(obj, ModuleType):
+                return obj.__spec__.parent == modulename
+            elif hasattr(obj, '__module__'):
+                return obj.__module__ == modulename
+            return obj is not None
+
+        for name, subtree in tree.items():
+            if name == '__module__' and isinstance(subtree, ModuleType):
+                submodule = subtree
+                init_dir = dir(submodule)
+
+                @wraps(submodule.__dir__)
+                def wrapped():
+                    s = set(attr for attr in init_dir if is_local(getattr(submodule, attr, None)))
+                    if hasattr(submodule, '__all__'):
+                        s.update(submodule.__all__)
+                    return list(s)
+
+                sys.modules[submodule.__name__].__dir__ = wrapped
+
+            else:
+                publicize_modules(f"{modulename}.{name}", subtree)
+
+    publicize_modules(*find_modules(os.path.split(__file__)[0]).popitem())
+
+finally:
+    del find_modules, publicize_modules

chromatic/_typing.py

@@ -1,13 +1,13 @@
 from __future__ import annotations
 
 import inspect
+import operator as op
 import re
 import types
 from collections import OrderedDict, namedtuple
 from collections.abc import Callable as ABC_Callable
 from functools import reduce, wraps
 from numbers import Number
-from operator import attrgetter, or_ as bitwise_or
 from typing import (
     Any,
     Callable,
@@ -79,7 +79,7 @@ FontArgType: TypeAlias = FreeTypeFont | UserFont | str | int
 
 def type_error_msg(err_obj, *expected, context: str = '', obj_repr=False):
     n_expected = len(expected)
-    name_slots = [f"{{{n}.__qualname__!r}}" for n in range(n_expected)]
+    name_slots = ['{%d.__qualname__!r}' % n for n in range(n_expected)]
     if name_slots and n_expected > 1:
         name_slots[-1] = f"or {name_slots[-1]}"
     names = (', ' if n_expected > 2 else ' ').join([context.strip()] + name_slots).format(*expected)
@@ -183,7 +183,7 @@ class SupportsUnion(Protocol[_T_contra, _T_co]):
 
 
 def unionize(__iterable: Iterable[SupportsUnion[_T_contra, _T_co]]) -> _T_co:
-    return reduce(bitwise_or, __iterable)
+    return reduce(op.or_, __iterable)
 
 
 _GenericAlias = type(Type[...]) | types.GenericAlias
@@ -213,7 +213,7 @@ class _BoundedDict[_KT, _VT](OrderedDict[_KT, _VT]):
 
 
 _SUBTYPE_CACHE: _BoundedDict[int, ...] = _BoundedDict()
-_ATTR_GETTERS: _BoundedDict[..., tuple[Callable[[Iterable], NamedTuple], attrgetter]] = (
+_ATTR_GETTERS: _BoundedDict[..., tuple[Callable[[Iterable], NamedTuple], op.attrgetter]] = (
     _BoundedDict()
 )
 
@@ -247,7 +247,10 @@ def _unique_attrs(obj) -> Optional['NamedTuple']:
         + 'Attrs'
     ).strip('_')
     UniqueAttrs = cast(NamedTuple, namedtuple(tup_name, field_names))
-    _ATTR_GETTERS[tp] = constructor, getter = UniqueAttrs._make, attrgetter(*attr_names)  # noqa
+    _ATTR_GETTERS[tp] = [constructor, getter] = [
+        UniqueAttrs._make,
+        op.attrgetter(*attr_names),  # noqa
+    ]
     return constructor(getter(obj))
 
 

chromatic/_version.py

@@ -1,8 +1,13 @@
-# file generated by setuptools_scm
+# file generated by setuptools-scm
 # don't change, don't track in version control
+
+__all__ = ["__version__", "__version_tuple__", "version", "version_tuple"]
+
 TYPE_CHECKING = False
 if TYPE_CHECKING:
-    from typing import Tuple, Union
+    from typing import Tuple
+    from typing import Union
+
     VERSION_TUPLE = Tuple[Union[int, str], ...]
 else:
     VERSION_TUPLE = object
@@ -12,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.2.0'
-__version_tuple__ = version_tuple = (0, 2, 0)
+__version__ = version = '0.2.2'
+__version_tuple__ = version_tuple = (0, 2, 2)

chromatic/ascii/_array.py

@@ -1,5 +1,29 @@
 from __future__ import annotations
 
+__all__ = [
+    'ansi2img',
+    'ansify',
+    'ansi_quantize',
+    'ascii2img',
+    'contrast_stretch',
+    'equalize_white_point',
+    'get_font_key',
+    'get_font_object',
+    'img2ansi',
+    'img2ascii',
+    'read_ans',
+    'render_ans',
+    'render_font_char',
+    'render_font_str',
+    'reshape_ansi',
+    'scale_saturation',
+    'shuffle_char_set',
+    '_scaled_hu_moments',
+    'to_sgr_array',
+    'AnsiImage',
+    '_otsu_mask',
+]
+
 import math
 import os.path
 import random
@@ -62,29 +86,6 @@ from ..data import UserFont
 if TYPE_CHECKING:
     from _typeshed import AnyStr_co
 
-__all__ = [
-    'ansi2img',
-    'ansi_quantize',
-    'ascii2img',
-    'contrast_stretch',
-    'equalize_white_point',
-    'get_font_key',
-    'get_font_object',
-    'img2ansi',
-    'img2ascii',
-    'read_ans',
-    'render_ans',
-    'render_font_char',
-    'render_font_str',
-    'reshape_ansi',
-    'scale_saturation',
-    'shuffle_char_set',
-    '_scaled_hu_moments',
-    'to_sgr_array',
-    'AnsiImage',
-    '_otsu_mask',
-]
-
 
 def get_font_key(font: FreeTypeFont):
     """Obtain a unique tuple pair from a FreeTypeFont object.
@@ -597,6 +598,7 @@ def img2ascii(
     return ascii_str
 
 
+@rgb_dispatch
 def img2ansi(
     __img: RGBImageLike | PathLike[str] | str,
     __font: FontArgType = 'arial.ttf',
@@ -605,7 +607,7 @@ def img2ansi(
     ansi_type: AnsiColorParam = DEFAULT_ANSI,
     sort_glyphs: bool | type[reversed] = True,
     equalize: bool | Literal['white_point'] = True,
-    bg: Color | Int3Tuple = (0, 0, 0),
+    bg: Color | Int3Tuple | str = (0, 0, 0),
 ):
     """Convert an image to an ANSI array.
 
@@ -665,7 +667,7 @@ def img2ansi(
     """
     if ansi_type is not DEFAULT_ANSI:
         ansi_type = get_ansi_type(ansi_type)
-    bg_wrapper = ColorStr('{}', color_spec={'bg': bg}, ansi_type=ansi_type, no_reset=True)
+    bg_wrapper = ColorStr('{}', color_spec={'bg': bg}, ansi_type=ansi_type, reset=False)
     base_ascii, color_arr = img2ascii(__img, __font, factor, char_set, sort_glyphs, ret_img=True)
     lines = base_ascii.splitlines()
     h, w = map(len, (lines, lines[0]))
@@ -697,7 +699,7 @@ def ascii2img(
     font_size=24,
     *,
     fg: Int3Tuple | str = (0, 0, 0),
-    bg: Int3Tuple | str = (255, 255, 255),
+    bg: Int3Tuple | str = (0xFF, 0xFF, 0xFF),
 ):
     """Render a literal string as an image.
 
@@ -803,7 +805,7 @@ def ansi2img(
         x_offset = 0
         for color_str in row:
             text_width = font.getbbox(color_str.base_str)[2]
-            if color_str._sgr_.is_reset():
+            if getattr(color_str, '_sgr_').is_reset():
                 fg_default = None
                 bg_default = input_bg
             if fg_color := getattr(color_str.fg, 'rgb', fg_default):
@@ -823,6 +825,38 @@ def ansi2img(
     return img
 
 
+def ansify(
+    __img: RGBImageLike | PathLike[str] | str,
+    /,
+    *,
+    font: FontArgType = UserFont.IBM_VGA_437_8X16,
+    font_size: int = 16,
+    factor: int = 200,
+    char_set: Iterable[str] = None,
+    sort_glyphs: bool | type[reversed] = True,
+    ansi_type: AnsiColorParam = DEFAULT_ANSI,
+    equalize: bool | Literal['white_point'] = True,
+    fg: Int3Tuple | str = (170, 170, 170),
+    bg: Int3Tuple | str | Literal['auto'] = (0, 0, 0),
+):
+    return ansi2img(
+        img2ansi(
+            __img,
+            font,
+            factor=factor,
+            char_set=char_set,
+            ansi_type=ansi_type,
+            sort_glyphs=sort_glyphs,
+            equalize=equalize,
+            bg=bg,
+        ),
+        font,
+        font_size=font_size,
+        fg_default=fg,
+        bg_default=bg,
+    )
+
+
 def _is_array(__obj: Any) -> TypeGuard[ndarray]:
     return isinstance(__obj, ndarray)
 
@@ -913,6 +947,7 @@ def to_sgr_array(__str: str, ansi_type: AnsiColorParam = DEFAULT_ANSI):
     resets = frozenset(resets_btok)
     pad_esc = {0x1B: '\x00\x1b'}
     x = []
+    sgr: SgrSequence | None
     for line in __str.splitlines():
         xs = []
         ansi_meta = {}
@@ -927,7 +962,7 @@ def to_sgr_array(__str: str, ansi_type: AnsiColorParam = DEFAULT_ANSI):
             else:
                 cs = new_cs(s)
             if sgr is None:
-                sgr = cs._sgr_
+                sgr: SgrSequence = getattr(cs, '_sgr_')
             if sgr.is_reset():
                 ansi_meta.clear()
             for k in resets.intersection(sgr.values()):
@@ -947,7 +982,7 @@ def to_sgr_array(__str: str, ansi_type: AnsiColorParam = DEFAULT_ANSI):
                     if not ansi_meta.get('bright'):
                         ansi_meta['bright'] = True
                     new_sgr = SgrSequence(
-                        sgr.values() + [p._value_ for p in prev._sgr_ if p.is_color()]
+                        sgr.values() + [p._value_ for p in getattr(prev, '_sgr_') if p.is_color()]
                     )
                     prev = cs = new_cs(cs.base_str, new_sgr)
             xs.append(cs.as_ansi_type(ansi_typ))
@@ -1009,9 +1044,10 @@ def read_ans[AnyStr: (str, bytes)](__path: PathLike[AnyStr] | AnyStr, encoding='
 class AnsiImage:
 
     @classmethod
-    def open[
-        AnyStr: (str, bytes)
-    ](
+    def open[AnyStr: (
+        str,
+        bytes,
+    )](
         cls,
         fp: PathLike[AnyStr] | AnyStr,
         shape: TupleOf2[int],
@@ -1076,7 +1112,7 @@ class AnsiImage:
         if not __table:
             return self
         table = {
-            k: (v if v not in frozenset(x for c in ' \t\n\r\v\f' for x in (c, ord(c))) else ' ')
+            k: v if v not in frozenset(x for c in ' \t\n\r\v\f' for x in (c, ord(c))) else ' '
             for (k, v) in __table.items()
             if k != ord('\n')
         }
@@ -1098,8 +1134,8 @@ class AnsiImage:
             return getattr(self, attr_name)[0]
         lines = []
         for line in self.data:
-            buffer = []
             if line:
+                buffer = []
                 initial = line[0]
                 for s in line[1:]:
                     if s.ansi_partition()[::2] == initial.ansi_partition()[::2]:
@@ -1109,7 +1145,7 @@ class AnsiImage:
                         initial = s
                 buffer.append(initial)
                 lines.append(''.join(buffer))
-        setattr(self, attr_name, ('\n'.join(lines) + SGR_RESET, self.shape))
+        setattr(self, attr_name, ('\n'.join(lines) + SGR_RESET.decode(), self.shape))
         return getattr(self, attr_name)[0]