mirror of
https://github.com/pre-commit/pre-commit-hooks.git
synced 2026-04-07 20:26:54 +00:00
57 lines
1.3 KiB
Python
57 lines
1.3 KiB
Python
from __future__ import annotations
|
|
|
|
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
|