관리-도구

편집 파일: wcwidth.py

import unicodedata
from functools import lru_cache

@lru_cache(100)
def wcwidth(c: str) -> int:
    """Determine how many columns are needed to display a character in a terminal.

Returns -1 if the character is not printable.
    Returns 0, 1 or 2 for other characters.
    """
    o = ord(c)

# ASCII fast path.
    if 0x20 <= o < 0x07F:
        return 1

# Some Cf/Zp/Zl characters which should be zero-width.
    if (
        o == 0x0000
        or 0x200B <= o <= 0x200F
        or 0x2028 <= o <= 0x202E
        or 0x2060 <= o <= 0x2063
    ):
        return 0

category = unicodedata.category(c)

# Control characters.
    if category == "Cc":
        return -1

# Combining characters with zero width.
    if category in ("Me", "Mn"):
        return 0

# Full/Wide east asian characters.
    if unicodedata.east_asian_width(c) in ("F", "W"):
        return 2

return 1

def wcswidth(s: str) -> int:
    """Determine how many columns are needed to display a string in a terminal.

Returns -1 if the string contains non-printable characters.
    """
    width = 0
    for c in unicodedata.normalize("NFC", s):
        wc = wcwidth(c)
        if wc < 0:
            return -1
        width += wc
    return width