Skip to content

Commit 4e65ff7

Browse files
fix
1 parent 2b8e48c commit 4e65ff7

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

examples/src/branch_key_id_supplier_example.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -25,15 +25,17 @@ def get_branch_key_id(
2525
"""Returns branch key ID from the tenant ID in input's encryption context."""
2626
encryption_context: Dict[str, str] = param.encryption_context
2727

28-
if b"tenant" not in encryption_context:
28+
print(f"{encryption_context=}")
29+
30+
if "tenant" not in encryption_context:
2931
raise ValueError("EncryptionContext invalid, does not contain expected tenant key value pair.")
3032

31-
tenant_key_id: str = encryption_context.get(b"tenant")
33+
tenant_key_id: str = encryption_context.get("tenant")
3234
branch_key_id: str
3335

34-
if tenant_key_id == b"TenantA":
36+
if tenant_key_id == "TenantA":
3537
branch_key_id = self.branch_key_id_for_tenant_A
36-
elif tenant_key_id == b"TenantB":
38+
elif tenant_key_id == "TenantB":
3739
branch_key_id = self.branch_key_id_for_tenant_B
3840
else:
3941
raise ValueError(f"Item does not contain valid tenant ID: {tenant_key_id=}")

0 commit comments

Comments
 (0)