mirrors/core-lightning
1
0
Fork 0
mirror of https://github.com/ElementsProject/lightning.git synced 2025-01-17 19:03:42 +01:00
Code Issues Projects Releases Wiki Activity

pyln.proto.message: separate fundamental types from other subtypes.

Browse Source 
This will be useful for the next patch, which introduces per-bolt
modules.  This makes it easier for them generate variables for each
field type they parse (they don't want to export u16, for example)

Signed-off-by: Rusty Russell <rusty@rustcorp.com.au>
This commit is contained in:
Rusty Russell 2020-06-04 12:14:16 +09:30 committed by Christian Decker
parent e4b5679f5d
commit daa707d213
1 changed files with 14 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
contrib/pyln-proto/pyln/proto/message/message.py
View File

@ -11,12 +11,13 @@ class MessageNamespace(object):
domain, such as within a given BOLT"""
def __init__(self, csv_lines=[]):
self.subtypes = {}
self.fundamentaltypes = {}
self.tlvtypes = {}
self.messagetypes = {}
# For convenience, basic types go in every namespace
for t in fundamental_types():
self.add_subtype(t)
self.add_fundamentaltype(t)
self.load_csv(csv_lines)
@ -26,6 +27,10 @@ domain, such as within a given BOLT"""
return ValueError('Already have {}'.format(prev))
self.subtypes[t.name] = t
def add_fundamentaltype(self, t):
assert not self.get_type(t.name)
self.fundamentaltypes[t.name] = t
def add_tlvtype(self, t):
prev = self.get_type(t.name)
if prev:
@ -51,6 +56,11 @@ domain, such as within a given BOLT"""
return m
return None
def get_fundamentaltype(self, name):
if name in self.fundamentaltypes:
return self.fundamentaltypes[name]
return None
def get_subtype(self, name):
if name in self.subtypes:
return self.subtypes[name]
@ -62,7 +72,9 @@ domain, such as within a given BOLT"""
return None
def get_type(self, name):
t = self.get_subtype(name)
t = self.get_fundamentaltype(name)
if not t:
t = self.get_subtype(name)
if not t:
t = self.get_tlvtype(name)
return t