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