本文教你如何通过合并职责、消除冗余判断和善用文件写入模式，将分散的文件存在性检查与创建逻辑整合为单一、可读性强且不易出错的函数。
本文教你如何通过合并职责、消除冗余判断和善用文件写入模式，将分散的文件存在性检查与创建逻辑整合为单一、可读性强且不易出错的函数。
在实际开发中，函数设计不应仅满足“能运行”，更要追求职责单一、逻辑内聚、边界清晰。你原始代码中将“检查文件是否存在 + 询问用户 + 删除旧文件”与“创建文件”拆分为两个独立函数（check_credentials_file 和 create_credentials_file），虽结构分明，却带来三类典型问题：
- 流程割裂：主逻辑（main）需显式串联两步，增加调用复杂度；
- 语义冗余：os.remove() 后立即用 "w" 模式重写，实属多余——"w" 本就覆盖原内容；
- 状态泄露：check_credentials_file 返回布尔值，但其副作用（删除文件）与返回值语义不一致，易引发误用。
✅ 更优解是将用户交互、存在性判断与文件写入统一收束于一个函数内，既减少外部依赖，又提升可维护性。以下是重构后的专业实践：
import os
def create_credentials_file(user, password, domain, filepath):
"""
创建或覆写凭证文件。
若文件已存在，则提示用户确认是否覆写；否决后直接退出。
使用 'w' 模式自动覆盖，无需手动删除。
"""
if os.path.exists(filepath):
choice = input(f"文件 '{filepath}' 已存在，是否覆写？(y/N): ").strip().lower()
if choice != "y":
print("操作已取消，未创建/修改文件。")
return False # 明确返回操作结果
try:
with open(filepath, "w", encoding="utf-8") as f:
f.write(f"username={user}\n")
f.write(f"password={password}\n")
f.write(f"domain={domain}\n")
print(f"✅ 凭证文件已成功写入：{filepath}")
return True
except OSError as e:
print(f"❌ 写入失败：无法访问路径 '{filepath}' — {e}")
return False
# 调用示例（简洁无分支）
if __name__ == "__main__":
create_credentials_file(
user=samba_user,
password=samba_pass,
domain=samba_domain,
filepath=credentials_filepath
)? 关键改进点说明：
- 消除中间状态：不再分离“检查”与“创建”，避免 main() 中隐含的控制流耦合；
- 利用 "w" 语义：Python 的 "w" 模式天然具备覆写能力，os.remove() 完全冗余，删除后反而增加异常风险（如删完瞬间文件被其他进程创建）；
- 增强健壮性：添加 encoding="utf-8" 防止中文或特殊字符乱码；对 OSError（比 IOError 更准确）捕获并给出上下文明确的错误提示；
- 返回语义化结果：函数返回 True/False 表达操作成败，便于上层扩展（如日志记录、重试机制）；
- 用户交互前置：在可能产生副作用（覆写）前即时确认，符合最小惊讶原则（Principle of Least Astonishment）。
? 延伸建议：
- 若需更高安全性，应避免明文存储密码，改用加密配置或系统密钥管理服务；
- 对生产环境，推荐使用 pathlib.Path 替代 os.path，代码更现代、可读性更强；
- 将用户输入逻辑抽离为独立函数（如 ask_confirmation(prompt)），便于单元测试与国际化。
好的函数逻辑，不是写得最多，而是用最少的代码、最直白的意图、最稳的边界处理，完成最确定的事。
