Minor changes to satisfy type checker

lethe/LATEST
Jan Petykiewicz 4 years ago
parent c9adca61b6
commit 25b2cecec9

@ -218,8 +218,8 @@ class Record(metaclass=ABCMeta):
def read_refname(stream: io.BufferedIOBase, def read_refname(stream: io.BufferedIOBase,
is_present: bool, is_present: Union[bool, int],
is_reference: bool is_reference: Union[bool, int]
) -> Union[None, int, NString]: ) -> Union[None, int, NString]:
""" """
Helper function for reading a possibly-absent, possibly-referenced NString. Helper function for reading a possibly-absent, possibly-referenced NString.
@ -242,8 +242,8 @@ def read_refname(stream: io.BufferedIOBase,
def read_refstring(stream: io.BufferedIOBase, def read_refstring(stream: io.BufferedIOBase,
is_present: bool, is_present: Union[bool, int],
is_reference: bool is_reference: Union[bool, int],
) -> Union[None, int, AString]: ) -> Union[None, int, AString]:
""" """
Helper function for reading a possibly-absent, possibly-referenced `AString`. Helper function for reading a possibly-absent, possibly-referenced `AString`.
@ -937,18 +937,18 @@ class Property(Record):
s = 0x01 & (byte >> 0) s = 0x01 & (byte >> 0)
name = read_refname(stream, c, n) name = read_refname(stream, c, n)
values: Optional[List[property_value_t]]
if v == 0: if v == 0:
if u < 0x0f: if u < 0x0f:
value_count = u value_count = u
else: else:
value_count = read_uint(stream) value_count = read_uint(stream)
values = [read_property_value(stream) for _ in range(value_count)] values: Optional[List[property_value_t]] = [read_property_value(stream)
for _ in range(value_count)]
else: else:
values = None values = None
if u != 0: if u != 0:
raise InvalidDataError('Malformed property record header') raise InvalidDataError('Malformed property record header')
record = Property(name, values, s) record = Property(name, values, bool(s))
logger.debug('Record ending at 0x{:x}:\n {}'.format(stream.tell(), record)) logger.debug('Record ending at 0x{:x}:\n {}'.format(stream.tell(), record))
return record return record

Loading…
Cancel
Save