Pythonで文字列のバイト数を簡単に計算する方法（lenb関数）

import unicodedata

def lenb(countstring):
    """
    文字列の長さをバイト数でカウントする関数

    Args:
        countstring (str): カウントする文字列

    Returns:
        int: 文字列のバイト数。全角文字は2として半角文字は1としてカウント

    Example:
        >>> lenb('abc')  # 全て半角文字
        3
        >>> lenb('こんにちは')  # 全て全角文字
        10
        >>> lenb('abcこんにちは')  # 半角と全角の混在
        13
    """

    length = 0

    # 1文字ずつ処理
    for char in countstring:
        # unicodedata.east_asian_widthを用いて文字が全角('F', 'W', 'A')か半角（それ以外）を判断
        if unicodedata.east_asian_width(char) in 'FWA':
            length += 2  # 全角文字の場合、2バイトとして加算
        else:
            length += 1  # 半角文字の場合、1バイトとして加算

    return length




##############################################################
# 関数のテスト
if __name__ == "__main__":

    print("\n数字")
    text = "0123456789"
    print("len  ", len(text))
    print("lenb ", lenb(text))

    print("\n全角数字")
    text = "０１２３４５６７８９"
    print("len  ", len(text))
    print("lenb ", lenb(text))

    print("\nかな")
    text = "ｱｲｳｴｵｶｷｸｹｺ"
    print("len  ", len(text))
    print("lenb ", lenb(text))

    print("\n全角かな")
    text = "あいうえおカキクケコ"
    print("len  ", len(text))
    print("lenb ", lenb(text))

    print("\n半角記号")
    text = "[]:;/.,#$~"
    print("len  ", len(text))
    print("lenb ", lenb(text))

    print("\n全角記号")
    text = "「」：；￥・。＃＄～"
    print("len  ", len(text))
    print("lenb ", lenb(text))

    print("\nローマ数字")
    text = "ⅠⅡⅢⅣⅤⅥⅦⅧⅨⅩ"
    print("len  ", len(text))
    print("lenb ", lenb(text))

    print("\nエスケープシーケンス")
    text = "\a\b\f\n\r\t\v\\\'\""
    print("len  ", len(text))
    print("lenb ", lenb(text))

数字
len   10
lenb  10

全角数字
len   10
lenb  20

かな
len   10
lenb  10

全角かな
len   10
lenb  20

半角記号
len   10
lenb  10

全角記号
len   10
lenb  20

ローマ数字
len   10
lenb  20

エスケープシーケンス
len   10
lenb  10