将所有函数名写入csv文件

import csv
import os

# 遍历代码库
def traverse_codebase(directory):
    function_names = []
    for root, dirs, files in os.walk(directory):
        for file in files:
            if file.endswith(".py"):  # 假设代码库中的文件为Python文件
                file_path = os.path.join(root, file)
                function_names.extend(extract_function_names(file_path))
    return function_names

# 提取函数名
def extract_function_names(file_path):
    function_names = []
    # 在这里实现提取函数名的逻辑，可以使用正则表达式、语法解析器等方法
    # 示例代码仅作为演示，实际实现需要根据具体语言和代码结构进行调整
    with open(file_path, "r") as file:
        lines = file.readlines()
        for line in lines:
            if "def " in line:
                function_name = line.split("def ")[1].split("(")[0]
                function_names.append(function_name)
    return function_names

# 创建CSV文件并写入函数名
def write_function_names_to_csv(function_names, csv_file):
    with open(csv_file, "w", newline="") as file:
        writer = csv.writer(file)
        writer.writerow(["Function Name"])  # 写入表头
        for function_name in function_names:
            writer.writerow([function_name])

# 主函数
def main():
    codebase_directory = "/path/to/codebase"  # 替换为实际的代码库目录
    csv_file = "/path/to/functions.csv"  # 替换为实际的CSV文件路径
    function_names = traverse_codebase(codebase_directory)
    write_function_names_to_csv(function_names, csv_file)
    print("All function names have been written to the CSV file.")

if __name__ == "__main__":
    main()