让python根据用户输入自动命名文件

import os

def create_file_with_user_input():
    # 获取用户输入的文件名
    file_name = input("请输入文件名（不包括扩展名）：")
    
    # 添加文件扩展名
    file_name_with_extension = f"{file_name}.txt"
    
    # 检查文件是否已经存在
    if os.path.exists(file_name_with_extension):
        print(f"文件 '{file_name_with_extension}' 已经存在。")
    else:
        # 创建并写入文件
        with open(file_name_with_extension, 'w') as file:
            file.write("这是一个自动命名的文件。\n")
        print(f"文件 '{file_name_with_extension}' 已创建。")

if __name__ == "__main__":
    create_file_with_user_input()

python create_file.py

import os

def create_file_with_user_input():
    # 获取用户输入的文件名
    file_name = input("请输入文件名（不包括扩展名）：")
    
    # 添加文件扩展名
    file_name_with_extension = f"{file_name}.txt"
    
    # 检查文件是否已经存在，如果存在则生成一个不冲突的文件名
    counter = 1
    original_file_name = file_name_with_extension
    while os.path.exists(file_name_with_extension):
        file_name_with_extension = f"{file_name}_{counter}.txt"
        counter += 1
    
    # 创建并写入文件
    with open(file_name_with_extension, 'w') as file:
        file.write("这是一个自动命名的文件。\n")
    print(f"文件 '{file_name_with_extension}' 已创建。")

if __name__ == "__main__":
    create_file_with_user_input()