Skip to content

Commit f810f47

Browse files
committed
pyln.proto.message: python-fluency feedback from @darosior
Signed-off-by: Rusty Russell <[email protected]>
1 parent 5fdd72a commit f810f47

File tree

2 files changed

+9
-17
lines changed

2 files changed

+9
-17
lines changed

contrib/pyln-proto/pyln/proto/message/array_types.py

Lines changed: 2 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -34,19 +34,11 @@ def val_to_str(self, v, otherfields):
3434
if self.elemtype.name == 'byte':
3535
return bytes(v).hex()
3636

37-
s = ''
38-
sep = ''
39-
for i in v:
40-
s += sep + self.elemtype.val_to_str(i, otherfields)
41-
sep = ','
42-
37+
s = ','.join(self.elemtype.val_to_str(i, otherfields) for i in v)
4338
return '[' + s + ']'
4439

4540
def val_to_bin(self, v, otherfields):
46-
b = bytes()
47-
for i in v:
48-
b += self.elemtype.val_to_bin(i, otherfields)
49-
return b
41+
return b''.join(self.elemtype.val_to_bin(i, otherfields) for i in v)
5042

5143
def arr_from_bin(self, bytestream, otherfields, arraysize):
5244
"""arraysize None means take rest of bytestream exactly"""

contrib/pyln-proto/pyln/proto/message/message.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@ def get_tlvtype(self, name):
6363

6464
def get_type(self, name):
6565
t = self.get_subtype(name)
66-
if not t:
66+
if t is None:
6767
t = self.get_tlvtype(name)
6868
return t
6969

@@ -178,7 +178,7 @@ def _field_from_csv(self, namespace, parts, ellipsisok=False, option=None):
178178
179179
"""
180180
basetype = namespace.get_type(parts[1])
181-
if not basetype:
181+
if basetype is None:
182182
raise ValueError('Unknown type {}'.format(parts[1]))
183183

184184
# Fixed number, or another field.
@@ -345,7 +345,7 @@ def field_from_csv(namespace, parts):
345345
raise ValueError("msgdata expected 4 CSV parts, not {}"
346346
.format(parts))
347347
messagetype = namespace.get_msgtype(parts[0])
348-
if not messagetype:
348+
if messagetype is None:
349349
raise ValueError("unknown subtype {}".format(parts[0]))
350350

351351
field = messagetype._field_from_csv(namespace, parts[1:4],
@@ -383,7 +383,7 @@ def type_from_csv(namespace, parts):
383383
raise ValueError("tlvtype expected 4 CSV parts, not {}"
384384
.format(parts))
385385
tlvstream = namespace.get_tlvtype(parts[0])
386-
if not tlvstream:
386+
if tlvstream is None:
387387
tlvstream = TlvStreamType(parts[0])
388388
namespace.add_tlvtype(tlvstream)
389389

@@ -400,7 +400,7 @@ def field_from_csv(namespace, parts):
400400
.format(parts))
401401

402402
tlvstream = namespace.get_tlvtype(parts[0])
403-
if not tlvstream:
403+
if tlvstream is None:
404404
raise ValueError("unknown tlvtype {}".format(parts[0]))
405405

406406
field = tlvstream.find_field(parts[1])
@@ -560,7 +560,7 @@ def from_bin(namespace, binmsg):
560560
off = 2
561561

562562
mtype = namespace.get_msgtype_by_number(typenum)
563-
if not mtype:
563+
if mtype is None:
564564
raise ValueError('Unknown message type number {}'.format(typenum))
565565

566566
fields = {}
@@ -578,7 +578,7 @@ def from_str(namespace, s, incomplete_ok=False):
578578
parts = s.split()
579579

580580
mtype = namespace.get_msgtype(parts[0])
581-
if not mtype:
581+
if mtype is None:
582582
raise ValueError('Unknown message type name {}'.format(parts[0]))
583583

584584
args = {}

0 commit comments

Comments
 (0)