1
0

follow-up previous commit

This commit is contained in:
ThomasV
2019-03-05 17:37:01 +01:00
parent 46aa5c1958
commit 436c313790
2 changed files with 6 additions and 6 deletions

View File

@@ -521,16 +521,16 @@ class ChannelDB(PrintError):
old_policy.timestamp = new_policy.timestamp
@sql
def get_policy_for_node(self, node) -> Optional['Policy']:
def get_policy_for_node(self, channel_info, node) -> Optional['Policy']:
"""
raises when initiator/non-initiator both unequal node
"""
if node.hex() not in (self.node1_id, self.node2_id):
if node.hex() not in (channel_info.node1_id, channel_info.node2_id):
raise Exception("the given node is not a party in this channel")
n1 = self.DBSession.query(Policy).filter_by(short_channel_id = self.short_channel_id, start_node = self.node1_id).one_or_none()
n1 = self.DBSession.query(Policy).filter_by(short_channel_id = channel_info.short_channel_id, start_node = channel_info.node1_id).one_or_none()
if n1:
return n1
n2 = self.DBSession.query(Policy).filter_by(short_channel_id = self.short_channel_id, start_node = self.node2_id).one_or_none()
n2 = self.DBSession.query(Policy).filter_by(short_channel_id = channel_info.short_channel_id, start_node = channel_info.node2_id).one_or_none()
return n2
@sql