From 854a2ddd8793b67c07b68a4f515ad981b7b42c94 Mon Sep 17 00:00:00 2001 From: Kris Nuttycombe Date: Mon, 14 Jun 2021 10:54:59 -0600 Subject: [PATCH] Allow unrecognized receivers. --- unified_addrs.py | 14 ++++---------- 1 file changed, 4 insertions(+), 10 deletions(-) diff --git a/unified_addrs.py b/unified_addrs.py index 7ddbfb4..87f5701 100644 --- a/unified_addrs.py +++ b/unified_addrs.py @@ -49,16 +49,10 @@ def decode_unified(addr_str): assert [1, 2, 3].count(b) > 0, "receiver type " + str(b) + " not recognized" s = 1 elif s == 1: - receiver_len = b - if receiver_type == 1: - assert receiver_len == 20 - elif receiver_type == 2: - assert receiver_len == 43 - elif receiver_type == 3: - assert receiver_len == 43 - else: - assert False, "incorrect receiver length" - s = 2 + receiver_len = b + expected_len == {1: 20, 2: 43, 3: 43}.get(receiver_type) + if expected_len is not None: + assert receiver_len == expected_len, "incorrect receiver length" elif s == 2: if len(acc) < receiver_len: acc.append(b)