Applies the transfer guard level within transfer_guard_lib.
(state, key, val)
| 1215 | |
| 1216 | |
| 1217 | def _update_transfer_guard(state, key, val): |
| 1218 | """Applies the transfer guard level within transfer_guard_lib.""" |
| 1219 | if val is None: |
| 1220 | setattr(state, key, None) |
| 1221 | elif val == "allow": |
| 1222 | setattr(state, key, transfer_guard_lib.TransferGuardLevel.ALLOW) |
| 1223 | elif val == "log": |
| 1224 | setattr(state, key, transfer_guard_lib.TransferGuardLevel.LOG) |
| 1225 | elif val == "disallow": |
| 1226 | setattr(state, key, transfer_guard_lib.TransferGuardLevel.DISALLOW) |
| 1227 | elif val == "log_explicit": |
| 1228 | setattr(state, key, transfer_guard_lib.TransferGuardLevel.LOG_EXPLICIT) |
| 1229 | elif val == "disallow_explicit": |
| 1230 | setattr(state, key, transfer_guard_lib.TransferGuardLevel.DISALLOW_EXPLICIT) |
| 1231 | else: |
| 1232 | assert False, f"Invalid transfer guard level {val}" |
| 1233 | |
| 1234 | |
| 1235 | transfer_guard_host_to_device = config.define_enum_state( |