AaronFiore/electrum
1
0
Fork 0
Code Activity

lnmsg: pass filename as parameter instead of boolean

Browse Source
This commit is contained in:
ThomasV
2026-01-28 12:02:27 +01:00
parent cfe2a57f2c
commit e1103266e6
1 changed files with 4 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
electrum/lnmsg.py
View File

@@ -323,7 +323,7 @@ def _parse_msgtype_intvalue_for_onion_wire(value: str) -> int:
class LNSerializer:
def __init__(self, *, for_onion_wire: bool = False):
def __init__(self, *, name: str = 'peer_wire'):
# TODO msg_type could be 'int' everywhere...
self.msg_scheme_from_type = {} # type: Dict[bytes, List[Sequence[str]]]
self.msg_type_from_name = {} # type: Dict[str, bytes]
@@ -334,10 +334,7 @@ class LNSerializer:
self.subtypes = {} # type: Dict[str, Dict[str, Sequence[str]]]
if for_onion_wire:
path = os.path.join(os.path.dirname(__file__), "lnwire", "onion_wire.csv")
else:
path = os.path.join(os.path.dirname(__file__), "lnwire", "peer_wire.csv")
path = os.path.join(os.path.dirname(__file__), "lnwire", name + ".csv")
with open(path, newline='') as f:
csvreader = csv.reader(f)
for row in csvreader:
@@ -345,7 +342,7 @@ class LNSerializer:
if row[0] == "msgtype":
# msgtype,<msgname>,<value>[,<option>]
msg_type_name = row[1]
if for_onion_wire:
if name == 'onion_wire':
msg_type_int = _parse_msgtype_intvalue_for_onion_wire(str(row[2]))
else:
msg_type_int = int(row[2])
@@ -668,4 +665,4 @@ encode_msg = _inst.encode_msg
decode_msg = _inst.decode_msg
OnionWireSerializer = LNSerializer(for_onion_wire=True)
OnionWireSerializer = LNSerializer(name='onion_wire')