【Python】通过注释插桩替换代码实现开源自动化

需求提出

在特定的标签注释后写上开源后的代码实现开源替换

答疑解惑

调用如下的代码即可实现

def replace_java_code_in_one_line_by_tag(patch_file_path, update_java_code_line_tag):
    """
    本方法对包含update_java_code_line_tag的之前本行内所有内容进行删除操作;
    适用于对java文件的代码替换，即在包含update_java_code_line_tag之后的该行注释内填写本行应该被替换的代码；
    :param patch_file_path:处理文件路径
    :param update_java_code_line_tag:特定字符
    :return:
    """
    print ("handle_update_java_code_line_tag {} for {}".format(update_java_code_line_tag, patch_file_path))

    with open(patch_file_path, 'r') as patch_file:
        content = patch_file.readlines()

    for (index, line) in enumerate(content):
        if update_java_code_line_tag in line :
            print ("find {} in line {}".format(update_java_code_line_tag, index))
            replace_content_start_pos = line.find(update_java_code_line_tag) + len(update_java_code_line_tag)
            print_green("原有代码:{}\n修改后的代码:{}".format(line[0:line.find(update_java_code_line_tag)],line[replace_content_start_pos:]))
            content[index] = line[replace_content_start_pos:]
    with open(patch_file_path, 'w+') as f:
        f.writelines(content)
    # todo
def replace_all_java_file_code(source_demo_root_path, update_java_code_line_tag):
    """
    本方法是对所有的java文件进行修改单行代码替换操作。
    主要依托全局java文件遍历调用replace_java_code_in_one_line_by_tag方法实现上述需求。
    :param source_demo_root_path:全局搜索目录路径
    :param update_java_code_line_tag: 修改代码特定标签
    :return:
    """
    for root, dirs, files in os.walk(source_demo_root_path,topdown=True):
        # root 表示当前正在访问的文件夹路径
        # dirs 表示该文件夹下的子目录名list
        # files 表示该文件夹下的文件list
        # 遍历文件
        for f in files:
            if f.endswith(".java"):
                replace_java_code_in_one_line_by_tag(os.path.join(root, f),update_java_code_line_tag)