From 55a0043ab77ed6a59d27b675847d9dd5101c9e55 Mon Sep 17 00:00:00 2001 From: ThomasV Date: Fri, 28 Feb 2020 10:34:04 +0100 Subject: [PATCH] follow-up previous commit (fix tests) --- electrum/channel_db.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/electrum/channel_db.py b/electrum/channel_db.py index faaccc6bf..f6bc54417 100644 --- a/electrum/channel_db.py +++ b/electrum/channel_db.py @@ -339,7 +339,8 @@ class ChannelDB(SqlDB): self._channels[channel_info.short_channel_id] = channel_info self._channels_for_node[channel_info.node1_id].add(channel_info.short_channel_id) self._channels_for_node[channel_info.node2_id].add(channel_info.short_channel_id) - self.save_channel(channel_info.short_channel_id, msg['raw']) + if 'raw' in msg: + self.save_channel(channel_info.short_channel_id, msg['raw']) def print_change(self, old_policy: Policy, new_policy: Policy): # print what changed between policies @@ -397,7 +398,8 @@ class ChannelDB(SqlDB): self.verify_channel_update(payload) policy = Policy.from_msg(payload) self._policies[key] = policy - self.save_policy(policy.key, payload['raw']) + if 'raw' in payload: + self.save_policy(policy.key, payload['raw']) # self.update_counts() return CategorizedChannelUpdates( @@ -491,7 +493,8 @@ class ChannelDB(SqlDB): continue # save self._nodes[node_id] = node_info - self.save_node_info(node_id, msg_payload['raw']) + if 'raw' in msg_payload: + self.save_node_info(node_id, msg_payload['raw']) for addr in node_addresses: self._addresses[node_id].add((addr.host, addr.port, 0)) self.save_node_addresses(node_id, node_addresses)