python实现wc

usage = "usage: %prog [options] arg1 arg2..."
parser = OP(usage)

# 往OptionParser对象中添加Options
parser.add_option(‘-l‘, ‘--line‘,# 对应-l 选项
                  dest=‘lines‘,
                  action=‘store_true‘,
                  default=False,
                  help=‘counts lines only‘
                  )
parser.add_option(‘-c‘, ‘--chars‘,# 对应 -c选项
                  dest=‘chars‘,
                  action=‘store_true‘,
                  default=False,
                  help=‘counts chars only‘
                  )
parser.add_option(‘-w‘, ‘--words‘,# 对应-w选项
                  dest=‘words‘,
                  action=‘store_true‘,
                  default=False,
                  help=‘counts words only‘
                  )
parser.add_option(‘-a‘, ‘--another‘,# 对应-a选项
                  dest=‘more‘,
                  action=‘store_true‘,
                  default=False,
                  help=‘use for counting the numbers of comment lines,empty lines and char line‘
                  )
#  如果使用-g参数，会保存相关数据到SVG文件
parser.add_option(‘-g‘, ‘--graph‘,#对应-g选项
                  dest=‘graph‘,
                  action=‘store_true‘,
                  default=False,
                  help=‘use for describing data by graph.svg‘
                  )
# 调用optionparser的解析函数

(options, args) = parser.parse_args()

# 以下是判断路径为文件或者目录的函数
def is_not_dir(files):
    if not os.path.exists(files):# 判断路径是否存在
        sys.stderr.write("%s No such file or directory\n" % files)
        return 0
    if os.path.isdir(files):# 判断是否为目录
        sys.stderr.write("%s Is a directory\n" % files)
        return
    return 2

def read_file(file):
    with open(file, ‘r‘) as f:
        l = 0
        c = 0
        w = 0
        comment_lines = 0
        space_lines = 0
        char_lines = 0
        for line in f.readlines():#对每行文本分别处理
            l += 1
            c += len(line.strip())
            # 统计单词数
            match_list = re.findall(‘[a-zA-Z]+‘, line.lower())
            w += len(match_list)# 统计字符数，不算前后的空格
            # to judge if a line is a comment line
            line = line.strip()#剥除前后的空格
            if line.startswith("#") or line.startswith(r‘"""‘) or line.endswith("#") or line.endswith(r‘"""‘):#注释行判断条件
                comment_lines += 1
            elif not line:
                space_lines += 1
            else:
                char_lines += 1
        print(‘the current file is: %s‘ % f)

        l -= 1# 不算最好一行空白行
        # 打印数据
        show_data(l, c, w, comment_lines, space_lines, char_lines)
        # 使用列表返回行数，注释行，空白行，字符行以便于测试
        return list(l, comment_lines, space_lines, char_lines)

def read_files(files):
    file_list = []#文件路径列表
    for file in files:# 每个文件单独处理
        judge = is_not_dir(file)
        if judge == 2: # 此路径为文件
            # use the function of with open to automatically open and close files
            read_file()
        elif judge == 1: #此路径为目录
            for item in os.listdir(file):
                item_path = os.path.join(file, item) #合并路径
                file_list.append(item_path)# 添加到文件列表
            read_files(file_list)
        else:
            continue

    if len(files) > 1:  # 如果有多个文件，则输出总数
        print(‘the total number is:‘)
        show_data(sum_l, sum_c, sum_w, sum_comment, sum_space, sum_char_line)

def use_pygal(x_list, y_list, svg_name):
    # 设置风格
    my_style = LS(‘#333366‘, base_style=LCS)
    my_config = pygal.Config()
    my_config.show_legend = False
    # 坐标旋转25度
    my_config.x_label_rotation = 25
    my_config.title_font_size = 24
    # x,y轴坐标大小
    my_config.label_font_size = 16
    # 主坐标大小
    my_config.major_label_font_size = 18
    my_config.show_y_guides = True
    # 设置width 充分利用浏览器界面
    my_config.width = 1000

    # 使用pygal绘制直方图
    chart = pygal.Bar(my_config, style=my_style)
    chart.title = "source file‘s data of wordcount"
    chart.x_labels = x_list

    chart.add(‘‘, y_list)

    # 保存svg图
    chart.render_to_file(svg_name)

def show_data(ll, cc, ww, cl, sl, chl):
    global sum_l #总行数
    global sum_c
    global sum_w
    global sum_comment #总注释行数
    global sum_space
    global sum_char_line
　　
    #记录总数据

    sum_l += ll
    sum_c += cc
    sum_w += ww
    sum_char_line += chl
    sum_space += sl
    sum_comment += cl

    if not (options.words or options.lines or options.chars):#如果不使用参数，则默认输出行数，单词数，字符数
        print(ll)
        print(ww)
        print(cc)
    if options.words:
        print(ww)
    if options.lines:
        print(ll)
    if options.chars:
        print(cc)
    if options.more:
        print("the number of comment lines is:", cl)
        print("the number of empty lines is:", sl)
        print("the number of char lines is:", chl)
    if options.graph:
        # 列表用以构造svg图
        x_list_arg = [‘lines number‘, ‘words number‘, ‘chars number‘, ‘comment lines number‘, ‘empty lines number‘,
                      ‘chars lines number‘]
        y_list_arg = [ll, ww, cc, cl, sl, chl]
        use_pygal(x_list_arg, y_list_arg, svg_name=‘word_count.svg‘)#绘图