def read(cls, fp, version=1, padding=1):
signature = read_fmt('4s', fp)[0]
if signature not in cls._SIGNATURES:
logger.warning('Invalid signature (%r)' % (signature))
fp.seek(-4, 1)
return None
key = read_fmt('4s', fp)[0]
try:
key = Tag(key)
except ValueError:
message = 'Unknown key: %r' % (key)
warn(message)
logger.warning(message)
fmt = cls._length_format(key, version)
raw_data = read_length_block(fp, fmt=fmt, padding=padding)
kls = TYPES.get(key)
if kls:
data = kls.frombytes(raw_data, version=version)
# _raw_data = data.tobytes(version=version,
# padding=1 if padding == 4 else 4)
# assert raw_data == _raw_data, '%r: %s vs %s' % (
# kls, trimmed_repr(raw_data), trimmed_repr(_raw_data)
# )
else:
message = 'Unknown tagged block: %r, %s' % (key,
trimmed_repr(raw_data))
warn(message)
logger.warning(message)
data = raw_data
return cls(signature, key, data)
def _repr_pretty_(self, p, cycle):
if cycle:
return '{{...}'
with p.group(2, '{', '}'):
p.breakable('')
for idx, key in enumerate(self._items):
if idx:
p.text(',')
p.breakable()
value = self._items[key]
try:
p.text(Tag(key).name)
except ValueError:
p.pretty(key)
p.text(': ')
if isinstance(value.data, bytes):
p.text(trimmed_repr(value.data))
else:
p.pretty(value.data)
p.breakable('')