From 9d9592e9e1813a6259cc05686876536feaf4a050 Mon Sep 17 00:00:00 2001 From: Jamie Hardt Date: Sun, 5 Jan 2020 19:20:12 -0800 Subject: [PATCH] Update wave_reader.py output INFO and test for if these aren't present --- wavinfo/wave_reader.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/wavinfo/wave_reader.py b/wavinfo/wave_reader.py index 268c75e..f9abb7e 100644 --- a/wavinfo/wave_reader.py +++ b/wavinfo/wave_reader.py @@ -148,6 +148,12 @@ class WavInfoReader(): for field in attr._fields: yield scope, field, attr.__getattribute__(field) - bext_dict = self.bext.to_dict() - for key in bext_dict.keys(): - yield 'bext', key, bext_dict[key] + if self.bext is not None: + bext_dict = (self.bext or {}).to_dict() + for key in bext_dict.keys(): + yield 'bext', key, bext_dict[key] + + if self.info is not None: + info_dict = self.info.to_dict() + for key in info_dict.keys(): + yield 'info', key, info_dict[key]