diff options
Diffstat (limited to 'scripts/update_payload/payload.py')
| -rw-r--r-- | scripts/update_payload/payload.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/scripts/update_payload/payload.py b/scripts/update_payload/payload.py index fe3a4502..6b28046e 100644 --- a/scripts/update_payload/payload.py +++ b/scripts/update_payload/payload.py @@ -124,8 +124,11 @@ class Payload(object): if zipfile.is_zipfile(payload_file): with zipfile.ZipFile(payload_file) as zfp: with zfp.open("payload.bin") as payload_fp: - payload_file = io.BytesIO(payload_fp.read()) - self.payload_file = payload_file + self.payload_file = io.BytesIO(payload_fp.read()) + elif isinstance(payload_file, str): + self.payload_file = open(payload_file, "rb") + else: + self.payload_file = payload_file self.payload_file_offset = payload_file_offset self.manifest_hasher = None self.is_init = False |