electrum

commit 55a0043ab77ed6a59d27b675847d9dd5101c9e55
parent 97900c09851e14fe0b5137b16cfb9b58fb4d4fd0
Author: ThomasV <thomasv@electrum.org>
Date:   Fri, 28 Feb 2020 10:34:04 +0100

follow-up previous commit (fix tests)

Diffstat:
M
electrum/channel_db.py
 | 
9
++++++---

1 file changed, 6 insertions(+), 3 deletions(-)
diff --git 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)