获取具有字符串部分名称的所有类

import os

def get_classes_with_partial_name(search_string, directory):
    classes = []
    for root, dirs, files in os.walk(directory):
        for file in files:
            if file.endswith(".py"):  # 仅搜索Python代码文件
                with open(os.path.join(root, file), 'r') as f:
                    content = f.read()
                    class_lines = [line.strip() for line in content.split('\n') if line.strip().startswith('class')]
                    matching_classes = [line.split(' ')[1] for line in class_lines if search_string in line.split(' ')[1]]
                    classes.extend(matching_classes)
    return classes

search_string = "partial_name"
directory = "path/to/code/directory"
results = get_classes_with_partial_name(search_string, directory)
print(results)