mirror of
https://github.com/python/cpython.git
synced 2025-10-31 13:41:24 +00:00
f27f8c790a
Co-authored-by: Łukasz Langa <lukasz@langa.pl> Co-authored-by: Marta Gómez Macías <mgmacias@google.com> Co-authored-by: Lysandros Nikolaou <lisandrosnik@gmail.com> Co-authored-by: Hugo van Kemenade <1324225+hugovk@users.noreply.github.com> Co-authored-by: Jelle Zijlstra <jelle.zijlstra@gmail.com>
18 lines
411 B
Python
18 lines
411 B
Python
import re
|
|
import unicodedata
|
|
|
|
ANSI_ESCAPE_SEQUENCE = re.compile(r"\x1b\[[ -@]*[A-~]")
|
|
|
|
|
|
def str_width(c: str) -> int:
|
|
w = unicodedata.east_asian_width(c)
|
|
if w in ('N', 'Na', 'H', 'A'):
|
|
return 1
|
|
return 2
|
|
|
|
|
|
def wlen(s: str) -> int:
|
|
length = sum(str_width(i) for i in s)
|
|
|
|
# remove lengths of any escape sequences
|
|
return length - sum(len(i) for i in ANSI_ESCAPE_SEQUENCE.findall(s))
|