AI原生开发:ClaudeCode与国产大模型协同实战 第6章:解剖祖传屎山——让替你吃下读懂烂代码的苦

# 严重的状态耦合、魔数、面条逻辑global_db = {}def process_order(oid, op, ext):    o = global_db.get(oid)    if not o:        if op == 1: # 创建            global_db[oid] = {'s': 1, 'amt': ext.get('amt', 0), 'paid': 0}            return 100 # 成功        return 404 # 找不到    if o['s'] == 1: # 待支付        if op == 2: # 支付            if ext.get('pay_amt', 0) >= o['amt']:                o['s'] = 2                o['paid'] = ext.get('pay_amt', 0)                return 100            else:                return 501 # 金额不足        elif op == 5: # 取消            o['s'] = 5            return 100        else:            return 500 # 非法操作    elif o['s'] == 2: # 已支付        if op == 3: # 发货            o['s'] = 3            o['track'] = ext.get('track', '')            return 100        elif op == 6: # 退款            o['s'] = 6            return 100        else:            return 500    elif o['s'] == 3: # 已发货        if op == 4: # 确认收货            o['s'] = 4            return 100        else:            return 500    return 500 # 兜底异常

# Role你是一个精通重构的代码外科医生。# Task我接手了一段祖传Python代码 legacy_order.py，里面充满了魔数和面条逻辑。我需要你帮我：1. 梳理出这段代码的完整业务逻辑，列出所有状态(s)和操作(op)的含义。2. 使用 Mermaid 语法画出状态流转图。3. 使用状态模式重构这段代码，消除 if-else 和魔数，要求代码分层、可读性极佳。# 原始代码（把上面的 legacy_order.py 代码完整粘贴在这里）# Directive- 重构后的代码必须保持原有的业务逻辑100%不变。- 必须使用枚举类替代魔数。- 必须使用状态模式。- 必须保证我能直接运行，包含测试验证逻辑不变。

from enum import Enum, autofrom dataclasses import dataclassfrom typing import Dict, Optional# ================== 1. 消除魔数，定义清晰枚举 ==================class OrderStatus(Enum):    PENDING = 1      # 待支付    PAID = 2         # 已支付    SHIPPED = 3      # 已发货    COMPLETED = 4    # 已完成    CANCELLED = 5    # 已取消    REFUNDING = 6    # 退款中class Operation(Enum):    CREATE = 1       # 创建    PAY = 2          # 支付    SHIP = 3         # 发货    CONFIRM = 4      # 确认收货    CANCEL = 5       # 取消    REFUND = 6       # 退款class ResultCode(Enum):    SUCCESS = 100    NOT_FOUND = 404    AMOUNT_INSUFFICIENT = 501    INVALID_OPERATION = 500# ================== 2. 定义数据结构 ==================@dataclassclass Order:    oid: str    status: OrderStatus = OrderStatus.PENDING    amount: float = 0.0    paid: float = 0.0    track_no: str = ""# ================== 3. 状态模式：将每个状态的行为封装 ==================class OrderContext:    def __init__(self):        self.db: Dict[str, Order] = {}    def process(self, oid: str, op: Operation, ext: dict) -> ResultCode:        # 创建逻辑特殊处理        if op == Operation.CREATE:            if oid in self.db:                return ResultCode.INVALID_OPERATION            self.db[oid] = Order(oid=oid, amount=ext.get('amt', 0))            return ResultCode.SUCCESS        order = self.db.get(oid)        if not order:            return ResultCode.NOT_FOUND        # 路由到具体的状态处理方法        handler = {            OrderStatus.PENDING: self._handle_pending,            OrderStatus.PAID: self._handle_paid,            OrderStatus.SHIPPED: self._handle_shipped,        }.get(order.status)        if not handler:            return ResultCode.INVALID_OPERATION # 终态无法操作        return handler(order, op, ext)    def _handle_pending(self, order: Order, op: Operation, ext: dict) -> ResultCode:        if op == Operation.PAY:            if ext.get('pay_amt', 0) >= order.amount:                order.status = OrderStatus.PAID                order.paid = ext.get('pay_amt', 0)                return ResultCode.SUCCESS            return ResultCode.AMOUNT_INSUFFICIENT        elif op == Operation.CANCEL:            order.status = OrderStatus.CANCELLED            return ResultCode.SUCCESS        return ResultCode.INVALID_OPERATION    def _handle_paid(self, order: Order, op: Operation, ext: dict) -> ResultCode:        if op == Operation.SHIP:            order.status = OrderStatus.SHIPPED            order.track_no = ext.get('track', '')            return ResultCode.SUCCESS        elif op == Operation.REFUND:            order.status = OrderStatus.REFUNDING            return ResultCode.SUCCESS        return ResultCode.INVALID_OPERATION    def _handle_shipped(self, order: Order, op: Operation, ext: dict) -> ResultCode:        if op == Operation.CONFIRM:            order.status = OrderStatus.COMPLETED            return ResultCode.SUCCESS        return ResultCode.INVALID_OPERATION# ================== 4. 验证逻辑一致性 ==================if __name__ == "__main__":    ctx = OrderContext()    # 测试正常流转    print("--- 正常流转测试 ---")    assert ctx.process("O1", Operation.CREATE, {"amt": 100}) == ResultCode.SUCCESS    assert ctx.process("O1", Operation.PAY, {"pay_amt": 100}) == ResultCode.SUCCESS    assert ctx.process("O1", Operation.SHIP, {"track": "SF123"}) == ResultCode.SUCCESS    assert ctx.process("O1", Operation.CONFIRM, {}) == ResultCode.SUCCESS    print("正常流转通过！")    # 测试异常拦截    print("\n--- 异常拦截测试 ---")    assert ctx.process("O2", Operation.PAY, {}) == ResultCode.NOT_FOUND  # 不存在的订单    ctx.process("O2", Operation.CREATE, {"amt": 100})    assert ctx.process("O2", Operation.PAY, {"pay_amt": 50}) == ResultCode.AMOUNT_INSUFFICIENT # 金额不足    assert ctx.process("O2", Operation.SHIP, {}) == ResultCode.INVALID_OPERATION # 未支付不能发货    print("异常拦截通过！")

#!/bin/bash# 防止AI越界修改核心文件的 Git Hook# 定义允许修改的文件白名单（比如我们只允许重构这个订单模块）ALLOWED_FILES=("legacy_order.py" "modern_order.py" "test_order.py")# 获取本次提交修改的文件列表CHANGED_FILES=$(git diff --cached --name-only --diff-filter=ACM)for file in $CHANGED_FILES; do    echo "检查文件: $file"    allowed=false    for allowed_file in "${ALLOWED_FILES[@]}"; do        if [[ "$file" == *"$allowed_file"* ]]; then            allowed=true            break        fi    done    if [ "$allowed" = false ]; then        echo "❌ 错误: 尝试修改受限文件 '$file'！"        echo "AI可能越界了，本次提交被拦截。请检查代码！"        exit 1    fidoneecho "✅ 文件检查通过，允许提交。"exit 0