如何为你的脚本快速加上作者介绍和说明？附源码！

本文链接：https://blog.csdn.net/m0_67819674/article/details/137243944

如何为你的脚本快速加上作者介绍和说明？

文件批处理是我们经常需要用的，用python批量操作文件真的十分迅速与准确。下面有个案例：

我需要为我的.py文件添加一个作者介绍以及说明（包括时间、联系方式等），如何才能实现呢？

比如现在，我的每一篇源代码中，都需要有一个作者介绍，那我们是一个一个敲的吗？还是写一份复制粘贴的吗？其实都不是，这种重复的事情，作为程序员，我们应该写脚本，例如使用python进行文件的批处理，下面详细介绍一下：

例如：这是我的一段介绍：我需要把他加载到我的所有以.py文件结尾的第一行上。

"""
@contact: 1679538451@qq.com
@file: xxx(文件名).py
@time: (current_time) eg:2024/3/10 19：05
@author: 瑞翼工人
"""

附源码：把需要修改的文件的类型和作者介绍替换成您自己的

import os
import re
from datetime import datetime


def add_author_info(file_path, author_info):
    with open(file_path, 'r',encoding='utf-8') as file:
        content = file.read()

    # 判断是否已存在作者信息，如果已存在则直接返回
    if re.search(r'^\s*\"\"\"@contact:.*?\"\"\"', content, re.MULTILINE):
        return

    # 获取当前时间并格式化
    current_time = datetime.now().strftime("%Y/%m/%d %H:%M")
    # 将作者信息添加到文件开头
    with open(file_path, 'w',encoding='utf-8') as file:
        file.write(
            author_info.format(file_name=os.path.basename(file_path), current_time=current_time) + '\n\n' + content)


def main(directory_path, author_info):
    # 获取目录下所有的.py文件
    for root, dirs, files in os.walk(directory_path):
        for file in files:
            if file.endswith('.py'):
                file_path = os.path.join(root, file)
                add_author_info(file_path, author_info)


if __name__ == "__main__":
    # 定义作者信息
    author_info = """
\"\"\"
@contact: 1679538451@qq.com
@file: {file_name}
@time: {current_time}
@author: 瑞翼工人
\"\"\"
"""
    # 指定目录路径
    directory_path = '/'#更换成你的程序所在的目录
    # 调用主函数
    main(directory_path, author_info)