重构前

import hashlib

import sqlite3

class AuthManager:

def __init__(self, db_path):

    self.db_path = db_path

def register_user(self, username, password):

    hashed = hashlib.sha256(password.encode()).hexdigest()

    conn = sqlite3.connect(self.db_path)

    cur = conn.cursor()

    cur.execute("INSERT INTO users (username, password) VALUES (?, ?)", (username, hashed))

    conn.commit()

    conn.close()

def login(self, username, password):

    conn = sqlite3.connect(self.db_path)

    cur = conn.cursor()

    cur.execute("SELECT password FROM users WHERE username=?", (username,))

    row = cur.fetchone()

    conn.close()

    if row:

        return row[0] == hashlib.sha256(password.encode()).hexdigest()

    return False

上述代码将哈希运算、数据访问和业务逻辑混杂在一起。让我们提示 Claude 进行增量重构：

"Claude，请重构此类，将数据库操作与认证逻辑分离。引入一个可复用的数据库工具函数，完善文档字符串，并保持原有行为不变。"

Claude 重构后的结果可能如下所示：