递归计算代码行数，包括压缩(zip)文件

import os
import zipfile

def count_code_lines(directory):
    total_lines = 0

    for root, dirs, files in os.walk(directory):
        for file in files:
            file_path = os.path.join(root, file)
            if file.endswith(('.py', '.java', '.cpp')):
                with open(file_path, 'r') as f:
                    lines = f.readlines()
                    code_lines = sum(1 for line in lines if line.strip() and not line.strip().startswith('#'))
                    total_lines += code_lines

    return total_lines

def count_code_lines_in_zip(zip_file):
    total_lines = 0

    with zipfile.ZipFile(zip_file, 'r') as z:
        for file_info in z.infolist():
            if not file_info.is_dir():
                file_extension = os.path.splitext(file_info.filename)[1]
                if file_extension.endswith(('.py', '.java', '.cpp')):
                    with z.open(file_info) as f:
                        lines = f.readlines()
                        code_lines = sum(1 for line in lines if line.strip() and not line.strip().startswith('#'))
                        total_lines += code_lines

    return total_lines

# 示例用法
directory_path = '/path/to/directory'
zip_file_path = '/path/to/zip_file.zip'

code_lines_in_directory = count_code_lines(directory_path)
code_lines_in_zip_file = count_code_lines_in_zip(zip_file_path)

print(f"代码行数（目录）: {code_lines_in_directory}")
print(f"代码行数（压缩文件）: {code_lines_in_zip_file}")