import os
import argparse


def add_prefix_to_files(directory, prefix):
 """
 对指定目录下的所有文件添加前缀。

 :param directory: 文件夹路径
 :param prefix: 要添加的前缀
 """
 for filename in os.listdir(directory):
 file_path = os.path.join(directory, filename)

 # 检查是否为文件
 if os.path.isfile(file_path):
 new_filename = f"{prefix}{filename}"
 new_file_path = os.path.join(directory, new_filename)

 # 重命名文件
 os.rename(file_path, new_file_path)
 print(f"Renamed '{filename}' to '{new_filename}'")


def main():
 parser = argparse.ArgumentParser(description="Add prefix to all files in a directory")
 parser.add_argument("directory", type=str, help="The directory containing files to rename")
 parser.add_argument("prefix", type=str, help="The prefix to add to files")
 args = parser.parse_args()

 # 调用函数
 add_prefix_to_files(args.directory, args.prefix)


if __name__ == "__main__":
 main()