Python实现统计文本中的字符数量

import string  # string内置模块
import os  # os文件操作模块

def count_string(str_=None):
    count_en = count_dg = count_sp = count_zh = count_pu = 0
    if str_ is None:
        print("字符串str_为空！")
        return count_en, count_dg, count_sp, count_zh, count_pu
    for st in str_:
        # 英文
        if st in string.ascii_letters:
            count_en += 1
        # 数字
        elif st.isdigit():
            count_dg += 1
        # 空格
        elif st.isspace():
            count_sp += 1
        # 中文
        elif st.isalpha():
            count_zh += 1
        # 特殊字符
        else:
            count_pu += 1
    print("字符串str_：", str_)
    return count_en, count_dg, count_sp, count_zh, count_pu

def read_text(text_path=None):
    count_en_all = count_dg_all = count_sp_all = count_zh_all = count_pu_all = 0
    if text_path is None:
        print('文本文档的路径不能为空！')
        return

    if not os.path.isfile(text_path):
        print('文本文档的路径不存在，请重新输入！')
        return

    with open(text_path, 'r', encoding='utf-8') as file_:
        for line_ in file_.readlines():
            if line_.strip() != '':
                count_en, count_dg, count_sp, count_zh, count_pu = count_string(line_)
                count_en_all = count_en_all + count_en
                count_dg_all = count_dg_all + count_dg
                count_sp_all = count_sp_all + count_sp
                count_zh_all = count_zh_all + count_zh
                count_pu_all = count_pu_all + count_pu

    print('当前文本文档{}，结果统计如下：'.format(text_path))
    print('英文字符：', count_en_all)
    print('数字：', count_dg_all)
    print('空格：', count_sp_all)
    print('中文：', count_zh_all)
    print('特殊字符：', count_pu_all)

read_text('D:/test-data-work/data.txt')

# 当前文本文档D:/test-data-work/data.txt，结果统计如下：
# 英文字符：108
# 数字：209
# 空格：53
# 中文：288
# 特殊字符：90

for file_name in os.listdir('dir_path'):
    pass