上下文隔离与预算分配

mkdir l6-context
cd l6-context

"""按角色和租户构造受预算约束的资料包。"""
from dataclasses import dataclass

@dataclass(frozen=True)
class Record:
    id: str
    tenant: str
    roles: frozenset
    estimate: int
    priority: int
    text: str

RECORDS = [
    Record("policy", "shared", frozenset({"planner", "reviewer"}), 5, 1, "未发货可退已付金额"),
    Record("order", "shop-A", frozenset({"planner", "executor"}), 4, 2, "O-17已付120元"),
    Record("stock", "shop-A", frozenset({"executor"}), 5, 3, "O-17尚未发货"),
    Record("other-customer", "shop-B", frozenset({"executor"}), 3, 1, "另一客户订单"),
]
GOAL = {"text": "核查O-17退款资格；禁止直接付款", "estimate": 3}

def build_context(role, tenant, budget, records=RECORDS):
    if type(budget) is not int or budget < GOAL["estimate"]:
        raise ValueError("预算不足以容纳目标和限制")
    selected, used, skipped = [], GOAL["estimate"], []
    visible = sorted((row for row in records
                      if row.tenant in {tenant, "shared"} and role in row.roles),
                     key=lambda row: (row.priority, row.id))
    # 必须信息先占额度，可选资料只能使用剩余空间。
    for row in visible:
        if used + row.estimate <= budget:
            selected.append(row)
            used += row.estimate
        else:
            skipped.append(row.id)
    return {"goal": GOAL["text"], "records": selected, "used": used, "skipped": skipped}

def show(role, budget):
    package = build_context(role, "shop-A", budget)
    ids = ",".join(row.id for row in package["records"]) or "无"
    skipped = ",".join(package["skipped"]) or "无"
    return f"{role}：资料={ids}；估算额度={package['used']}/{budget}；因预算略去={skipped}"

def main():
    print(show("planner", 12))
    print(show("executor", 12))
    print(show("executor", 7))
    print("必须信息：" + GOAL["text"])
    has_other = any(row.id == "other-customer" for row in build_context("executor", "shop-A", 1000)["records"])
    print("隔离检查：shop-A资料包" + ("含有" if has_other else "没有") + "other-customer")

if __name__ == "__main__":
    main()

ls
python -X utf8 context_demo.py

planner：资料=policy,order；估算额度=12/12；因预算略去=无
executor：资料=order,stock；估算额度=12/12；因预算略去=无
executor：资料=order；估算额度=7/7；因预算略去=stock
必须信息：核查O-17退款资格；禁止直接付款
隔离检查：shop-A资料包没有other-customer

python -X utf8 -m unittest -v test_context_demo.py