fix: [decoded] fix download file

This commit is contained in:
Terrtia 2023-06-30 16:22:30 +02:00
parent 0fa0984ec4
commit 35f678245f
No known key found for this signature in database
GPG key ID: 1E1B1F50D84613D0

View file

@ -138,7 +138,7 @@ class Decoded(AbstractDaterangeObject):
with open(filepath, 'rb') as f: with open(filepath, 'rb') as f:
content = f.read() content = f.read()
return content return content
elif r_str == 'bytesio': elif r_type == 'bytesio':
with open(filepath, 'rb') as f: with open(filepath, 'rb') as f:
content = BytesIO(f.read()) content = BytesIO(f.read())
return content return content
@ -149,7 +149,7 @@ class Decoded(AbstractDaterangeObject):
with zipfile.ZipFile(zip_content, "w") as zf: with zipfile.ZipFile(zip_content, "w") as zf:
# TODO: Fix password # TODO: Fix password
# zf.setpassword(b"infected") # zf.setpassword(b"infected")
zf.writestr(self.id, self.get_content().getvalue()) zf.writestr(self.id, self.get_content(r_type='bytesio').getvalue())
zip_content.seek(0) zip_content.seek(0)
return zip_content return zip_content