如何通过Python根据用户输入生成删除.config/nvim的Bash脚本

你可以通过判断用户输入、生成Bash脚本内容并写入文件来完成需求，下面是修改后的完整Python代码：

warning = input("Do you want to remove .config/nvim to install nvchad? ").strip().lower()
if warning == "yes":
    # 定义Bash脚本内容，先检查目录存在再删除
    bash_script = """#!/bin/bash
NVIM_CONFIG_DIR="$HOME/.config/nvim"
if [ -d "$NVIM_CONFIG_DIR" ]; then
    rm -rf "$NVIM_CONFIG_DIR"
    echo "已成功删除 $NVIM_CONFIG_DIR"
else
    echo "$NVIM_CONFIG_DIR 不存在，无需删除"
fi
"""
    # 将脚本写入文件
    with open("clean_nvim_config.sh", "w") as script_file:
        script_file.write(bash_script)
    
    # 可选：给脚本添加执行权限，方便直接运行
    import os
    os.chmod("clean_nvim_config.sh", 0o755)
    
    print("已生成clean_nvim_config.sh脚本，可通过./clean_nvim_config.sh执行")
else:
    print("操作已取消")

关键细节说明 #

• 对用户输入做strip().lower()处理，兼容大小写（如"YES"、" Yes"）和前后空格的情况
• Bash脚本里先判断目标目录是否存在，避免直接执行rm -rf时出现无意义的错误提示
• 使用with open语句确保文件安全写入，自动管理文件句柄
• 可选的os.chmod步骤给脚本添加执行权限，用户无需手动执行chmod +x

内容的提问来源于stack exchange，提问作者user23505753